from docx import Document
from datetime import datetime

def create_report(title, author, content):
    # 创建一个新的文档
    doc = Document()

    # 添加标题
    doc.add_heading(title, level=1)

    # 添加作者和日期
    doc.add_paragraph(f'作者：{author}')
    doc.add_paragraph(f'日期：{datetime.now().strftime("%Y年%m月%d日")}')

    # 添加内容
    for section_title, section_content in content.items():
        doc.add_heading(section_title, level=2)
        doc.add_paragraph(section_content)

    # 保存文档
    doc.save(f'{title}.docx')

# 示例数据
report_title = '2023年工作总结报告'
report_author = '张三'
report_content = {
    '摘要': '本报告总结了2023年的工作情况及未来展望。',
    '工作完成情况': '完成了多个项目，达成了预期目标。',
    '未来计划': '计划在2024年继续提高团队协作效率。',
}

# 生成报告
create_report(report_title, report_author, report_content)

def add_table(doc, data):
    # 添加表格，数据格式为嵌套列表
    table = doc.add_table(rows=1, cols=len(data[0]))
    hdr_cells = table.rows[0].cells
    for i, header in enumerate(data[0]):
        hdr_cells[i].text = header
    
    for row in data[1:]:
        row_cells = table.add_row().cells
        for i, item in enumerate(row):
            row_cells[i].text = str(item)

# 示例数据
table_data = [
    ['项目', '状态', '完成度'],
    ['项目A', '已完成', '100%'],
    ['项目B', '进行中', '50%'],
]

# 创建报告并添加表格
doc = Document()
create_report(report_title, report_author, report_content)
add_table(doc, table_data)
doc.save(f'{report_title}.docx')