67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
import sqlite3
|
||
|
||
|
||
def InsertUserToDB(username_list, db_path='../src/db_ntfs_info.db', table_name='db_user'):
|
||
"""
|
||
向用户表中插入多个用户名。
|
||
|
||
参数:
|
||
username_list: list of str，要插入的用户名列表
|
||
db_path: str，SQLite 数据库路径
|
||
table_name: str，用户表名
|
||
|
||
返回:
|
||
int: 成功插入的新用户数量
|
||
"""
|
||
if not isinstance(username_list, list):
|
||
raise TypeError("username_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,
|
||
UserName TEXT UNIQUE NOT NULL
|
||
);
|
||
"""
|
||
cursor.execute(create_table_sql)
|
||
|
||
# 插入语句（忽略重复）
|
||
insert_sql = f"""
|
||
INSERT OR IGNORE INTO {table_name} (UserName)
|
||
VALUES (?)
|
||
"""
|
||
|
||
# 构建数据格式
|
||
data_to_insert = [(name,) for name in username_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__":
|
||
users = ["Alice", "Bob", "Charlie", "David"]
|
||
count = InsertUserToDB(users)
|
||
print(f"共插入 {count} 个新用户。")
|