• 使用pyhton的python-docx框架实现导出数据表结构为word格式

    导出数据库表文档到word。

    使用pyhton的python-docx框架实现导出数据库表结构为word格式。

    前言： 

            本人自学python，语法不精通，参考人员注意甄别。

    问题：

            由于近期甲方项目需要验收，需要完成数据库er，和数据库文档，由于没有搜索到需要的插件/方案（网上只有半自动化，每次只能自己手动操作一个个导出，效率太低）故自己尝试手写了一个。

    实现样式如下：

    下面是单个的样式，代码内部会遍历所有的表，有多少个表就会往word内写入多少个表格

    后面只需要自己配置头部的公司信息，和其他自己所需要的文件即可，最后手动生成目录

     上python代码：

    部分代码是写死的，例如（上图）表左，表头

    1. # -*- coding:utf-8 -*-
    2. # 数据库验收文档导出er关系表信息
    3. import sys
    4. import os
    5. from docx.opc.oxml import parse_xml
    6. from docx.oxml.ns import nsdecls
    7. from docx.shared import RGBColor, Cm, Pt
    8. curPath = os.path.abspath(os.path.dirname(__file__))
    9. rootPath = os.path.split(curPath)[0]
    10. sys.path.append(rootPath)
    11. import pymysql
    12. from pymysql.cursors import DictCursor
    13. from docx import Document
    14.  
    15. # mysql超时时间设置
    16. connect_timeout = 10
    17.  
    18. # 创建mysql链接
    19. def connMysql():
    20.     return pymysql.connect(host="url地址", port=数据库端口, user="数据库用户名",password="数据库密码", db="库名称", charset="utf8")
    21.  
    22.  
    23. # 获取数据库所有表名称
    24. def getAllDBName():
    25.     conn = connMysql()
    26.     # 建立游标，制定游标类型，返回字典
    27.     cur = conn.cursor(DictCursor);
    28.     # 执行sql语句
    29.     cur.execute("select table_name,table_comment from information_schema.tables where table_schema='数据库名称'")  # param
    30.     # 返回查询所有
    31.     res = cur.fetchall()
    32.     # 关闭游标
    33.     cur.close()
    34.     # 关闭连接
    35.     conn.close()
    36.     # print("当前库下所有表的名称")
    37.     # for i in res:
    38.     #     print(i["table_name"])
    39.     return res
    40.  
    41.  
    42. # 获取当前表描述
    43. def getAllDBDDL(dbName, dbComment):
    44.     conn = connMysql()
    45.     # 建立游标，制定游标类型，返回字典
    46.     cur = conn.cursor(DictCursor);
    47.     # 执行sql语句
    48.     cur.execute(
    49.         "SELECT COLUMN_NAME 列名, COLUMN_TYPE 数据类型,DATA_TYPE 字段类型,IS_NULLABLE 是否为空,COLUMN_DEFAULT 默认值,COLUMN_COMMENT 备注 FROM INFORMATION_SCHEMA.COLUMNS WHERE table_schema = 'zhongtie' AND table_name = '%s'" % (
    50.             dbName))  # param
    51.     # 返回查询所有
    52.     res = cur.fetchall()
    53.     # 关闭游标
    54.     cur.close()
    55.     # 关闭连接
    56.     conn.close()
    57.     print(dbName + "[%s]" % (dbComment))
    58.     for i in res:
    59.         print(i)
    60.     return res
    61.  
    62.  
    63.  
    64. shading_list = locals()
    65.  
    66.  
    67. def leftComment(doc):
    68.     # 获取第一行第二列单元格
    69.     cell = doc.cell(0, 0)
    70.     # 设置单元格文本
    71.     cell.text = "表名"
    72.     cell = doc.cell(1, 0)
    73.     # 设置单元格文本
    74.     cell.text = "库名"
    75.     cell = doc.cell(2, 0)
    76.     # 设置表格背景色
    77.     for rgbsIndex in range(0, 3):
    78.         shading_list['shading_elm_' + str(i)] = parse_xml(
    79.             r''.format(nsdecls('w'), bgColor="#C0C0C0"))
    80.         doc.rows[rgbsIndex].cells[0]._tc.get_or_add_tcPr().append(shading_list['shading_elm_' + str(i)])
    81.     # 设置单元格文本
    82.     cell.text = "内容描述"
    83.  
    84.  
    85. def rightComment(doc, dbname):
    86.     cell = doc.cell(0, 1)
    87.     # 设置单元格文本
    88.     cell.text = dbname
    89.     cell = doc.cell(1, 1)
    90.     # 设置单元格文本
    91.     cell.text = "zhongtie"
    92.     cell = doc.cell(2, 1)
    93.     # 设置单元格文本
    94.     cell.text = "用于存储组数据。"
    95.  
    96.  
    97. def topComment(doc):
    98.     cell = doc.cell(0, 0)
    99.     # 设置单元格文本
    100.     cell.text = "序号"
    101.     cell = doc.cell(0, 1)
    102.     # 设置单元格文本
    103.     cell.text = "列名"
    104.     cell = doc.cell(0, 2)
    105.     # 设置单元格文本
    106.     cell.text = "类型"
    107.     cell = doc.cell(0, 3)
    108.     # 设置单元格文本
    109.     cell.text = "长度"
    110.     cell = doc.cell(0, 4)
    111.     # 设置单元格文本
    112.     cell.text = "为空"
    113.     cell = doc.cell(0, 5)
    114.     # 设置单元格文本
    115.     cell.text = "说明"
    116.     cell = doc.cell(0, 6)
    117.     # 设置单元格文本
    118.     cell.text = "主键"
    119.     # 设置表格背景色
    120.     for rgbsIndex in range(0, 7):
    121.         shading_list['shading_elm_' + str(i)] = parse_xml(
    122.             r''.format(nsdecls('w'), bgColor="#C0C0C0"))
    123.         doc.rows[0].cells[rgbsIndex]._tc.get_or_add_tcPr().append(shading_list['shading_elm_' + str(i)])
    124.  
    125. # 添加一级标题
    126. # document.add_heading('我是一级标题')
    127. #
    128. # decument.add_heading('我是二级标题', level=2)
    129. #
    130. # decument.add_heading('我是段落标题', level=0)
    131.  
    132. if __name__ == "__main__":
    133.     # 如果需要在完成的数据内插入多个标题则这里就往后移动 例如插入三个下面就从4下标开始，方便后期完成目录的生成
    134.     levelLine = 2
    135.     # 获取所有的数据库表名
    136.     res = getAllDBName()
    137.     # word文档
    138.     document = Document()
    139.     # 这里遍历获取到的所有的表名 和 注释 信息
    140.     for i in res:
    141.         # 组装名称 例 ： tableName[注释]
    142.         tableAndComment = "{0}[{1}]".format(i['table_name'], i['table_comment'])
    143.         # 组装名称 例 ： 1.tableName[注释] 方便插入标题使用
    144.         str1 = "{0}.{1}".format(levelLine, tableAndComment)
    145.         # 这里写入一级标题
    146.         T1_1 = document.add_heading('', level=1)
    147.         # 获取run 来操作其他参数
    148.         run_T1_1 = T1_1.add_run(str1)
    149.         # 设置字体格式 一级标题不适合设置字体样式 会导致难看？ ps：中文字体设置样式需要修改其他参数
    150.         # run_T1_1.font.name = '宋体'
    151.         # 设置标题颜色
    152.         # 通过run修改 字体颜色
    153.         run_T1_1.font.color.rgb = RGBColor(0, 0, 0)
    154.         run_T1_1.font.bold = True
    155.         # 获取表的DDL描述信息
    156.         dbAll = getAllDBDDL(i['table_name'], i['table_comment'])
    157.         # 创建表描述表格 3行 2列 实线
    158.         table = document.add_table(rows=3, cols=2, style="Table Grid")
    159.         # table.style.bg.color.rgb = RGBColor(255, 0, 0)
    160.         # 开启高度不匹配
    161.         table.rows[0].hight_mismatch = True
    162.         # 开启宽度不匹配
    163.         table.cell(0, 0).width_mismatch = True
    164.         # 设置列 高 宽 注释：有时有用有时候又没用 不需要自行注释
    165.         for topIndex in range(0, 3):
    166.             # table.rows[topIndex].hight = Cm(100000.23)
    167.             table.cell(topIndex, 0).width = Cm(7.53)
    168.             table.cell(topIndex, 1).hight = Cm(1.23)
    169.         # 填充表描述 左侧数据
    170.         leftComment(table)
    171.         # 填充表描述 右侧数据
    172.         rightComment(table, tableAndComment)
    173.         # 设置字体 大小
    174.         table.style.font.size = Pt(11)
    175.         # 中间插入空字符串防止两个表组合在一起
    176.         document.add_paragraph("")
    177.         # 创建表 根据获取的数据长度+1行 7列 实线
    178.         table1 = document.add_table(rows=len(dbAll) + 1, cols=7, style="Table Grid")
    179.         # 开启高度不匹配
    180.         table1.rows[0].hight_mismatch = True
    181.         # 开启宽度不匹配
    182.         table1.cell(0, 0).width_mismatch = True
    183.         # 设置列 高 宽 注释：有时有用有时候又没用
    184.         for indexs in range(0, len(dbAll) + 1):
    185.             # table1.rows[indexs].hight = Cm(10.88)
    186.             for x in range(0, 7):
    187.                 table1.cell(indexs, x).width = Cm(2.11)
    188.                 table1.cell(indexs, x).hight = Cm(1.33)
    189.         # 表头填充
    190.         topComment(table1)
    191.         # 游标来记录操作行
    192.         dbLine = 1
    193.         for db1 in dbAll:
    194.             cell = table1.cell(dbLine, 0)
    195.             # 设置单元格文本
    196.             cell.text = str(dbLine)
    197.             cell = table1.cell(dbLine, 1)
    198.             # 设置单元格文本
    199.             cell.text = db1["列名"]
    200.  
    201.             cell = table1.cell(dbLine, 2)
    202.             # 设置单元格文本
    203.             text1 = db1["数据类型"]
    204.             cell.text = db1["数据类型"]
    205.             splitStr = text1.split('(')
    206.             # 拆分数据长度注释
    207.             lenStr = ""
    208.             if len(splitStr) > 1:
    209.                 lenStr = splitStr[1].split(")")
    210.             cell = table1.cell(dbLine, 3)
    211.             # 设置单元格文本
    212.             cell.text = lenStr
    213.  
    214.             cell = table1.cell(dbLine, 4)
    215.             # 设置单元格文本
    216.             cell.text = db1["是否为空"]
    217.  
    218.             cell = table1.cell(dbLine, 5)
    219.             # 设置单元格文本
    220.             cell.text = db1["备注"]
    221.  
    222.             cell = table1.cell(dbLine, 6)
    223.             # 设置单元格文本
    224.             # 是否是主键判断 这里只适合每个表的第一列都是主键的操作，可以自行修改
    225.             isKey = "Y" if dbLine == 1 else "N"
    226.             cell.text = isKey
    227.             table1.style.font.name = '宋体'
    228.             table1.style.font.size = Pt(11)
    229.             dbLine += 1
    230.         levelLine += 1
    231.     # 目录地址根据需要自行修改，这里是导入到脚本统计目录下
    232.     document.save(r"filename.docx")

  • 相关阅读:
    Unity DropDown 组件 详解
    Py之removebg：removebg的简介、安装、使用方法之详细攻略
    IPV6地址详解
    计算机的错误计算（一百四十）
    入门力扣自学笔记121 C++ （题目编号1282）
    SpringBoot2.0数据访问之整合数据源（Druid）
    Elucidating the Design Space of Diffusion-Based Generative Models 阅读笔记
    微信小程序 springboot旅游景点门票预订服务系统
    Java学习----数据库
    河南双创蓝皮书发布：科技创新持续发力，​中创助推中部地区发展！
  • 原文地址：https://blog.csdn.net/CSDN_MrWang/article/details/127747297