python读取SQLite表个并生成pdf文件
代码用于创建含50列的SQLite数据库并插入500行随机浮点数据,随后读取数据,通过ReportLab生成横向PDF表格,包含格式化(两位小数)及表头、网格线等美观样式。
# 导入所需库
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")