修改界面

module/worker.py

@@ -0,0 +1,44 @@
+import os
+from pathlib import Path
+
+from PySide6.QtCore import QObject, Signal
+
+from module.doc import DocPaper
+from module.schema import Course, Student, Question
+from utils.function import resource_path
+
+
+class DTGWorker(QObject):
+    progress = Signal(int)
+    finished = Signal()
+
+    def __init__(
+            self,
+            input_student_filepath: str,
+            input_question_filepath: str,
+            output_filepath: str,
+            output_filename: str
+    ):
+        super().__init__()
+        self.input_filepath = input_student_filepath
+        self.input_question_filepath = input_question_filepath
+        self.output_filepath = output_filepath
+        self.output_filename = output_filename
+
+    def run(self):
+        course = Course.load_from_xls(self.input_filepath)
+        students = Student.load_from_xls(self.input_filepath)
+        questions = Question.load_from_xls(self.input_question_filepath)
+
+        d = DocPaper(self.output_filename, template_path=resource_path("template/template.docx"))
+        for index, student in enumerate(students):
+            if (p := int((index + 1) / len(students) * 100)) != 100:
+                self.progress.emit(p)
+            else:
+                self.progress.emit(99)
+            student.pick_question(questions)
+            d.add_paper(course, student)
+        d.save(self.output_filepath)
+        self.progress.emit(100)
+        os.startfile(Path(self.output_filepath) / f"{self.output_filename}.docx")
+        self.finished.emit()