import sqlite3

# 1. 建立連接（關(guān)鍵）
# - 連接本地文件：test.db（不存在則自動創(chuàng)建）
# - 內(nèi)存數(shù)據(jù)庫：sqlite3.connect(":memory:")（程序退出后數(shù)據(jù)丟失）
conn = sqlite3.connect("test.db")

# 2. 創(chuàng)建游標(biāo)（執(zhí)行SQL的工具）
cursor = conn.cursor()

# 創(chuàng)建users表（IF NOT EXISTS 避免重復(fù)創(chuàng)建）
create_table_sql = """
CREATE TABLE IF NOT EXISTS users (
    id INTEGER PRIMARY KEY AUTOINCREMENT,  -- 自增主鍵
    name TEXT NOT NULL,                    -- 姓名（非空）
    age INTEGER,                           -- 年齡
    email TEXT UNIQUE                      -- 郵箱（唯一，避免重復(fù)）
);
"""
cursor.execute(create_table_sql)
conn.commit()  # 建表屬于修改操作，需提交事務(wù)

# 方式1：單條插入
insert_single_sql = "INSERT INTO users (name, age, email) VALUES (?, ?, ?);"
# 參數(shù)以元組形式傳遞，匹配占位符數(shù)量
cursor.execute(insert_single_sql, ("Alice", 25, "alice@example.com"))

# 方式2：批量插入（效率更高，減少IO）
insert_batch_sql = "INSERT INTO users (name, age, email) VALUES (?, ?, ?);"
data_list = [
    ("Bob", 30, "bob@example.com"),
    ("Charlie", 28, "charlie@example.com"),
    ("David", 22, "david@example.com")
]
cursor.executemany(insert_batch_sql, data_list)

# 增刪改操作必須提交事務(wù)，否則數(shù)據(jù)不生效
conn.commit()
print("插入數(shù)據(jù)成功，最后插入的ID：", cursor.lastrowid)  # 獲取最后插入的主鍵ID

# 方式1：查詢單條數(shù)據(jù)（fetchone()）
select_single_sql = "SELECT * FROM users WHERE name = ?;"
cursor.execute(select_single_sql, ("Alice",))
single_result = cursor.fetchone()  # 返回元組：(1, 'Alice', 25, 'alice@example.com')
print("單條查詢結(jié)果：", single_result)

# 方式2：查詢多條數(shù)據(jù)（fetchall()）
select_all_sql = "SELECT id, name, age FROM users WHERE age > ?;"
cursor.execute(select_all_sql, (25,))
all_results = cursor.fetchall()  # 返回列表嵌套元組：[(2, 'Bob', 30), (3, 'Charlie', 28)]
print("\n多條查詢結(jié)果：")
for row in all_results:
    print(f"ID: {row[0]}, 姓名: {row[1]}, 年齡: {row[2]}")

# 方式3：指定條數(shù)查詢（fetchmany(n)）
cursor.execute("SELECT * FROM users;")
many_results = cursor.fetchmany(2)  # 僅獲取前2條
print("\n指定條數(shù)查詢結(jié)果：", many_results)

# 進(jìn)階：返回字典格式結(jié)果（更易讀）
# 創(chuàng)建游標(biāo)時指定 row_factory
conn.row_factory = sqlite3.Row
cursor = conn.cursor()
cursor.execute("SELECT * FROM users WHERE id = ?;", (1,))
dict_result = cursor.fetchone()
print("\n字典格式結(jié)果：", dict_result["name"], dict_result["email"])  # 可通過鍵取值

update_sql = "UPDATE users SET age = ? WHERE name = ?;"
cursor.execute(update_sql, (26, "Alice"))  # 將Alice的年齡改為26
conn.commit()

# 查看受影響的行數(shù)
print("更新影響的行數(shù)：", cursor.rowcount)  # 輸出：1

delete_sql = "DELETE FROM users WHERE id = ?;"
cursor.execute(delete_sql, (4,))  # 刪除ID為4的記錄
conn.commit()

print("刪除影響的行數(shù)：", cursor.rowcount)  # 輸出：1

# 先關(guān)游標(biāo)，再關(guān)連接
cursor.close()
conn.close()

import sqlite3

# 上下文管理器簡化操作
with sqlite3.connect("test.db") as conn:
    cursor = conn.cursor()
    # 執(zhí)行查詢
    cursor.execute("SELECT * FROM users;")
    print("上下文管理器查詢結(jié)果：", cursor.fetchall())
    # 增刪改無需手動commit，with塊結(jié)束會自動提交（出錯則回滾）
    cursor.execute("INSERT INTO users (name, age, email) VALUES (?, ?, ?);", ("Eve", 29, "eve@example.com"))

# 錯誤示例（易被注入，如name傳入 "Alice'; DROP TABLE users; --"）
name = "Alice"
sql = f"SELECT * FROM users WHERE name = '{name}';"  # 危險！

# 正確示例（占位符）
sql = "SELECT * FROM users WHERE name = ?;"
cursor.execute(sql, (name,))  # 安全

import sqlite3

try:
    conn = sqlite3.connect("test.db")
    cursor = conn.cursor()
    # 執(zhí)行可能出錯的操作（如插入重復(fù)郵箱）
    cursor.execute("INSERT INTO users (name, age, email) VALUES (?, ?, ?);", ("Frank", 35, "alice@example.com"))
    conn.commit()
except sqlite3.IntegrityError as e:
    # 捕獲唯一鍵沖突異常
    print("插入失?。何ㄒ绘I沖突", e)
    conn.rollback()  # 出錯回滾事務(wù)
except sqlite3.Error as e:
    print("數(shù)據(jù)庫錯誤：", e)
    conn.rollback()
finally:
    if cursor:
        cursor.close()
    if conn:
        conn.close()