# # 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) # sort only after the worker is done for efficiency self.setSortingEnabled(True) 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_())