40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
from apps.project.models import TestDemand
|
||
from docx.oxml import OxmlElement
|
||
from docx.oxml.ns import qn
|
||
|
||
def demand_sort_by_designKey(demand_obj: TestDemand) -> tuple[int, ...]:
|
||
"""仅限于测试项排序函数，传入sorted函数的key里面"""
|
||
parts = demand_obj.key.split('-')
|
||
sort_tuple = tuple(int(part) for part in parts)
|
||
return sort_tuple
|
||
|
||
def set_table_border(table, **kwargs):
|
||
"""docx-设置表格上下左右边框"""
|
||
# 获取或创建表格属性
|
||
tbl_pr = table._tbl.tblPr
|
||
|
||
# 查找并移除现有的边框设置
|
||
existing_borders = tbl_pr.find(qn('w:tblBorders'))
|
||
if existing_borders is not None:
|
||
tbl_pr.remove(existing_borders)
|
||
|
||
# 创建新的边框元素
|
||
borders = OxmlElement('w:tblBorders')
|
||
|
||
# 只设置外边框：top, left, bottom, right
|
||
# 不设置 insideV 和 insideH（内部边框）
|
||
for border_type in ['top', 'left', 'bottom', 'right']:
|
||
border_data = kwargs.get(border_type, {"sz": "12", "val": "single", "color": "#000000"})
|
||
border_elem = OxmlElement(f'w:{border_type}')
|
||
|
||
# 设置边框属性
|
||
border_elem.set(qn('w:val'), border_data.get('val', 'single')) # 线条类型
|
||
border_elem.set(qn('w:sz'), border_data.get('sz', '12')) # 线条粗细（8代表1磅）
|
||
border_elem.set(qn('w:color'), border_data.get('color', '#000000')) # 颜色
|
||
borders.append(border_elem) # type:ignore
|
||
|
||
# 将边框设置添加到表格属性中
|
||
tbl_pr.append(borders)
|
||
|
||
__all__ = ['demand_sort_by_designKey', 'set_table_border']
|