44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
# 初始化命令脚本
|
|
import sqlite3
|
|
import os
|
|
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')
|
|
# 建立管理员表
|
|
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() |