diff --git a/windows/command/init.py b/windows/command/init.py index 9219cd3..b3129b6 100644 --- a/windows/command/init.py +++ b/windows/command/init.py @@ -1,17 +1,44 @@ # 初始化命令脚本 import sqlite3 import os -import sys +import shutil # 初始化生成路径 # 确定当前路径 dir_path = os.path.dirname(os.path.abspath(__file__)) # 确定sqlite数据库初始位置 db_path = os.path.dirname(os.path.dirname(dir_path)) +# 删除db文件夹 +shutil.rmtree(f'{db_path}\\db') # 建立初始化db文件夹存放数据库 os.mkdir(f'{db_path}\\db') print(db_path) # 建立sqlite数据库 -conn = sqlite3.connect(f'{db_path}\\db\\admin.db') \ No newline at end of file +conn = sqlite3.connect(f'{db_path}\\db\\admin.db') +# 建立管理员表 +create_table = """CREATE TABLE admin( + id INTEGER PRIMARY KEY, + name TEXT NOT NULL, + passwd TEXT NOT NULL + )""" +conn.execute(create_table) +conn.commit() + +# 写入初始管理员名字和密码 +print("请输入您的管理员名字") +admin_name = input() +print("请输入您的管理员名字") +admin_passwd = input() +# 配置写入数据库表单 +INSERTE_SQL = f"INSERT INTO admin (name, passwd) VALUES ('{admin_name}', '{admin_passwd}');" +conn.execute(INSERTE_SQL) +conn.commit() + +# 列出您输入的管理员名字和密码 +# 暂时没想好怎么写 +# 正常的应该是读取数据库再输出,因为太晚了,先随便谢谢凑合 +print('名字', f'{admin_name}') +print('密码', f'{admin_passwd}') +conn.close() \ No newline at end of file