import sqlite3
import os

DB_FILE = "my_database.db"

def connect_db():
    """建立數(shù)據(jù)庫連接"""
    try:
        conn = sqlite3.connect(DB_FILE)
        return conn
    except sqlite3.Error as e:
        print(f"連接數(shù)據(jù)庫失敗: {e}")
        return None

def close_db(conn):
    """關(guān)閉數(shù)據(jù)庫連接"""
    if conn:
        conn.close()

# 示例：創(chuàng)建測試數(shù)據(jù)庫和表
def setup_test_db():
    if os.path.exists(DB_FILE):
        os.remove(DB_FILE) # 清理舊的測試文件
    conn = connect_db()
    if conn:
        cursor = conn.cursor()
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS users (
                id INTEGER PRIMARY KEY,
                name TEXT NOT NULL,
                email TEXT UNIQUE
            );
        ''')
        conn.commit()
        print(f"數(shù)據(jù)庫 '{DB_FILE}' 已創(chuàng)建，并創(chuàng)建了 'users' 表。")
        close_db(conn)

def cleanup_test_db():
    """刪除測試數(shù)據(jù)庫文件"""
    if os.path.exists(DB_FILE):
        os.remove(DB_FILE)
        print(f"數(shù)據(jù)庫 '{DB_FILE}' 已刪除。")

# setup_test_db() # 調(diào)用以創(chuàng)建測試數(shù)據(jù)庫
# cleanup_test_db() # 調(diào)用以刪除測試數(shù)據(jù)庫

SELECT name FROM sqlite_master WHERE type='table' AND name='{table_name}';

def check_table_exists_sqlite_master(db_file, table_name):
    """
    方法一：通過查詢 sqlite_master 表來檢查表是否存在。
    這是最推薦和最安全的方法，因?yàn)樗褂脜?shù)化查詢來處理表名。
    """
    conn = None
    try:
        conn = sqlite3.connect(db_file)
        cursor = conn.cursor()
        # 使用參數(shù)化查詢，防止 SQL 注入
        query = "SELECT name FROM sqlite_master WHERE type='table' AND name=?;"
        cursor.execute(query, (table_name,)) # 注意 table_name 是元組
        result = cursor.fetchone()
        return result is not None
    except sqlite3.Error as e:
        print(f"檢查表 '{table_name}' 存在性時發(fā)生 SQLite 錯誤: {e}")
        return False
    finally:
        if conn:
            conn.close()

def check_table_exists_pragma_table_info(db_file, table_name):
    """
    方法二：通過 PRAGMA table_info() 檢查表是否存在。
    此方法對表名使用字符串格式化，因此需要謹(jǐn)慎處理用戶輸入。
    """
    conn = None
    try:
        conn = sqlite3.connect(db_file)
        cursor = conn.cursor()
        # 注意：PRAGMA 語句通常不支持對表名進(jìn)行參數(shù)化查詢，
        # 因此這里使用 f-string 格式化。如果 table_name 來自用戶輸入，
        # 則存在 SQL 注入風(fēng)險，需要額外的驗(yàn)證或白名單機(jī)制。
        cursor.execute(f"PRAGMA table_info('{table_name}');")
        result = cursor.fetchall()
        return len(result) > 0 # 如果有列信息返回，則表存在
    except sqlite3.Error as e:
        print(f"檢查表 '{table_name}' 存在性時發(fā)生 SQLite 錯誤: {e}")
        return False
    finally:
        if conn:
            conn.close()

def check_table_exists_try_select(db_file, table_name):
    """
    方法三：嘗試從表中查詢數(shù)據(jù)并捕獲 OperationalError 異常。
    此方法使用異常處理進(jìn)行流程控制，且對表名使用字符串格式化。
    """
    conn = None
    try:
        conn = sqlite3.connect(db_file)
        cursor = conn.cursor()
        # 同樣，表名作為 SQL 結(jié)構(gòu)的一部分，這里使用 f-string 格式化。
        # 存在 SQL 注入風(fēng)險，需要額外的驗(yàn)證。
        cursor.execute(f"SELECT 1 FROM {table_name} LIMIT 1;")
        return True # 如果執(zhí)行成功，則表存在
    except sqlite3.OperationalError:
        return False # 捕獲到 OperationalError，說明表不存在
    except sqlite3.Error as e:
        print(f"檢查表 '{table_name}' 存在性時發(fā)生意外的 SQLite 錯誤: {e}")
        return False
    finally:
        if conn:
            conn.close()

def check_table_exists_with_context(db_file, table_name):
    """使用 with 語句管理連接，并使用推薦的 sqlite_master 方法。"""
    try:
        with sqlite3.connect(db_file) as conn: # 連接將自動關(guān)閉
            cursor = conn.cursor()
            query = "SELECT name FROM sqlite_master WHERE type='table' AND name=?;"
            cursor.execute(query, (table_name,))
            result = cursor.fetchone()
            return result is not None
    except sqlite3.Error as e:
        print(f"檢查表 '{table_name}' 存在性時發(fā)生 SQLite 錯誤: {e}")
        return False

import sqlite3
import os

DB_FILE = "my_database.db"

def setup_database_for_testing():
    """
    創(chuàng)建測試數(shù)據(jù)庫，包含一個 'users' 表和一個大小寫敏感的 '"Products"' 表，
    以及一個不存在的 'orders' 表。
    """
    if os.path.exists(DB_FILE):
        os.remove(DB_FILE)
    try:
        with sqlite3.connect(DB_FILE) as conn:
            cursor = conn.cursor()
            # 創(chuàng)建一個普通表 (大小寫不敏感)
            cursor.execute('''
                CREATE TABLE users (
                    id INTEGER PRIMARY KEY,
                    name TEXT NOT NULL
                );
            ''')
            # 創(chuàng)建一個大小寫敏感的表 (注意雙引號)
            cursor.execute('''
                CREATE TABLE "Products" (
                    item_id INTEGER PRIMARY KEY,
                    item_name TEXT NOT NULL
                );
            ''')
            conn.commit()
            print(f"測試數(shù)據(jù)庫 '{DB_FILE}' 已創(chuàng)建，包含 'users' 和 'Products' 表。")
    except sqlite3.Error as e:
        print(f"Error setting up test database: {e}")

def cleanup_database_after_testing():
    """刪除測試數(shù)據(jù)庫文件。"""
    if os.path.exists(DB_FILE):
        os.remove(DB_FILE)
        print(f"測試數(shù)據(jù)庫 '{DB_FILE}' 已刪除。")

def check_table_exists_master_method(table_name):
    """方法一：查詢 sqlite_master 表（推薦）。"""
    try:
        with sqlite3.connect(DB_FILE) as conn:
            cursor = conn.cursor()
            query = "SELECT name FROM sqlite_master WHERE type='table' AND name=?;"
            cursor.execute(query, (table_name,))
            return cursor.fetchone() is not None
    except sqlite3.Error as e:
        print(f"SQLite error for '{table_name}' with master method: {e}")
        return False

def check_table_exists_pragma_method(table_name):
    """方法二：使用 PRAGMA table_info()（注意 SQL 注入風(fēng)險）。"""
    try:
        with sqlite3.connect(DB_FILE) as conn:
            cursor = conn.cursor()
            # !! SQL 注入風(fēng)險：表名未經(jīng)參數(shù)化，來自用戶輸入時需嚴(yán)格驗(yàn)證 !!
            cursor.execute(f"PRAGMA table_info('{table_name}');")
            return len(cursor.fetchall()) > 0
    except sqlite3.Error as e:
        print(f"SQLite error for '{table_name}' with pragma method: {e}")
        return False

def check_table_exists_try_select_method(table_name):
    """方法三：嘗試查詢并捕獲異常（注意 SQL 注入風(fēng)險和流程控制）。"""
    try:
        with sqlite3.connect(DB_FILE) as conn:
            cursor = conn.cursor()
            # !! SQL 注入風(fēng)險：表名未經(jīng)參數(shù)化，來自用戶輸入時需嚴(yán)格驗(yàn)證 !!
            cursor.execute(f"SELECT 1 FROM {table_name} LIMIT 1;")
            return True
    except sqlite3.OperationalError:
        return False
    except sqlite3.Error as e:
        print(f"SQLite error for '{table_name}' with try-select method: {e}")
        return False

def main():
    setup_database_for_testing()

    print("\n--- 檢查 'users' 表 (存在，大小寫不敏感) ---")
    print(f"sqlite_master: {check_table_exists_master_method('users')}")
    print(f"PRAGMA: {check_table_exists_pragma_method('users')}")
    print(f"try-select: {check_table_exists_try_select_method('users')}")
    print(f"sqlite_master (大寫): {check_table_exists_master_method('USERS')}") # SQLite 默認(rèn)不區(qū)分大小寫
    print(f"PRAGMA (大寫): {check_table_exists_pragma_method('USERS')}")
    print(f"try-select (大寫): {check_table_exists_try_select_method('USERS')}")


    print("\n--- 檢查 'Products' 表 (存在，大小寫敏感) ---")
    print(f"sqlite_master: {check_table_exists_master_method('Products')}")
    print(f"PRAGMA: {check_table_exists_pragma_method('Products')}")
    print(f"try-select: {check_table_exists_try_select_method('Products')}")
    print(f"sqlite_master (小寫): {check_table_exists_master_method('products')}") # 應(yīng)該為 False
    print(f"PRAGMA (小寫): {check_table_exists_pragma_method('products')}")
    print(f"try-select (小寫): {check_table_exists_try_select_method('products')}")


    print("\n--- 檢查 'orders' 表 (不存在) ---")
    print(f"sqlite_master: {check_table_exists_master_method('orders')}")
    print(f"PRAGMA: {check_table_exists_pragma_method('orders')}")
    print(f"try-select: {check_table_exists_try_select_method('orders')}")

    cleanup_database_after_testing()

if __name__ == "__main__":
    main()