45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
|
from docx.oxml.ns import qn # qn作用是元素的.tag属性，自动帮你处理namespace
|
|||
|
|
from docx.shape import InlineShape
|
|||
|
|
from apps.createSeiTaiDocument.docXmlUtils import (
|
|||
|
|
Demand_table_xqms,
|
|||
|
|
Timing_diagram_width,
|
|||
|
|
Test_result_width,
|
|||
|
|
Horizatal_width
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def set_shape_size(shape: InlineShape):
|
|||
|
|
"""调用下面辅助函数，判断字典{'in_table': True, 'row_idx': 10, 'col_idx': 3}来设置大小"""
|
|||
|
|
shape_location = get_shape_location(shape)
|
|||
|
|
# 先判断是否在table中
|
|||
|
|
if shape_location['in_table']:
|
|||
|
|
# 在table中看是否是第一列，第一列则是时序图
|
|||
|
|
if shape_location['col_idx'] == 0:
|
|||
|
|
# 在第一列：说明是时序图
|
|||
|
|
shape.width = Timing_diagram_width
|
|||
|
|
else:
|
|||
|
|
shape.width = Test_result_width
|
|||
|
|
else:
|
|||
|
|
shape.width = Horizatal_width
|
|||
|
|
|
|||
|
|
def get_shape_location(shape: InlineShape):
|
|||
|
|
"""传入图片直接处理，注意是python-docx库，不是docxtpl"""
|
|||
|
|
# 获取父元素链
|
|||
|
|
parent_chain = list(shape._inline.iterancestors())
|
|||
|
|
# 检查是否在表格中
|
|||
|
|
for elem in parent_chain:
|
|||
|
|
if elem.tag == qn("w:tbl"):
|
|||
|
|
# 获取表格对象
|
|||
|
|
tbl = elem
|
|||
|
|
# 获取行对象并计算行索引
|
|||
|
|
tr = next(e for e in parent_chain if e.tag == qn('w:tr'))
|
|||
|
|
row_idx = tbl.index(tr)
|
|||
|
|
# 获取单元格对象并计算列索引
|
|||
|
|
tc = next(e for e in parent_chain if e.tag == qn('w:tc'))
|
|||
|
|
col_idx = tr.index(tc)
|
|||
|
|
return {
|
|||
|
|
'in_table': True,
|
|||
|
|
'row_idx': row_idx,
|
|||
|
|
'col_idx': col_idx
|
|||
|
|
}
|
|||
|
|
return {'in_table': False}
|