完成初始化程序命令

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')
+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()