python之使用docxtpl渲染word模板

word模板内容

docxtpl 使用的 Jinja2 模板语法

A1内容:{{A1}}
A2内容:{{A2}}
A3内容:{{A3}}

人员信息:

姓名年龄性别
{%tr for item in person %}
{{item.name}}{{item.age}}{{item.sex}}
{%tr endfor %}

安全信息:

IP地址端口规则名称
{%tr for item in secure %}
{{item.ip}}{{item.port}}{{item.rule}}
{%tr endfor %}

列表数据:
{% for val in dataList -%}
{{val}}
{% endfor -%}

python代码

# -*- coding: utf-8 -*-

from docxtpl import DocxTemplate
import dmPython

# ---------------------- 1. 达梦数据库配置 ----------------------
DM_CONFIG = {
    "user": "SYSDBA",          # 用户名
    "password": "SYSDBA",      # 密码
    "server": "localhost",     # IP
    "port": 5236,              # 端口
    "database": "DAMENG"       # 库名
}


def get_dm_data(sql):
    """从达梦查询数据,返回 docxtpl 需要的字典"""
    global cursor
    conn = None
    try:
        # 连接达梦
        conn = dmPython.connect(**DM_CONFIG)
        cursor = conn.cursor()

        # 写你的查询 SQL
        # sql = 'SELECT id, name, age, phone, address, create_time FROM user_info WHERE id = 1001'
        cursor.execute(sql)
        # 获取列名
        columns = [desc[0] for desc in cursor.description]
        # 获取一行数据
        # row = cursor.fetchone()
        # 获取所有数据
        rows = cursor.fetchall()

        # 自动组装成字典:{字段名: 值}
        # data_dict = dict(zip(columns, row))

        # 转为 [{}, {}, {}] 格式(docxtpl 循环专用)
        # 返回 docxtpl 需要的字典,需要根据业务需要进行组装
        data_dict = [dict(zip(columns, row)) for row in rows]

        return data_dict
    finally:
        if conn:
            cursor.close()
            conn.close()


def template_render(doc, context):
    doc.render(context)


def generate_example():
    pass


if __name__ == '__main__':
    # 1. 加载 Word 模板
    document = DocxTemplate("template.docx")

    # 2. 从达梦数据库获取要替换的数据
    """
      json中的key为模板中的占位符,value为实际替换后的内容
      注意:数据库返回的结果集可能只有一层JSON数据,比如:{"name": "张三","age":"11","sex":"man"},实际需要如2-1的样本数据,需要单独组装数据
    """
    # 写你的查询 SQL
    sql = 'SELECT id, name, age, phone, address, create_time FROM user_info WHERE id = 1001'
    # data = get_dm_data(sql)

    analyzer = dict(A1="A1值",A2="A2值",A3="A3值",A4=4,A5=5)
    table1 = [{"name": "张三","age":"11","sex":"man"},{"name": "王五","age":"13","sex":"woman"}]
    table2 = [{"ip": "192.168.0.11","port":"8081","rule":"test"},{"ip": "192.168.0.12","port":"8082","rule":"test2"},{"ip": "192.168.0.13","port":"8083","rule":"tes3"}]
    dataList = ["项一","项二","项三","项四"]

    data = {**analyzer,"person": table1,"secure": table2,"dataList": dataList}

    # 3. 执行替换
    template_render(document, data)

    # 4. 保存生成的新文档,名字自定义
    document.save("生成结果.docx")
    print("Word 文档生成完成!")
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值