Python實現數據庫數據自動化導出PDF報告

更新時間：2025年09月25日 10:17:07 作者：IT小本本

本文將介紹如何使用Python構建一個自動化工具,實現從MySQL數據庫提取員工數據,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下

技術棧與核心功能

主要技術組件

數據庫連接：使用pymysql實現Python與MySQL的交互，支持字符集配置和字典游標（便于數據處理）。

PDF生成：基于reportlab庫構建PDF文檔，支持表格樣式定制、中文字體渲染和動態(tài)內容生成。

異常處理：通過try-except結構和traceback模塊捕獲運行時錯誤，提供詳細日志便于調試。

文件管理：自動創(chuàng)建輸出目錄，使用時間戳確保文件名唯一，避免文件覆蓋。

功能亮點

自動檢測數據庫表結構，支持動態(tài)字段提取

為每位員工生成獨立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直接返回字典格式數據，便于后續(xù)處理；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

預處理邏輯：先校驗表是否存在，避免后續(xù)操作拋出未知錯誤。

數據格式：返回字段列表和字典數組，便于與PDF生成模塊對接。

3. 個性化PDF生成

def create_pdf_for_employee(employee_data, columns):
    try:
        # 目錄創(chuàng)建
        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放在程序目錄或指定路徑），解決中文亂碼問題。

樣式定制技巧：