62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
import os
|
||
from app import create_app
|
||
from extensions import db
|
||
from models import User, Device, UserDevicePermission
|
||
|
||
# 创建应用实例
|
||
app = create_app()
|
||
|
||
|
||
def init_db():
|
||
with app.app_context():
|
||
# ==========================================
|
||
# ⚠️ 警告：这会清空现有的数据库表结构并重建
|
||
# 如果只想更新 User 表，可以注释掉 db.drop_all()，
|
||
# 但因为增加了字段，直接重建是最稳妥的。
|
||
# ==========================================
|
||
print("正在清理旧数据库...")
|
||
db.drop_all()
|
||
|
||
print("正在创建新表结构...")
|
||
db.create_all()
|
||
|
||
print("✅ 数据库表结构创建完成 (devices.db 和 users.db)")
|
||
|
||
# ==========================================
|
||
# 🟢 1. 创建超级管理员 (Root)
|
||
# 即使代码里有后门，数据库里有一个对应的实体也是最好的
|
||
# ==========================================
|
||
admin = User(username='admin', role='admin')
|
||
admin.set_password('licahk') # 设置密码
|
||
db.session.add(admin)
|
||
print(f"👤 用户创建: [admin] (角色: 超级管理员)")
|
||
|
||
# ==========================================
|
||
# 🟡 2. 创建一个测试工程师 (可选)
|
||
# ==========================================
|
||
engineer = User(username='engineer01', role='engineer')
|
||
engineer.set_password('123456')
|
||
db.session.add(engineer)
|
||
print(f"👤 用户创建: [engineer01] (角色: 工程师)")
|
||
|
||
# ==========================================
|
||
# ⚪ 3. 创建一个测试普通客户 (可选)
|
||
# ==========================================
|
||
client = User(username='client01', role='client')
|
||
client.set_password('123456')
|
||
db.session.add(client)
|
||
print(f"👤 用户创建: [client01] (角色: 客户)")
|
||
|
||
# 提交更改
|
||
db.session.commit()
|
||
print("\n🚀 初始化完成！请运行 run.py 启动服务器。")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 再次确认防止误删
|
||
print("此操作会删除现有的 'users.db' 和 'devices.db' 中的数据并重建。")
|
||
confirm = input("确认继续吗? (y/n): ")
|
||
if confirm.lower() == 'y':
|
||
init_db()
|
||
else:
|
||
print("操作已取消。") |