39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
创建初始用户脚本
|
|
用于创建管理员账户
|
|
"""
|
|
|
|
from backend.app import create_app
|
|
from backend.models import db, User
|
|
|
|
def create_initial_user():
|
|
"""创建初始用户"""
|
|
app = create_app()
|
|
|
|
with app.app_context():
|
|
# 检查是否已有用户
|
|
existing_user = User.query.first()
|
|
|
|
if existing_user:
|
|
print(f"用户已存在: {existing_user.username}")
|
|
return
|
|
|
|
# 创建默认管理员用户
|
|
username = input("请输入用户名 (默认: admin): ").strip() or "admin"
|
|
password = input("请输入密码 (默认: admin123): ").strip() or "admin123"
|
|
|
|
user = User(username=username)
|
|
user.set_password(password)
|
|
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
|
|
print(f"用户创建成功!")
|
|
print(f"用户名: {username}")
|
|
print(f"请妥善保管密码!")
|
|
|
|
if __name__ == '__main__':
|
|
create_initial_user()
|