Python实现数据库数据自动化导出PDF报告_Python

本文将介绍如何使用python构建一个自动化工具，实现从mysql数据库提取员工数据，并为每位员工生成包含定制化表格的pdf报告。通过该方案，可显著提升数据导出效率，避免手动操作误差，同时支持灵活的格式定制。

需求：从mysql数据库提取员工数据，并为每位员工生成包含定制化表格的pdf报告。

最终想要的结果：

技术栈与核心功能

主要技术组件

数据库连接：使用pymysql实现python与mysql的交互，支持字符集配置和字典游标（便于数据处理）。

pdf生成：基于reportlab库构建pdf文档，支持表格样式定制、中文字体渲染和动态内容生成。

异常处理：通过try-except结构和traceback模块捕获运行时错误，提供详细日志便于调试。

文件管理：自动创建输出目录，使用时间戳确保文件名唯一，避免文件覆盖。

功能亮点

自动检测数据库表结构，支持动态字段提取

为每位员工生成独立pdf，包含姓名、薪资等关键信息

定制化表格样式（背景色、字体、边框等）

完整的中文字体支持（通过注册系统字体实现）

核心代码解析

1. 数据库连接与数据获取

import pymysql

def connect_to_database():
    try:
        connection = pymysql.connect(
            host='127.0.0.1',
            user='root',
            password='root',
            database='250515sj',
            charset='utf8mb4',
            cursorclass=pymysql.cursors.dictcursor
        )
        print("数据库连接成功！")
        return connection
    except exception as e:
        # 详细异常日志记录
        print(f"数据库连接错误: {str(e)}")
        print("详细错误信息:\n", traceback.format_exc())
        return none

关键配置：使用dictcursor直接返回字典格式数据，便于后续处理；charset='utf8mb4'确保支持 emoji 等特殊字符。

健壮性设计：通过异常捕获避免程序因数据库连接失败崩溃，并提供堆栈跟踪信息。

数据表内容：

2. 数据提取与表结构校验

def get_employees_data(connection):
    try:
        with connection.cursor() as cursor:
            # 表存在性校验
            cursor.execute("show tables like 'employees'")
            if not cursor.fetchone():
                print("错误：employees表不存在！")
                return none, none
                
            # 提取表结构与数据
            cursor.execute("show columns from employees")
            columns = [col['field'] for col in cursor.fetchall()]
            cursor.execute("select * from employees")
            data = cursor.fetchall()
            
            print(f"表结构: {columns}, 获取到 {len(data)} 条记录")
            return columns, data
    except exception as e:
        # 数据操作异常处理
        print(f"获取数据错误: {str(e)}", traceback.format_exc())
        return none, none

预处理逻辑：先校验表是否存在，避免后续操作抛出未知错误。

数据格式：返回字段列表和字典数组，便于与pdf生成模块对接。

3. 个性化pdf生成

def create_pdf_for_employee(employee_data, columns):
    try:
        # 目录创建
        output_dir = "employee_pdfs"
        os.makedirs(output_dir, exist_ok=true)
        
        # 字段筛选与中文表头映射
        export_data = [employee_data[col] for col in ['name', 'salary']]
        filename = f"{employee_data['name']}_{datetime.now().strftime('%y%m%d_%h%m%s')}.pdf"
        path = os.path.join(output_dir, filename)
        
        # pdf文档初始化
        doc = simpledoctemplate(path, pagesize=letter)
        styles = getsamplestylesheet()
        
        # 中文字体注册（关键步骤）
        pdfmetrics.registerfont(ttfont('simhei', 'simhei.ttf'))
        title_style = paragraphstyle(
            'customtitle',
            parent=styles['heading1'],
            fontname='simhei',
            fontsize=24,
            spaceafter=30
        )
        
        # 表格数据与样式
        table_data = [['姓名', '薪资'], export_data]
        table = table(table_data)
        table.setstyle(tablestyle([
            ('background', (0,0), (-1,0), colors.grey),    # 表头背景色
            ('textcolor', (0,0), (-1,0), colors.whitesmoke), # 表头文字颜色
            ('align', (0,0), (-1,-1), 'center'),            # 全局居中对齐
            ('fontname', (0,0), (-1,-1), 'simhei'),          # 中文字体应用
            ('grid', (0,0), (-1,-1), 1, colors.black)        # 边框设置
        ]))
        
        # 文档元素组装
        elements = [
            paragraph(f"{employee_data['name']}的个人信息", title_style),
            spacer(1, 20),
            table
        ]
        doc.build(elements)
        return path
    except exception as e:
        print(f"生成pdf错误: {str(e)}", traceback.format_exc())
        return none

中文字体解决方案：通过ttfont注册本地字体文件（需将simhei.ttf放在程序目录或指定路径），解决中文乱码问题。

样式定制技巧：