import sqlite3


def InsertExtensionsToDB(extensions, db_path='../src/db_ntfs_info.db', table_name='db_extend_name'):
    """
    将扩展名列表插入到数据库中，自动忽略重复项。

    参数:
        extensions: list of str，要插入的扩展名列表（如 ["txt", "jpg"]）
        db_path: str，SQLite 数据库路径
        table_name: str，扩展名表名

    返回:
        int: 成功插入的新记录数量
    """
    if not isinstance(extensions, list):
        raise TypeError("extensions 必须是一个列表")

    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()

    try:
        # 创建表（如果不存在）
        create_table_sql = f"""
        CREATE TABLE IF NOT EXISTS {table_name} (
            ID INTEGER PRIMARY KEY AUTOINCREMENT,
            ExtendName TEXT UNIQUE NOT NULL
        );
        """
        cursor.execute(create_table_sql)

        # 插入语句（忽略重复）
        insert_sql = f"""
        INSERT OR IGNORE INTO {table_name} (ExtendName)
        VALUES (?)
        """

        # 构造插入数据格式
        data_to_insert = [(ext.lower(),) for ext in extensions if ext.strip()]

        # 批量插入
        cursor.executemany(insert_sql, data_to_insert)
        conn.commit()

        inserted_count = cursor.rowcount
        if inserted_count > 0:
            print(f"✅ 成功插入 {inserted_count} 个扩展名")
        else:
            print("⚠️ 没有新的扩展名被插入（可能已存在）")

        return inserted_count

    except Exception as e:
        print(f"❌ 插入失败: {e}")
        conn.rollback()
        return 0

    finally:
        conn.close()


# 示例调用
if __name__ == "__main__":
    # 常见的文件扩展名列表（可选）
    common_extensions = ["txt", "log", "csv", "xls", "xlsx", "doc", "docx"]

    count = InsertExtensionsToDB(common_extensions)
    print(f"共插入 {count} 个新扩展名。")