73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
|
"""
|
||
|
|
表格渲染器
|
||
|
|
"""
|
||
|
|
|
||
|
|
from typing import List
|
||
|
|
import pandas as pd
|
||
|
|
from pptx.slide import Slide
|
||
|
|
from ppt_station.models.job import TableSpec
|
||
|
|
from ppt_station.utils.template_utils import find_shape_by_name
|
||
|
|
|
||
|
|
|
||
|
|
class TableRenderer:
|
||
|
|
"""表格渲染器"""
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def render(slide: Slide, table_spec: TableSpec, df: pd.DataFrame) -> None:
|
||
|
|
"""
|
||
|
|
渲染表格到幻灯片
|
||
|
|
|
||
|
|
Args:
|
||
|
|
slide: 幻灯片对象
|
||
|
|
table_spec: 表格配置
|
||
|
|
df: 数据 DataFrame
|
||
|
|
"""
|
||
|
|
shape = find_shape_by_name(slide, table_spec.target)
|
||
|
|
|
||
|
|
if not shape:
|
||
|
|
print(f"警告: 未找到表格 '{table_spec.target}'")
|
||
|
|
return
|
||
|
|
|
||
|
|
if not shape.has_table:
|
||
|
|
print(f"警告: '{table_spec.target}' 不是表格")
|
||
|
|
return
|
||
|
|
|
||
|
|
table = shape.table
|
||
|
|
data = df[table_spec.columns]
|
||
|
|
|
||
|
|
# 写入表头
|
||
|
|
row_start = 0
|
||
|
|
if table_spec.header:
|
||
|
|
for j, header in enumerate(table_spec.header):
|
||
|
|
if j < len(table.columns):
|
||
|
|
cell = table.cell(0, j)
|
||
|
|
cell.text = str(header)
|
||
|
|
row_start = 1
|
||
|
|
|
||
|
|
# 写入数据
|
||
|
|
for i in range(len(data)):
|
||
|
|
if i + row_start >= len(table.rows):
|
||
|
|
# 如果表格行数不够，需要添加行
|
||
|
|
print(f"警告: 表格 '{table_spec.target}' 行数不足，需要 {i + row_start + 1} 行")
|
||
|
|
break
|
||
|
|
|
||
|
|
for j in range(len(table_spec.columns)):
|
||
|
|
if j < len(table.columns):
|
||
|
|
cell = table.cell(i + row_start, j)
|
||
|
|
value = data.iat[i, j]
|
||
|
|
|
||
|
|
# 格式化数值
|
||
|
|
if table_spec.number_format and table_spec.columns[j] in table_spec.number_format:
|
||
|
|
fmt = table_spec.number_format[table_spec.columns[j]]
|
||
|
|
if isinstance(value, (int, float)):
|
||
|
|
if "%" in fmt:
|
||
|
|
cell.text = f"{value:.2%}"
|
||
|
|
elif "#,##0" in fmt:
|
||
|
|
cell.text = f"{value:,.0f}"
|
||
|
|
else:
|
||
|
|
cell.text = str(value)
|
||
|
|
else:
|
||
|
|
cell.text = str(value)
|
||
|
|
else:
|
||
|
|
cell.text = str(value)
|