first commit

2025-05-16 16:15:18 +08:00
commit 6961b70a7d
11 changed files with 750 additions and 0 deletions
--- a/module/doc.py
+++ b/module/doc.py
@@ -0,0 +1,62 @@
+import pathlib
+from copy import deepcopy
+
+from docx import Document
+from docx.shared import Cm
+
+from module.schema import Course, Student, Question
+
+
+class DocPaper:
+    def __init__(self, filename: str = 'Paper', template_path: str = '../template/template.docx'):
+        self._doc = Document()
+        self._template = Document(template_path)
+        self._filename = filename
+
+        section = self._doc.sections[0]
+        section.top_margin = Cm(2)
+        section.bottom_margin = Cm(1)
+        section.left_margin = Cm(1.4)
+        section.right_margin = Cm(1.4)
+
+    def add_paper(self, course: Course, student: Student):
+        temp_table = self._template.tables[0]
+        new_table = deepcopy(temp_table)
+        para = self._doc.add_paragraph()
+        para._p.addprevious(new_table._element)
+
+        data_list = {
+            '%CNAME%': course.name,
+            '%CLASS%': student.class_name,
+            '%SNAME%': student.name,
+            '%NO%': student.no,
+            '%SO%': student.so,
+            '%Q1%': student.picked_questions[0].topic,
+            '%Q2%': student.picked_questions[1].topic,
+            '%Q3%': student.picked_questions[2].topic
+        }
+
+        # 替换表格中的占位符
+        for row in new_table.rows:
+            for cell in row.cells:
+                for para in cell.paragraphs:
+                    for run in para.runs:
+                        for key, val in data_list.items():
+                            if key in run.text:
+                                run.text = run.text.replace(key, val)
+                                break
+
+    def save(self, path: str = './'):
+        self._doc.save(str(pathlib.Path(path) / f"{self._filename}.docx"))
+
+
+if __name__ == '__main__':
+    course = Course.load_from_xls('../files/21工程管理-工程造价Ⅱ-点名册-系统0828.xlsx')
+    students = Student.load_from_xls('../files/21工程管理-工程造价Ⅱ-点名册-系统0828.xlsx')
+    questions = Question.load_from_csv()
+
+    d = DocPaper()
+    for student in students:
+        student.pick_question(questions)
+        d.add_paper(course, student)
+    d.save()