76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
"""创建超级管理员账号。
|
||
|
||
使用方式：
|
||
cd D:/work/code/python/coding/baodanagent
|
||
python scripts/create_admin.py
|
||
|
||
默认账号：admin / Admin@123456
|
||
"""
|
||
import sys
|
||
import os
|
||
|
||
# 添加 api 目录到 Python 路径
|
||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'api'))
|
||
|
||
# 加载环境变量
|
||
from dotenv import load_dotenv
|
||
load_dotenv(os.path.join(os.path.dirname(__file__), '..', '.env'))
|
||
|
||
|
||
def create_admin():
|
||
"""创建超级管理员账号。"""
|
||
try:
|
||
from insurance.app import create_app
|
||
from insurance.db.compat import db
|
||
from insurance.models.wecom_user import WeComUserMapping
|
||
import bcrypt
|
||
import uuid
|
||
|
||
app = create_app()
|
||
|
||
with app.app_context():
|
||
username = "admin"
|
||
password = "Admin@123456"
|
||
|
||
# 检查是否已存在
|
||
existing = db.session.query(WeComUserMapping).filter_by(username=username).first()
|
||
if existing:
|
||
print(f"用户 {username} 已存在，跳过创建")
|
||
print(f" 角色: {existing.role}")
|
||
print(f" 状态: {existing.status}")
|
||
return
|
||
|
||
# 创建用户
|
||
password_hash = bcrypt.hashpw(password.encode(), bcrypt.gensalt()).decode()
|
||
mapping = WeComUserMapping(
|
||
wecom_userid=f"local_{uuid.uuid4().hex[:8]}",
|
||
internal_user_id=f"user_{uuid.uuid4().hex[:8]}",
|
||
username=username,
|
||
password_hash=password_hash,
|
||
role="super_admin",
|
||
department="管理组",
|
||
status="active",
|
||
)
|
||
db.session.add(mapping)
|
||
db.session.commit()
|
||
|
||
print("=" * 50)
|
||
print("超级管理员创建成功！")
|
||
print("=" * 50)
|
||
print(f" 用户名: {username}")
|
||
print(f" 密码: {password}")
|
||
print(f" 角色: super_admin")
|
||
print("=" * 50)
|
||
print("登录后即可看到管理后台菜单。")
|
||
|
||
except Exception as e:
|
||
print(f"创建失败: {e}")
|
||
print(f"\n请确保：")
|
||
print(f"1. 数据库已启动并创建")
|
||
print(f"2. .env 文件中 DATABASE_URL 已配置")
|
||
print(f"3. 已安装所需依赖 (pip install -r requirements.txt)")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
create_admin()
|