63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
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(2)
|
|
section.left_margin = Cm(2)
|
|
section.right_margin = Cm(2)
|
|
|
|
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()
|