import sqlite3

class InitDB():
    def __init__(self, parent=None, title="初始化数据库", initial_data=None):
        super().__init__(parent)

    @staticmethod
    def init_db():
        # 连接到SQLite数据库（如果数据库不存在，则会自动创建）
        conn = sqlite3.connect('buff.db')
        cursor = conn.cursor()

        # 创建表
        cursor.execute('''CREATE TABLE IF NOT EXISTS sh_account_info
                     (phone TEXT, email TEXT, code TEXT, cookie TEXT,xcsrftoken TEXT)''')

        # 创建表（如果表不存在）
        cursor.execute('''CREATE TABLE IF NOT EXISTS shpz_settings
                     (pause_time REAL,thread_count INTEGER,needed_msg_check_pause_time INTEGER,
                needed_msg_check_try_times INTEGER)''')

        # 创建表
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS accounts (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                phone_number TEXT NOT NULL,
                email TEXT NOT NULL,
                email_alias TEXT NOT NULL,
                cookie TEXT NOT NULL
            )
        ''')

        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS categories (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                name TEXT NOT NULL,
                code TEXT NOT NULL,
                selected BOOLEAN NOT NULL DEFAULT 0  -- 新增字段，用于存储选择状态
            )
        ''')
        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS qualities (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                code TEXT NOT NULL,
                name TEXT NOT NULL,
                selected BOOLEAN NOT NULL DEFAULT 0  -- 新增字段，用于存储选择状态
            )
        ''')
        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS types (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                code TEXT NOT NULL,
                name TEXT NOT NULL,
                selected BOOLEAN NOT NULL DEFAULT 0  -- 新增字段，用于存储选择状态
            )
        ''')
        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS appearances (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                code TEXT NOT NULL,
                name TEXT NOT NULL,
                selected BOOLEAN NOT NULL DEFAULT 0  -- 新增字段，用于存储选择状态
            )
        ''')
        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS collectibles (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                code TEXT NOT NULL,
                name TEXT NOT NULL,
                selected BOOLEAN NOT NULL DEFAULT 0  -- 新增字段，用于存储选择状态
            )
        ''')
        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS config (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                min_price REAL,
                max_price REAL,
                sort_mode TEXT,
                stop_condition TEXT,
                success_count INTEGER,
                scan_count INTEGER,
                page_size INTEGER,
                pause_time INTEGER,
                max_pages INTEGER,
                ref_price_adjust INTEGER,
                sweep_price_adjust INTEGER,
                needed_msg_check_pause_time INTEGER,
                needed_msg_check_try_times INTEGER,
                delete_ids TEXT,
                cron_time TEXT,
                sell_num INTEGER
            )
        ''')
        # 创建表（如果表不存在）
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS select_logs (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                item_id TEXT NOT NULL,
                item_name TEXT NOT NULL,
                min_wear TEXT NOT NULL,
                max_wear TEXT NOT NULL,
                min_full_wear_price REAL NOT NULL,
                sweep_price REAL NOT NULL,
                judge_price REAL NOT NULL,
                timestamp TEXT NOT NULL
            )
        ''')
        conn.commit()
        conn.close()