#!/usr/bin/env python3
"""
专业报告模版生成脚本 V2.2
修复循环表格语法（poi-tl 需要标记行 + 模板行的结构）
"""

from docx import Document
from docx.shared import Pt, Cm
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.enum.table import WD_TABLE_ALIGNMENT
from docx.oxml.ns import qn
from docx.oxml import OxmlElement
import os


def set_font(run, name='宋体', size=12, bold=False):
    """设置字体"""
    run.font.name = name
    run._element.rPr.rFonts.set(qn('w:eastAsia'), name)
    run.font.size = Pt(size)
    run.bold = bold


def add_heading(doc, text, size=14):
    """添加章节标题"""
    para = doc.add_paragraph()
    run = para.add_run(text)
    set_font(run, '黑体', size, bold=True)
    para.paragraph_format.space_before = Pt(12)
    para.paragraph_format.space_after = Pt(6)
    return para


def add_body(doc, text, indent=True):
    """添加正文段落"""
    para = doc.add_paragraph()
    run = para.add_run(text)
    set_font(run, '宋体', 12)
    if indent:
        para.paragraph_format.first_line_indent = Cm(0.74)
    return para


def add_label_para(doc, label, value):
    """添加 标签：值 段落"""
    para = doc.add_paragraph()
    label_run = para.add_run(label)
    set_font(label_run, '宋体', 12, bold=True)
    value_run = para.add_run(value)
    set_font(value_run, '宋体', 12)
    return para


def set_cell_background(cell, color):
    """设置单元格背景色"""
    shading = OxmlElement('w:shd')
    shading.set(qn('w:val'), 'clear')
    shading.set(qn('w:fill'), color)
    cell._tc.get_or_add_tcPr().append(shading)


def set_cell_vertical_center(cell):
    """单元格内容垂直居中"""
    vAlign = OxmlElement('w:vAlign')
    vAlign.set(qn('w:val'), 'center')
    cell._tc.get_or_add_tcPr().append(vAlign)


def style_header_cell(cell, bg_color='DCE6F1'):
    """表头单元格样式：加粗 + 居中 + 浅蓝底纹 + 垂直居中"""
    set_cell_background(cell, bg_color)
    set_cell_vertical_center(cell)
    cell.paragraphs[0].alignment = WD_ALIGN_PARAGRAPH.CENTER
    for run in cell.paragraphs[0].runs:
        run.bold = True


def style_body_cell(cell, align=WD_ALIGN_PARAGRAPH.CENTER):
    """数据单元格样式：垂直居中"""
    set_cell_vertical_center(cell)
    cell.paragraphs[0].alignment = align


def set_table_widths(table, widths_cm):
    """设置表格列宽（厘米）"""
    for row in table.rows:
        for i, width in enumerate(widths_cm):
            if i < len(row.cells):
                row.cells[i].width = Cm(width)


def create_professional_template(template_name, device_type, template_dir):
    """创建专业报告模版"""

    doc = Document()

    # 页面设置：A4 + 标准页边距
    section = doc.sections[0]
    section.page_width = Cm(21.0)
    section.page_height = Cm(29.7)
    section.left_margin = Cm(2.54)
    section.right_margin = Cm(2.54)
    section.top_margin = Cm(2.54)
    section.bottom_margin = Cm(2.54)

    # 设置文档默认样式：宋体小四、1.5 倍行距
    style = doc.styles['Normal']
    style.font.name = '宋体'
    style._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体')
    style.font.size = Pt(12)
    style.paragraph_format.line_spacing = 1.5

    # ===== 1. 标题 =====
    title = doc.add_paragraph()
    title.alignment = WD_ALIGN_PARAGRAPH.CENTER
    run = title.add_run(f'{device_type}监测分析报告')
    set_font(run, '黑体', 18, bold=True)
    title.paragraph_format.space_after = Pt(12)

    # ===== 2. 基本信息 =====
    info_para = doc.add_paragraph()
    r1 = info_para.add_run('报告编号：')
    set_font(r1, '宋体', 12, bold=True)
    r2 = info_para.add_run('{{report.no}}')
    set_font(r2, '宋体', 12)
    r3 = info_para.add_run('    生成时间：')
    set_font(r3, '宋体', 12, bold=True)
    r4 = info_para.add_run('{{report.generatedAt}}')
    set_font(r4, '宋体', 12)

    add_label_para(doc, '报告周期：', '{{report.periodDesc}}')
    add_label_para(doc, '设备信息：', '设备名称 {{device.name}}｜设备编号 {{device.serialNumber}}｜所属产品 {{device.productName}}')
    add_label_para(doc, '监测指标：', '{{device.metricName}}')
    add_label_para(doc, '控制值：', '{{controlValue}} {{controlValueUnit}}')

    # ===== 3. 项目概况 =====
    add_heading(doc, '一、项目概况')
    add_body(doc, '{{summary}}')

    # ===== 4. 安全评估 =====
    add_heading(doc, '二、安全评估')
    add_label_para(doc, '安全等级：', '{{safetyLevel}}')
    add_label_para(doc, '评估说明：', '{{safetyDescription}}')

    # ===== 5. 统计指标 =====
    add_heading(doc, '三、统计指标')

    table = doc.add_table(rows=10, cols=2)
    table.style = 'Table Grid'
    table.alignment = WD_TABLE_ALIGNMENT.CENTER

    header_cells = table.rows[0].cells
    header_cells[0].text = '统计指标'
    header_cells[1].text = '数值'
    style_header_cell(header_cells[0])
    style_header_cell(header_cells[1])

    metrics = [
        ('当前值', '{{analysis.statistics.currentWithUnit}}'),
        ('初始值', '{{analysis.statistics.initialWithUnit}}'),
        ('最大值', '{{analysis.statistics.maxWithUnit}}'),
        ('最小值', '{{analysis.statistics.minWithUnit}}'),
        ('平均值', '{{analysis.statistics.avgWithUnit}}'),
        ('中位数', '{{analysis.statistics.medianWithUnit}}'),
        ('变化量', '{{analysis.statistics.deltaWithUnit}}'),
        ('累计变化', '{{analysis.statistics.totalDeltaWithUnit}}'),
        ('数据量', '{{analysis.statistics.count}}'),
    ]

    for i, (label, value) in enumerate(metrics, start=1):
        row = table.rows[i]
        row.cells[0].text = label
        row.cells[1].text = value
        style_body_cell(row.cells[0])
        style_body_cell(row.cells[1])

    set_table_widths(table, [4.5, 11.0])

    # ===== 6. 测点数据（循环表格） =====
    add_heading(doc, '四、测点数据')

    # poi-tl LoopRowTableRenderPolicy 需要的结构：
    # 第1行：表头
    # 第2行：{{listName}} 标记行（可以是空单元格）
    # 第3行：模板行，使用 [field] 语法
    points_table = doc.add_table(rows=3, cols=8)
    points_table.style = 'Table Grid'
    points_table.alignment = WD_TABLE_ALIGNMENT.CENTER

    # 表头
    points_headers = ['测点', '初始值', '当前值', '变化量', '变化速率', '最大值', '最小值', '预警状态']
    points_header_row = points_table.rows[0]
    for i, header in enumerate(points_headers):
        points_header_row.cells[i].text = header
        style_header_cell(points_header_row.cells[i])

    # 标记行：{{metricPoints}}
    points_marker_row = points_table.rows[1]
    points_marker_row.cells[0].text = '{{metricPoints}}'

    # 模板行：[field] 语法
    points_data_row = points_table.rows[2]
    points_data = [
        '[modelName]',
        '[initialValue]',
        '[currentValue]',
        '[changeValue]',
        '[changeRate]',
        '[maxValue]',
        '[minValue]',
        '[warningStatus]',
    ]
    for i, data in enumerate(points_data):
        points_data_row.cells[i].text = data
        style_body_cell(points_data_row.cells[i])

    set_table_widths(points_table, [2.6, 1.84, 1.84, 1.84, 1.84, 1.84, 1.84, 1.84])

    # ===== 7. 趋势分析 =====
    add_heading(doc, '五、趋势分析')
    add_body(doc, '{{analysis.trend.description}}')

    # ===== 8. 数据质量 =====
    add_heading(doc, '六、数据质量')
    add_body(doc, '{{analysis.quality.description}}')

    # ===== 9. 预警状态 =====
    add_heading(doc, '七、预警状态')
    add_label_para(doc, '状态：', '{{analysis.warning.status}}')
    add_body(doc, '{{analysis.warning.description}}', indent=False)

    # ===== 10. 时程曲线（主指标 + 分组指标） =====
    add_heading(doc, '八、时程曲线')

    chart_para1 = doc.add_paragraph()
    chart_para1.alignment = WD_ALIGN_PARAGRAPH.CENTER
    chart_para1.add_run('{{@charts.timeSeriesChart}}')

    chart_para2 = doc.add_paragraph()
    chart_para2.alignment = WD_ALIGN_PARAGRAPH.CENTER
    chart_para2.add_run('{{@charts.chart2}}')

    chart_para3 = doc.add_paragraph()
    chart_para3.alignment = WD_ALIGN_PARAGRAPH.CENTER
    chart_para3.add_run('{{@charts.chart3}}')

    # ===== 11. 专业分析建议 =====
    add_heading(doc, '九、专业分析建议')
    add_body(doc, '{{professionalAdvice}}')

    # ===== 12. 结论（循环表格） =====
    add_heading(doc, '十、结论')

    # poi-tl LoopRowTableRenderPolicy 结构
    conclusion_table = doc.add_table(rows=3, cols=2)
    conclusion_table.style = 'Table Grid'
    conclusion_table.alignment = WD_TABLE_ALIGNMENT.CENTER

    # 表头
    conclusion_header_row = conclusion_table.rows[0]
    conclusion_header_row.cells[0].text = '序号'
    conclusion_header_row.cells[1].text = '结论内容'
    style_header_cell(conclusion_header_row.cells[0])
    style_header_cell(conclusion_header_row.cells[1])

    # 标记行
    conclusion_marker_row = conclusion_table.rows[1]
    conclusion_marker_row.cells[0].text = '{{conclusions}}'

    # 模板行
    conclusion_data_row = conclusion_table.rows[2]
    conclusion_data_row.cells[0].text = '[seq]'
    conclusion_data_row.cells[1].text = '[text]'
    style_body_cell(conclusion_data_row.cells[0])
    style_body_cell(conclusion_data_row.cells[1], WD_ALIGN_PARAGRAPH.LEFT)

    set_table_widths(conclusion_table, [1.5, 14.0])

    # 保存文档
    output_path = os.path.join(template_dir, f'{template_name}_V2.docx')
    doc.save(output_path)
    print(f'已生成模版: {output_path}')
    return output_path


def main():
    """主函数"""
    template_dir = '/Users/admin/IdeaProjects/fastbee/fastbee-report/src/main/resources/templates'

    device_types = {
        'DEVICE_CRACK_METER': '裂缝计',
        'DEVICE_DEEP_DISPLACEMENT': '深部位移',
        'DEVICE_AXIAL_FORCE': '轴力/应力',
        'DEVICE_STATIC_LEVEL': '静力水准',
        'DEVICE_VIBRATING_WIRE': '振弦',
        'DEVICE_STRAIN_GAUGE': '应变计',
        'DEVICE_WATER_LEVEL': '水位计',
        'DEVICE_RANGING_METER': '测距仪',
        'DEVICE_INCLINOMETER': '倾角计',
        'DEVICE_THERMOMETER': '温度计',
        'DEVICE_RAIN_GAUGE': '雨量计',
    }

    print('开始生成专业报告模版 V2.2（修复循环表格语法）...')
    print(f'模版目录: {template_dir}')
    print()

    for template_name, device_type in device_types.items():
        create_professional_template(template_name, device_type, template_dir)

    print()
    print('所有模版生成完成！')


if __name__ == '__main__':
    main()