合并达成度功能
This commit is contained in:
505
module/achievement_doc.py
Normal file
505
module/achievement_doc.py
Normal file
@@ -0,0 +1,505 @@
|
||||
import io
|
||||
import os
|
||||
import traceback
|
||||
from typing import Callable
|
||||
|
||||
from docx import Document
|
||||
from docx.enum.section import WD_ORIENT
|
||||
from docx.enum.table import WD_ALIGN_VERTICAL, WD_ROW_HEIGHT_RULE
|
||||
from docx.enum.text import WD_ALIGN_PARAGRAPH, WD_LINE_SPACING
|
||||
from docx.oxml import OxmlElement
|
||||
from docx.oxml.ns import qn
|
||||
from docx.shared import Pt, Cm
|
||||
from docx.text.run import Run
|
||||
|
||||
from module import LOGLEVEL
|
||||
from module.achievement_excel import ExcelReader
|
||||
|
||||
|
||||
class DocxWriter:
|
||||
full_file_path: str
|
||||
|
||||
def __init__(self, save_path: str, filename: str, excel_reader: ExcelReader,
|
||||
signal: Callable[[str, str], None] = lambda x, y: print(x)):
|
||||
super().__init__()
|
||||
self.save_path = save_path
|
||||
self.filename = filename
|
||||
self.excel_reader = excel_reader
|
||||
self.full_file_path = os.path.join(save_path, filename) + '.docx'
|
||||
self.signal = signal
|
||||
|
||||
def write(self) -> int:
|
||||
try:
|
||||
if not os.path.exists(self.save_path):
|
||||
os.makedirs(self.save_path)
|
||||
if os.path.exists(self.full_file_path):
|
||||
self.signal(f"文件'{self.filename}'已存在，将覆盖原文件", LOGLEVEL.WARNING)
|
||||
doc = Document()
|
||||
for section in doc.sections:
|
||||
new_width, new_height = Cm(29.7), Cm(21) # A4纸张
|
||||
# 设置纸张方向为横向
|
||||
section.orientation = WD_ORIENT.LANDSCAPE
|
||||
|
||||
# 设置页边距（例如：上下左右均为2厘米）
|
||||
section.top_margin = Cm(2)
|
||||
section.bottom_margin = Cm(2)
|
||||
section.left_margin = Cm(2)
|
||||
section.right_margin = Cm(1)
|
||||
|
||||
section.page_width = new_width
|
||||
section.page_height = new_height
|
||||
|
||||
paragraph = doc.add_paragraph()
|
||||
# 设置段落居中对齐
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||||
run = paragraph.add_run(f"《{self.excel_reader.course_name}》课程目标达成情况评价报告")
|
||||
self.set_run_font(run, 16, '黑体', '黑体', True)
|
||||
|
||||
paragraph = doc.add_paragraph()
|
||||
run = paragraph.add_run(
|
||||
f"1.班级{'（年级）' if len(self.excel_reader.class_list) == 1 else ''}课程目标达成情况分析")
|
||||
self.set_run_font(run, 14, 'Times New Roman', '黑体', True)
|
||||
|
||||
# 班级课程目标达成情况分析表
|
||||
class_info_str = "班级：{} 学生人数：{} 任课教师：{}"
|
||||
for r_index in range(len(self.excel_reader.class_list)):
|
||||
self.signal(f"正在生成 {self.excel_reader.class_list[r_index]} 表格", LOGLEVEL.INFO)
|
||||
paragraph = doc.add_paragraph()
|
||||
# 设置段落为分散对齐
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.DISTRIBUTE
|
||||
run = paragraph.add_run(class_info_str.format(
|
||||
self.excel_reader.class_list[r_index],
|
||||
self.excel_reader.class_number[r_index],
|
||||
self.excel_reader.course_teacher_name
|
||||
))
|
||||
self.set_run_font(run, 10.5, 'Times New Roman', '宋体')
|
||||
|
||||
if len(self.excel_reader.class_list) == 1:
|
||||
rows = 14 + self.excel_reader.class_number[r_index] + self.excel_reader.kpi_number
|
||||
else:
|
||||
rows = 12 + self.excel_reader.class_number[r_index] + self.excel_reader.kpi_number
|
||||
cols = 2 + self.calculate_columns(self.excel_reader.achievement_level[r_index]) # 加上第一列
|
||||
|
||||
table = doc.add_table(rows=rows, cols=cols)
|
||||
|
||||
# 设置外侧框线粗1.5磅，内侧框线粗0.5磅
|
||||
self.set_table_borders(table)
|
||||
|
||||
# 合并前六行头两个单元格
|
||||
for row in range(6):
|
||||
cell_start = table.cell(row, 0)
|
||||
cell_end = table.cell(row, 1)
|
||||
cell_start.merge(cell_end)
|
||||
# 合并前三行的单元格
|
||||
for row in range(3):
|
||||
col_span = 2 # 从第三列开始
|
||||
for performance in self.excel_reader.achievement_level[r_index]:
|
||||
non_none_count = 3 - performance.scores.count(None)
|
||||
if non_none_count > 1:
|
||||
cell_start = table.cell(row, col_span)
|
||||
cell_end = table.cell(row, col_span + non_none_count - 1)
|
||||
cell_start.merge(cell_end)
|
||||
col_span += non_none_count
|
||||
|
||||
X = self.excel_reader.kpi_number + 5 # X的值
|
||||
if len(self.excel_reader.class_list) == 1:
|
||||
X += 2
|
||||
|
||||
# 第一、二行合并第二个单元格
|
||||
for row in range(rows - X, rows - X + 2):
|
||||
table.cell(row, 0).merge(table.cell(row, 1))
|
||||
|
||||
# 第三行的合并操作
|
||||
# 第一个单元格与第三+len(a[0])-1行第二个单元格合并
|
||||
table.cell(rows - X + 2, 0).merge(table.cell(rows - X + 2 + self.excel_reader.kpi_number - 1, 1))
|
||||
for i in range(self.excel_reader.kpi_number):
|
||||
# 第三个与第四个单元格合并
|
||||
table.cell(rows - X + 2 + i, 2).merge(table.cell(rows - X + i + 2, 3))
|
||||
# 从第五个单元格合并到最后
|
||||
start_cell = table.cell(rows - X + 2 + i, 4)
|
||||
end_cell = table.cell(rows - X + 2 + i, cols - 1)
|
||||
start_cell.merge(end_cell)
|
||||
|
||||
# 第三+len(a[0])行的合并操作
|
||||
# 合并第一个到第二个单元格
|
||||
table.cell(rows - X + 2 + self.excel_reader.kpi_number, 0).merge(
|
||||
table.cell(rows - X + 2 + self.excel_reader.kpi_number, 1))
|
||||
# 合并第三个到最后一个单元格
|
||||
start_cell = table.cell(rows - X + 2 + self.excel_reader.kpi_number, 2)
|
||||
end_cell = table.cell(rows - X + 2 + self.excel_reader.kpi_number, cols - 1)
|
||||
start_cell.merge(end_cell)
|
||||
|
||||
if len(self.excel_reader.class_list) == 1:
|
||||
# 合并前两个单元格
|
||||
for i in range(2):
|
||||
start = rows - X + 3 + i + self.excel_reader.kpi_number
|
||||
table.cell(start, 0).merge(table.cell(start, 1))
|
||||
# 依次合并后面的单元格
|
||||
for row in range(start, start + self.excel_reader.kpi_number):
|
||||
col_span = 2 # 从第三列开始
|
||||
for performance in self.excel_reader.achievement_level[r_index]:
|
||||
non_none_count = 3 - performance.scores.count(None)
|
||||
if non_none_count > 1:
|
||||
cell_start = table.cell(row, col_span)
|
||||
cell_end = table.cell(row, col_span + non_none_count - 1)
|
||||
cell_start.merge(cell_end)
|
||||
col_span += non_none_count
|
||||
|
||||
start = rows - X + 3 + self.excel_reader.kpi_number
|
||||
if len(self.excel_reader.class_list) == 1:
|
||||
start += 2
|
||||
end = rows
|
||||
for row in range(start, end):
|
||||
start_cell = table.cell(row, 1)
|
||||
end_cell = table.cell(row, cols - 1)
|
||||
start_cell.merge(end_cell)
|
||||
|
||||
# 填充表格数据
|
||||
self.put_data_to_table(table, self.excel_reader.get_word_template, r_index)
|
||||
|
||||
doc.add_page_break()
|
||||
|
||||
if len(self.excel_reader.class_list) > 1:
|
||||
paragraph = doc.add_paragraph()
|
||||
run = paragraph.add_run("2.年级课程目标达成情况分析")
|
||||
self.set_run_font(run, 14, 'Times New Roman', '黑体', True)
|
||||
|
||||
# 2.年级课程目标达成情况分析
|
||||
class_info_str = "班级：{} 课程负责人：{}"
|
||||
paragraph = doc.add_paragraph()
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.DISTRIBUTE
|
||||
run = paragraph.add_run(class_info_str.format(
|
||||
self.excel_reader.total_class_str,
|
||||
self.excel_reader.course_lead_teacher_name
|
||||
))
|
||||
self.set_run_font(run, 10.5, "Times New Roman", '宋体')
|
||||
|
||||
rows = 10 + len(self.excel_reader.class_list)
|
||||
cols = 2 + self.calculate_columns(self.excel_reader.achievement_level[0]) # 加上第一列
|
||||
|
||||
table = doc.add_table(rows=rows, cols=cols)
|
||||
|
||||
# 设置外侧框线粗1.5磅，内侧框线粗0.5磅
|
||||
self.set_table_borders(table)
|
||||
|
||||
# 合并前六行头两个单元格
|
||||
for row in range(rows - 2):
|
||||
cell_start = table.cell(row, 0)
|
||||
cell_end = table.cell(row, 1)
|
||||
cell_start.merge(cell_end)
|
||||
# 合并前三行的单元格
|
||||
for row in range(3):
|
||||
col_span = 2 # 从第三列开始
|
||||
for performance in self.excel_reader.achievement_level[0]:
|
||||
non_none_count = 3 - performance.scores.count(None)
|
||||
if non_none_count > 1:
|
||||
cell_start = table.cell(row, col_span)
|
||||
cell_end = table.cell(row, col_span + non_none_count - 1)
|
||||
cell_start.merge(cell_end)
|
||||
col_span += non_none_count
|
||||
# 合并中间X行的单元格
|
||||
for row in range(6, 8 + len(self.excel_reader.class_list)):
|
||||
col_span = 2
|
||||
for performance in self.excel_reader.achievement_level[0]:
|
||||
non_none_count = 3 - performance.scores.count(None)
|
||||
if non_none_count > 1:
|
||||
cell_start = table.cell(row, col_span)
|
||||
cell_end = table.cell(row, col_span + non_none_count - 1)
|
||||
cell_start.merge(cell_end)
|
||||
col_span += non_none_count
|
||||
# 合并最后两行的单元格
|
||||
for row in range(2):
|
||||
cell_start = table.cell(rows - row - 1, 1)
|
||||
cell_end = table.cell(rows - row - 1, cols - 1)
|
||||
cell_start.merge(cell_end)
|
||||
|
||||
# 填充数据
|
||||
self.put_data_to_table(table, self.excel_reader.get_word_template_part_2)
|
||||
|
||||
doc.add_page_break()
|
||||
|
||||
# 3. 课程目标达成情况的合理性评价
|
||||
paragraph = doc.add_paragraph()
|
||||
run = paragraph.add_run(f"{3 if len(self.excel_reader.class_list) > 1 else 2}"
|
||||
f". 课程目标达成情况的合理性评价")
|
||||
self.set_run_font(run, 14, 'Times New Roman', '黑体', True)
|
||||
|
||||
rows = 9
|
||||
cols = 4
|
||||
table = doc.add_table(rows=rows, cols=cols)
|
||||
# 设置外侧框线粗1.5磅，内侧框线粗0.5磅
|
||||
self.set_table_borders(table)
|
||||
|
||||
# 合并第一行
|
||||
cell_start = table.cell(0, 0)
|
||||
cell_end = table.cell(0, cols - 1)
|
||||
cell_start.merge(cell_end)
|
||||
# 合并第二行至最后
|
||||
for i in range(1, 9):
|
||||
if i == 2:
|
||||
continue
|
||||
cell_start = table.cell(i, 1)
|
||||
cell_end = table.cell(i, cols - 1)
|
||||
cell_start.merge(cell_end)
|
||||
# 填充数据
|
||||
self.put_data_to_table(table, self.excel_reader.get_word_template_part_3)
|
||||
|
||||
# 应用样式
|
||||
self.signal("正在应用文字样式", LOGLEVEL.INFO)
|
||||
# 遍历文档中的所有段落
|
||||
for index, paragraph in enumerate(doc.paragraphs):
|
||||
if index == 0:
|
||||
self.set_paragraph_format(paragraph, WD_LINE_SPACING.ONE_POINT_FIVE)
|
||||
else:
|
||||
self.set_paragraph_format(paragraph)
|
||||
|
||||
# 遍历文档中的所有表格及其单元格
|
||||
t_len = len(doc.tables)
|
||||
part_1_table_index = [x for x in range(len(self.excel_reader.class_list))]
|
||||
part_2_table_index = [len(part_1_table_index)] if len(self.excel_reader.class_list) != 1 else []
|
||||
part_3_table_index = [t_len - 1]
|
||||
|
||||
for t_index, table in enumerate(doc.tables):
|
||||
self.set_table_borders(table)
|
||||
for r_index, row in enumerate(table.rows):
|
||||
row.height_rule = WD_ROW_HEIGHT_RULE.AT_LEAST
|
||||
row.height = Cm(0.7)
|
||||
for c_index, cell in enumerate(row.cells):
|
||||
cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
|
||||
self.set_cell_margins(cell, start=57, end=57)
|
||||
for paragraph in cell.paragraphs:
|
||||
for run in paragraph.runs:
|
||||
self.set_run_font(run, 10.5, 'Times New Roman', '宋体')
|
||||
self.set_paragraph_format(paragraph)
|
||||
special_cell = []
|
||||
if t_index in part_1_table_index:
|
||||
ll = [(9 + x + self.excel_reader.class_number[t_index], 4) for x in
|
||||
range(self.excel_reader.kpi_number)]
|
||||
start_line = 9 + self.excel_reader.kpi_number
|
||||
special_cell.extend(ll)
|
||||
special_cell.extend([
|
||||
(1, 2),
|
||||
(2, 2),
|
||||
(start_line + self.excel_reader.class_number[t_index], 2),
|
||||
])
|
||||
if len(self.excel_reader.class_list) == 1:
|
||||
special_cell.extend([
|
||||
(start_line + 3 + self.excel_reader.class_number[t_index], 1),
|
||||
(start_line + 4 + self.excel_reader.class_number[t_index], 1),
|
||||
])
|
||||
lst = [(start_line + 3 + self.excel_reader.class_number[t_index], 1)]
|
||||
else:
|
||||
special_cell.extend([
|
||||
(start_line + 1 + self.excel_reader.class_number[t_index], 1),
|
||||
(start_line + 2 + self.excel_reader.class_number[t_index], 1),
|
||||
])
|
||||
lst = [(start_line + 1 + self.excel_reader.class_number[t_index], 1)]
|
||||
if (r_index, c_index) in lst:
|
||||
self.insert_image_from_data(cell,
|
||||
self.excel_reader.pic_list[t_index],
|
||||
Cm(7 * self.excel_reader.kpi_number)
|
||||
)
|
||||
if (len(self.excel_reader.class_list) == 1 and
|
||||
r_index == start_line + 2 + self.excel_reader.class_number[t_index]):
|
||||
self.change_font_to_wingding2(paragraph)
|
||||
elif t_index in part_2_table_index:
|
||||
special_cell = [
|
||||
(1, 2),
|
||||
(2, 2),
|
||||
(8 + len(self.excel_reader.class_list), 1),
|
||||
(9 + len(self.excel_reader.class_list), 1),
|
||||
]
|
||||
if r_index == 7 + len(self.excel_reader.class_list):
|
||||
self.change_font_to_wingding2(paragraph)
|
||||
elif t_index in part_3_table_index:
|
||||
special_cell = [
|
||||
(1, 1),
|
||||
(2, 1),
|
||||
(3, 1),
|
||||
(4, 1),
|
||||
(5, 1),
|
||||
(6, 1),
|
||||
(7, 1),
|
||||
(8, 1),
|
||||
]
|
||||
if r_index == 0:
|
||||
for run in paragraph.runs:
|
||||
self.set_run_font(run, 10.5, 'Times New Roman', '宋体', True)
|
||||
else:
|
||||
self.change_font_to_wingding2(paragraph)
|
||||
if any(r_index == pair[0] and c_index >= pair[1] for pair in special_cell):
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.LEFT
|
||||
else:
|
||||
paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
|
||||
|
||||
doc.save(self.full_file_path)
|
||||
|
||||
except PermissionError as e:
|
||||
raise Exception(f"""
|
||||
简要错误信息：{str(e)}
|
||||
这可能是由于Word文件被占用所导致的，请关闭相应文件后再试。
|
||||
""")
|
||||
except Exception as _:
|
||||
error_message = traceback.format_exc()
|
||||
raise Exception(f"""
|
||||
原始错误信息：
|
||||
{error_message}
|
||||
""")
|
||||
|
||||
def set_save_path(self, save_path: str):
|
||||
self.save_path = save_path
|
||||
self.full_file_path = os.path.join(save_path, self.filename)
|
||||
|
||||
def set_filename(self, filename: str):
|
||||
self.filename = filename
|
||||
self.full_file_path = os.path.join(self.save_path, filename)
|
||||
|
||||
def set_run_font(self, run: Run, size, western_font, chinese_font, bold=False):
|
||||
run.font.size = Pt(size)
|
||||
run.font.name = western_font
|
||||
run.bold = bold
|
||||
run._element.rPr.rFonts.set(qn('w:eastAsia'), chinese_font)
|
||||
|
||||
def set_cell_border(self, cell, **kwargs):
|
||||
"""
|
||||
设置单元格边框
|
||||
kwargs: top, bottom, left, right, inside_h, inside_v
|
||||
"""
|
||||
tc = cell._tc
|
||||
tcPr = tc.get_or_add_tcPr()
|
||||
|
||||
# 创建一个新的'TC边框'元素
|
||||
tcBorders = OxmlElement('w:tcBorders')
|
||||
|
||||
# 对于每个边框方向
|
||||
for key, value in kwargs.items():
|
||||
if value is not None:
|
||||
tag = 'w:{}'.format(key)
|
||||
border = OxmlElement(tag)
|
||||
border.set(qn('w:val'), 'single')
|
||||
border.set(qn('w:sz'), str(int(value * 8)))
|
||||
border.set(qn('w:space'), '0')
|
||||
border.set(qn('w:color'), 'auto')
|
||||
tcBorders.append(border)
|
||||
|
||||
# 将边框添加到单元格属性中
|
||||
tcPr.append(tcBorders)
|
||||
|
||||
def set_paragraph_format(self, paragraph, line_spacing_rule: WD_LINE_SPACING = WD_LINE_SPACING.SINGLE):
|
||||
paragraph_format = paragraph.paragraph_format
|
||||
paragraph_format.line_spacing_rule = line_spacing_rule # 单倍行距
|
||||
paragraph_format.space_before = Pt(0) # 段前间距
|
||||
paragraph_format.space_after = Pt(0) # 段后间距
|
||||
|
||||
def calculate_columns(self, performance_list):
|
||||
"""
|
||||
计算表格的列数
|
||||
:param performance_list:
|
||||
:return:
|
||||
"""
|
||||
col_count = 0
|
||||
for performance in performance_list:
|
||||
col_count += 3 - performance.scores.count(None) # 非None值的数量
|
||||
return col_count
|
||||
|
||||
def put_data_to_table(self, table, gen_func, *args):
|
||||
"""填充表格数据"""
|
||||
data = gen_func(*args)
|
||||
for i, row in enumerate(table.rows):
|
||||
for j, cell in enumerate(row.cells):
|
||||
if cell.text == "":
|
||||
# 填充数据
|
||||
# 这里需要根据您的数据结构来调整
|
||||
d = next(data)
|
||||
cell.text = str(d)
|
||||
|
||||
def set_cell_margins(self, cell, **kwargs):
|
||||
"""
|
||||
设置单元格的边距。
|
||||
cell: 要修改的单元格实例
|
||||
kwargs: 边距值，以twips为单位（1/1440英寸）
|
||||
示例：set_cell_margins(cell, top=50, start=50, bottom=50, end=50)
|
||||
"""
|
||||
tc = cell._tc
|
||||
tcPr = tc.get_or_add_tcPr()
|
||||
tcMar = OxmlElement('w:tcMar')
|
||||
|
||||
for m in ["top", "start", "bottom", "end"]:
|
||||
if m in kwargs:
|
||||
node = OxmlElement("w:{}".format(m))
|
||||
node.set(qn('w:w'), str(kwargs.get(m)))
|
||||
node.set(qn('w:type'), 'dxa')
|
||||
tcMar.append(node)
|
||||
|
||||
tcPr.append(tcMar)
|
||||
|
||||
def is_start_of_merged_cells(self, cell):
|
||||
""" 检查单元格是否是合并单元格的起始单元格 """
|
||||
merge = cell._element.xpath('./w:tcPr/w:vMerge')
|
||||
if merge:
|
||||
return merge[0].get(qn('w:val')) == 'restart'
|
||||
return False
|
||||
|
||||
def is_merged_vertically(self, cell):
|
||||
""" 检查单元格是否垂直合并 """
|
||||
merge = cell._element.xpath('./w:tcPr/w:vMerge')
|
||||
return bool(merge)
|
||||
|
||||
def set_table_borders(self, table, outside_sz=1.5, inside_sz=0.5):
|
||||
"""
|
||||
设置整个表格的边框样式
|
||||
outside_sz: 外侧边框大小
|
||||
inside_sz: 内部分割线大小
|
||||
"""
|
||||
for row in table.rows:
|
||||
for cell in row.cells:
|
||||
self.set_cell_border(cell, top=inside_sz, bottom=inside_sz, left=inside_sz, right=inside_sz)
|
||||
|
||||
# 设置外侧边框
|
||||
for cell in table.rows[0].cells:
|
||||
self.set_cell_border(cell, top=outside_sz)
|
||||
for cell in table.rows[-1].cells:
|
||||
self.set_cell_border(cell, bottom=outside_sz)
|
||||
for row in table.rows:
|
||||
self.set_cell_border(row.cells[0], left=outside_sz)
|
||||
self.set_cell_border(row.cells[-1], right=outside_sz)
|
||||
|
||||
def insert_image_from_data(self, cell, image_data, width=Cm(21), height=Cm(4.5)):
|
||||
# 创建一个临时文件
|
||||
with io.BytesIO(image_data) as image_stream:
|
||||
image_stream.seek(0)
|
||||
# 插入图片
|
||||
# 在单元格中添加一个段落
|
||||
paragraph = cell.paragraphs[0]
|
||||
# 在段落中插入图片
|
||||
run = paragraph.add_run()
|
||||
run.add_picture(image_stream, width=width, height=height)
|
||||
|
||||
def is_chinese(self, char):
|
||||
"""判断字符是否为中文"""
|
||||
if '\u4e00' <= char <= '\u9fff':
|
||||
return True
|
||||
return False
|
||||
|
||||
def change_font_to_wingding2(self, paragraph):
|
||||
for run in paragraph.runs:
|
||||
if not run.text.strip(): # 跳过不包含文本的runs
|
||||
continue
|
||||
new_text = ""
|
||||
for char in run.text:
|
||||
if char in ['R', '£']:
|
||||
# 对于特殊字符，创建新的run
|
||||
if new_text:
|
||||
new_run = paragraph.add_run(new_text)
|
||||
self.set_run_font(new_run, 10.5, 'Times New Roman', '宋体')
|
||||
new_text = ""
|
||||
special_run = paragraph.add_run(char)
|
||||
self.set_run_font(special_run, 10.5, 'Wingdings 2', '宋体')
|
||||
else:
|
||||
# 累积其他字符
|
||||
new_text += char
|
||||
if new_text:
|
||||
# 创建剩余文本的run
|
||||
new_run = paragraph.add_run(new_text)
|
||||
self.set_run_font(new_run, 10.5, 'Times New Roman', '宋体')
|
||||
run.clear()
|
||||
Reference in New Issue
Block a user