# coding:utf-8
# @Time:2022-06-20 12:28
# @Author:Kevin
# @Software:PyCharm
import sqlite3 as sl
# 獲取列名/字段名
db = sl.connect('mytest.db')
cur = db.cursor()
cur.execute("select * from table_data")
col_name_list = [tuple[0] for tuple in cur.description]
print(col_name_list)

# cur.execute("PRAGMA table_info(table_data)")
# print(cur.fetchall())

# 獲取sqlite3數(shù)據(jù)庫(kù)mydb.db中的表名和表字段名
 
import sqlite3
 
 
def sqlite_void():
    conn = sqlite3.connect('d:/data/mydb.db')
    cur = conn.cursor()
    # 獲取表名，保存在table_name列表
    cur.execute("select name from sqlite_master where type='table'")
    rows = cur.fetchall()
    table_name = [row[0] for row in rows]
 
    # 獲取表的列名（字段名），保存在col_names列表,每個(gè)表的字段名集為一個(gè)元組
    col_names = []
    for i in table_name:
        cur.execute('pragma table_info({})'.format(i))
        col_name = cu.fetchall()
        col_name = [x[1] for x in col_name]
        col_name = tuple(col_name)
        col_names.append(col_name)
 
if __name__ == '__main__':
    sqlite_void()

# 導(dǎo)入所需庫(kù)
import sqlite3                  # 用于操作SQLite數(shù)據(jù)庫(kù)
import random                   # 用于生成隨機(jī)數(shù)據(jù)
from reportlab.lib.pagesizes import landscape, letter  # 報(bào)表實(shí)驗(yàn)室?guī)欤喉?yè)面尺寸（橫向letter紙）
from reportlab.platypus import SimpleDocTemplate, LongTable  # 報(bào)表實(shí)驗(yàn)室組件：文檔模板、長(zhǎng)表格
from reportlab.lib import colors  # 報(bào)表實(shí)驗(yàn)室顏色庫(kù)
from reportlab.lib.styles import getSampleStyleSheet  # 報(bào)表實(shí)驗(yàn)室預(yù)定義樣式
 
# ==================== 步驟1：創(chuàng)建數(shù)據(jù)庫(kù)并生成測(cè)試數(shù)據(jù) ====================
# 連接到SQLite數(shù)據(jù)庫(kù)（如果不存在則自動(dòng)創(chuàng)建）
conn = sqlite3.connect('mydatabase.db')
# 創(chuàng)建數(shù)據(jù)庫(kù)游標(biāo)對(duì)象，用于執(zhí)行SQL語(yǔ)句
c = conn.cursor()
 
# 動(dòng)態(tài)生成50個(gè)列名（col1到col50）
columns = [f'col{i+1}' for i in range(50)]
# 構(gòu)建創(chuàng)建表的SQL語(yǔ)句：50個(gè)REAL類(lèi)型（浮點(diǎn)數(shù)）的列
create_table_sql = f'CREATE TABLE IF NOT EXISTS tab1 ({", ".join([f"{col} REAL" for col in columns])})'
# 執(zhí)行SQL語(yǔ)句創(chuàng)建表（IF NOT EXISTS避免重復(fù)創(chuàng)建報(bào)錯(cuò)）
c.execute(create_table_sql)
 
# 生成500行隨機(jī)數(shù)據(jù)（每行50個(gè)0-100之間的浮點(diǎn)數(shù)）
data = []
for _ in range(500):  # 循環(huán)500次生成500行
    # 每行包含50個(gè)隨機(jī)浮點(diǎn)數(shù)（范圍0-100）
    row = [random.uniform(0, 100) for _ in range(50)]
    data.append(row)
 
# 使用executemany批量插入數(shù)據(jù)（比循環(huán)execute更高效）
# SQL語(yǔ)句中的?是占位符，對(duì)應(yīng)data中的每個(gè)元素
c.executemany(f'INSERT INTO tab1 VALUES ({", ".join(["?"]*50)})', data)
# 提交事務(wù)（將緩存中的數(shù)據(jù)寫(xiě)入數(shù)據(jù)庫(kù)）
conn.commit()
# 關(guān)閉數(shù)據(jù)庫(kù)連接
conn.close()
 
# ==================== 步驟2：從數(shù)據(jù)庫(kù)讀取數(shù)據(jù) ====================
# 重新連接數(shù)據(jù)庫(kù)（因?yàn)橹耙殃P(guān)閉）
conn = sqlite3.connect('mydatabase.db')
c = conn.cursor()
# 執(zhí)行查詢(xún)語(yǔ)句獲取所有數(shù)據(jù)
c.execute('SELECT * FROM tab1')
# 獲取所有查詢(xún)結(jié)果（返回元組的列表，每個(gè)元組代表一行數(shù)據(jù)）
rows = c.fetchall()
# 獲取表的列名（通過(guò)cursor.description屬性，每個(gè)元素的第一個(gè)值是列名）
columns = [desc[0] for desc in c.description]
# 關(guān)閉數(shù)據(jù)庫(kù)連接
conn.close()
 
# ==================== 步驟3：生成PDF報(bào)表 ====================
# 創(chuàng)建PDF文檔模板（橫向letter紙尺寸，文件名為tab1_data.pdf）
doc = SimpleDocTemplate("tab1_data.pdf", pagesize=landscape(letter))
# 獲取預(yù)定義的文本樣式（用于表格內(nèi)容）
styles = getSampleStyleSheet()
 
# 格式化數(shù)據(jù)為字符串（保留2位小數(shù)，提高可讀性）
formatted_rows = []
for row in rows:
    # 將每個(gè)單元格的浮點(diǎn)數(shù)格式化為"%.2f"形式的字符串
    formatted_row = [f"{cell:.2f}" for cell in row]
    formatted_rows.append(formatted_row)
 
# 在表格頂部插入表頭行（使用原始列名）
formatted_rows.insert(0, columns)
 
# 創(chuàng)建長(zhǎng)表格（支持跨頁(yè)顯示）
# colWidths參數(shù)設(shè)置每列寬度（這里每列固定80點(diǎn)，總寬度50 * 80=4000點(diǎn)，適合橫向紙）
table = LongTable(formatted_rows, colWidths=[80] * 50)
 
# 設(shè)置表格樣式（通過(guò)元組列表定義不同區(qū)域的樣式）
table.setStyle([
    # 表頭區(qū)域樣式（第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),                # 表頭底部?jī)?nèi)邊距（12點(diǎn)）
    
    # 數(shù)據(jù)行區(qū)域樣式（第1行到最后一行，所有列）
    ('BACKGROUND', (0,1), (-1,-1), colors.beige),        # 背景色（米色）
    
    # 全局網(wǎng)格線(xiàn)樣式（所有單元格）
    ('GRID', (0,0), (-1,-1), 1, colors.black),           # 網(wǎng)格線(xiàn)寬度1點(diǎn)，黑色
    
    # 全局垂直對(duì)齊方式（所有單元格頂部對(duì)齊）
    ('VALIGN', (0,0), (-1,-1), 'TOP')
])
 
# 將表格添加到PDF文檔并構(gòu)建輸出
doc.build([table])
 
# 輸出完成提示
print("PDF已成功生成: tab1_data.pdf")