Move into version control

1 files changed, 223 insertions, 0 deletions

diff --git a/gui.py b/gui.py
new file mode 100644
index 0000000..ffa8116
--- /dev/null
+++ b/gui.py
@@ -0,0 +1,223 @@
+#
+# This document *and this document only* uses the Qt naming convention instead
+# of PEP because the PyQt5 bindings do not have pythonic names
+#
+
+import sys
+import json
+import enum
+import html
+import logging
+
+from PyQt5.QtGui import QFont
+from PyQt5.Qt import QStyle
+from PyQt5.QtCore import Qt, QThread, pyqtSlot
+from PyQt5.QtWidgets import QApplication, QWidget, QTreeWidget, QTreeWidgetItem, QGridLayout, QHBoxLayout, QPushButton, QProgressBar, QTabWidget, QPlainTextEdit
+
+import moodle
+
+log = logging.getLogger("muddle.gui")
+
+class MoodleItem:
+    class Type(enum.Enum):
+        ROOT       = 0
+        # root
+        COURSE     = 1
+        # sections
+        SECTION    = 2
+        # modules
+        MODULE     = 3
+        FORUM      = 3
+        RESOURCE   = 3
+        FOLDER     = 3
+        ATTENDANCE = 3
+        LABEL      = 3
+        QUIZ       = 3
+        # contents
+        CONTENT    = 4
+        FILE       = 4
+        URL        = 4
+
+    def __init__(self, nodetype, leaves=[], **kwargs):
+        self.leaves = leaves
+        self.type = nodetype
+
+        # TODO: check required attributes
+        for k, v in kwargs.items():
+            setattr(self, k, v)
+
+        self.setupQt()
+
+    # TODO: Qt objects should be on the main thread
+    # prob cause of the crash
+    def setupQt(self):
+        self.qt = QTreeWidgetItem()
+
+        font = QFont("Monospace")
+        font.setStyleHint(QFont.Monospace)
+        self.qt.setFont(0, font)
+
+        icons = {
+            MoodleItem.Type.COURSE   : QApplication.style().standardIcon(QStyle.SP_DriveNetIcon),
+            MoodleItem.Type.FOLDER   : QApplication.style().standardIcon(QStyle.SP_DirIcon),
+            MoodleItem.Type.FILE     : QApplication.style().standardIcon(QStyle.SP_FileIcon),
+            MoodleItem.Type.URL      : QApplication.style().standardIcon(QStyle.SP_FileLinkIcon),
+        }
+
+        if icons.get(self.type):
+            self.qt.setIcon(0, icons[self.type])
+
+        if self.type == MoodleItem.Type.FILE:
+            flags = self.qt.flags()
+            self.qt.setFlags(flags | Qt.ItemIsUserCheckable)
+            self.qt.setCheckState(0, Qt.Unchecked)
+
+        self.qt.setChildIndicatorPolicy(QTreeWidgetItem.DontShowIndicatorWhenChildless)
+        self.qt.setText(0, html.unescape(self.title))
+
+    def insert(self, node):
+        self.leaves.append(node)
+        if not self.type == MoodleItem.Type.ROOT:
+            self.qt.addChild(node.qt)
+
+    def remove(self, node):
+        self.leaves.remove(node)
+        self.qt.removeChild(node.qt)
+        # TODO: remove from child
+
+class MoodleFetcher(QThread):
+    def __init__(self, parent, instance_url, token):
+        super().__init__()
+
+        self.api = moodle.RestApi(instance_url, token)
+        self.apihelper = moodle.ApiHelper(self.api)
+        self.moodleItems = MoodleItem(MoodleItem.Type.ROOT, title="ROOT")
+
+    def run(self):
+        # This is beyond bad but I don't have access to the moodle documentation, so I had to guess
+        courses = self.api.core_enrol_get_users_courses(userid=self.apihelper.get_userid()).json()
+        for course in courses:
+            courseItem = MoodleItem(MoodleItem.Type.COURSE, 
+                    id = course["id"], title = course["shortname"])
+
+            self.moodleItems.insert(courseItem)
+
+            sections = self.api.core_course_get_contents(courseid=courseItem.id).json()
+
+            for section in sections:
+                sectionItem = MoodleItem(MoodleItem.Type.SECTION,
+                        id = section["id"], title = section["name"])
+
+                courseItem.insert(sectionItem)
+
+                modules = section["modules"] if "modules" in section else []
+                for module in modules:
+                    moduleType = {
+                        "folder"     : MoodleItem.Type.FOLDER,
+                        "resource"   : MoodleItem.Type.RESOURCE,
+                        "forum"      : MoodleItem.Type.FORUM,
+                        "attendance" : MoodleItem.Type.ATTENDANCE,
+                        "label"      : MoodleItem.Type.LABEL,
+                        "quiz"       : MoodleItem.Type.QUIZ,
+                    }
+
+                    moduleItem = MoodleItem(moduleType.get(module["modname"]) or MoodleItem.Type.MODULE,
+                            id = module["id"], title = module["name"])
+
+                    sectionItem.insert(moduleItem)
+
+                    contents = module["contents"] if "contents" in module else []
+                    for content in contents:
+                        contentType = {
+                            "url" : MoodleItem.Type.URL,
+                            "file" : MoodleItem.Type.FILE,
+                        }
+
+                        contentItem = MoodleItem(contentType.get(content["type"]) or MoodleItem.Type.MODULE,
+                                title = content["filename"], url = content["fileurl"])
+
+                        moduleItem.insert(contentItem)
+
+
+class MoodleTreeView(QTreeWidget):
+    def __init__(self, parent, instance_url, token):
+        super().__init__(parent)
+
+        self.worker = MoodleFetcher(self, instance_url, token)
+        self.worker.finished.connect(self.onWorkerDone)
+        self.worker.start()
+
+        self.initUi()
+        self.show()
+
+    def initUi(self):
+        self.setHeaderHidden(True)
+        self.itemDoubleClicked.connect(self.onItemDoubleClicked, Qt.QueuedConnection)
+
+    @pyqtSlot(QTreeWidgetItem, int)
+    def onItemDoubleClicked(self, item, col):
+        log.debug(f"double clicked on item with type {str(item.type)}")
+        if item.type == MoodleItem.Type.FILE:
+            pass
+
+    @pyqtSlot()
+    def onWorkerDone(self):
+        log.debug("worker done")
+        for leaf in self.worker.moodleItems.leaves:
+            self.addTopLevelItem(leaf.qt)
+
+class QPlainTextEditLogger(logging.Handler):
+    def __init__(self, parent):
+        super().__init__()
+        self.widget = QPlainTextEdit(parent)
+        self.widget.setReadOnly(True)
+
+    def emit(self, record):
+        msg = self.format(record)
+        self.widget.appendPlainText(msg)
+
+    def write(self, m):
+        pass
+
+class Muddle(QTabWidget):
+    def __init__(self, instance_url, token):
+        super().__init__()
+        self.instance_url = instance_url
+        self.token = token
+        self.initUi()
+
+    def initUi(self):
+        self.setWindowTitle("Muddle")
+
+        # moodle tab
+        self.tabmoodle = QWidget()
+        self.addTab(self.tabmoodle, "Moodle")
+
+        self.tabmoodle.setLayout(QGridLayout())
+        self.tabmoodle.layout().addWidget(MoodleTreeView(self, self.instance_url, self.token), 0, 0, 1, -1)
+
+        self.tabmoodle.downloadbtn = QPushButton("Download")
+        self.tabmoodle.selectallbtn = QPushButton("Select All")
+        self.tabmoodle.deselectallbtn = QPushButton("Deselect All")
+        self.tabmoodle.progressbar = QProgressBar()
+
+        self.tabmoodle.layout().addWidget(self.tabmoodle.downloadbtn, 1, 0)
+        self.tabmoodle.layout().addWidget(self.tabmoodle.selectallbtn, 1, 1)
+        self.tabmoodle.layout().addWidget(self.tabmoodle.deselectallbtn, 1, 2)
+        self.tabmoodle.layout().addWidget(self.tabmoodle.progressbar, 2, 0, 1, -1)
+
+        # log tabs
+        handler = QPlainTextEditLogger(self)
+        handler.setFormatter(logging.Formatter("%(name)s - %(levelname)s - %(message)s"))
+        logging.getLogger("muddle").addHandler(handler)
+
+        self.tablogs = handler.widget
+        self.addTab(self.tablogs, "Logs")
+
+        self.show()
+
+
+def start(instance_url, token):
+    app = QApplication(sys.argv)
+    ex = Muddle(instance_url, token)
+    sys.exit(app.exec_())