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

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

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

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

pip install pymysql reportlab