python读取SQLite表个并生成pdf文件

代码用于创建含50列的SQLite数据库并插入500行随机浮点数据，随后读取数据，通过ReportLab生成横向PDF表格，包含格式化（两位小数）及表头、网格线等美观样式。
python 复制代码
# 导入所需库
import sqlite3                  # 用于操作SQLite数据库
import random                   # 用于生成随机数据
from reportlab.lib.pagesizes import landscape, letter  # 报表实验室库：页面尺寸（横向letter纸）
from reportlab.platypus import SimpleDocTemplate, LongTable  # 报表实验室组件：文档模板、长表格
from reportlab.lib import colors  # 报表实验室颜色库
from reportlab.lib.styles import getSampleStyleSheet  # 报表实验室预定义样式

# ==================== 步骤1：创建数据库并生成测试数据 ====================
# 连接到SQLite数据库（如果不存在则自动创建）
conn = sqlite3.connect('mydatabase.db')
# 创建数据库游标对象，用于执行SQL语句
c = conn.cursor()

# 动态生成50个列名（col1到col50）
columns = [f'col{i+1}' for i in range(50)]
# 构建创建表的SQL语句：50个REAL类型（浮点数）的列
create_table_sql = f'CREATE TABLE IF NOT EXISTS tab1 ({", ".join([f"{col} REAL" for col in columns])})'
# 执行SQL语句创建表（IF NOT EXISTS避免重复创建报错）
c.execute(create_table_sql)

# 生成500行随机数据（每行50个0-100之间的浮点数）
data = []
for _ in range(500):  # 循环500次生成500行
    # 每行包含50个随机浮点数（范围0-100）
    row = [random.uniform(0, 100) for _ in range(50)]
    data.append(row)

# 使用executemany批量插入数据（比循环execute更高效）
# SQL语句中的?是占位符，对应data中的每个元素
c.executemany(f'INSERT INTO tab1 VALUES ({", ".join(["?"]*50)})', data)
# 提交事务（将缓存中的数据写入数据库）
conn.commit()
# 关闭数据库连接
conn.close()

# ==================== 步骤2：从数据库读取数据 ====================
# 重新连接数据库（因为之前已关闭）
conn = sqlite3.connect('mydatabase.db')
c = conn.cursor()
# 执行查询语句获取所有数据
c.execute('SELECT * FROM tab1')
# 获取所有查询结果（返回元组的列表，每个元组代表一行数据）
rows = c.fetchall()
# 获取表的列名（通过cursor.description属性，每个元素的第一个值是列名）
columns = [desc[0] for desc in c.description]
# 关闭数据库连接
conn.close()

# ==================== 步骤3：生成PDF报表 ====================
# 创建PDF文档模板（横向letter纸尺寸，文件名为tab1_data.pdf）
doc = SimpleDocTemplate("tab1_data.pdf", pagesize=landscape(letter))
# 获取预定义的文本样式（用于表格内容）
styles = getSampleStyleSheet()

# 格式化数据为字符串（保留2位小数，提高可读性）
formatted_rows = []
for row in rows:
    # 将每个单元格的浮点数格式化为"%.2f"形式的字符串
    formatted_row = [f"{cell:.2f}" for cell in row]
    formatted_rows.append(formatted_row)

# 在表格顶部插入表头行（使用原始列名）
formatted_rows.insert(0, columns)

# 创建长表格（支持跨页显示）
# colWidths参数设置每列宽度（这里每列固定80点，总宽度50 * 80=4000点，适合横向纸）
table = LongTable(formatted_rows, colWidths=[80] * 50)

# 设置表格样式（通过元组列表定义不同区域的样式）
table.setStyle([
    # 表头区域样式（第0行，所有列）
    ('BACKGROUND', (0,0), (-1,0), colors.grey),          # 背景色（灰色）
    ('TEXTCOLOR', (0,0), (-1,0), colors.whitesmoke),     # 文字颜色（白色）
    ('ALIGN', (0,0), (-1,-1), 'CENTER'),                 # 所有单元格文字居中
    ('FONTNAME', (0,0), (-1,0), 'Helvetica-Bold'),       # 表头字体（加粗）
    ('BOTTOMPADDING', (0,0), (-1,0), 12),                # 表头底部内边距（12点）
    
    # 数据行区域样式（第1行到最后一行，所有列）
    ('BACKGROUND', (0,1), (-1,-1), colors.beige),        # 背景色（米色）
    
    # 全局网格线样式（所有单元格）
    ('GRID', (0,0), (-1,-1), 1, colors.black),           # 网格线宽度1点，黑色
    
    # 全局垂直对齐方式（所有单元格顶部对齐）
    ('VALIGN', (0,0), (-1,-1), 'TOP')
])

# 将表格添加到PDF文档并构建输出
doc.build([table])

# 输出完成提示
print("PDF已成功生成: tab1_data.pdf")