import sqlite3


def InsertGroupToDB(group_name_list, db_path='../src/db_ntfs_info.db', table_name='db_group'):
    """
    向用户组表中插入多个组名。

    参数:
        group_name_list: list of str，要插入的组名列表
        db_path: str，SQLite 数据库路径
        table_name: str，用户组表名

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

    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,
            GroupName TEXT UNIQUE NOT NULL
        );
        """
        cursor.execute(create_table_sql)

        # 构建插入数据格式
        insert_sql = f"""
        INSERT OR IGNORE INTO {table_name} (GroupName)
        VALUES (?)
        """

        data_to_insert = [(name,) for name in group_name_list]

        # 批量插入
        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__":
    groups = ["Copier", "Admin", "Guest", "Developer"]
    count = InsertGroupToDB(groups)
    print(f"共插入 {count} 个新组名。")