feat: 娣诲姞澶氫釜鏂板姛鑳藉拰淇 - 鍖呮嫭鐢ㄦ埛绠＄悊銆佹暟鎹簱杩佺Щ銆丟it鎺ㄩ€佸伐鍏风瓑
This commit is contained in:
赵杰 Jie Zhao （雄狮汽车科技）
173
create_admin_user.py
Normal file
173
create_admin_user.py
Normal file
@@ -0,0 +1,173 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
创建或修复默认管理员用户
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
from src.core.database import db_manager
|
||||
from src.core.models import User
|
||||
from werkzeug.security import generate_password_hash, check_password_hash
|
||||
from sqlalchemy import text, inspect
|
||||
|
||||
print("=" * 60)
|
||||
print("创建/修复管理员用户")
|
||||
print("=" * 60)
|
||||
|
||||
try:
|
||||
with db_manager.get_session() as session:
|
||||
# 检查表结构
|
||||
inspector = inspect(db_manager.engine)
|
||||
if 'users' not in inspector.get_table_names():
|
||||
print("错误: users表不存在，请先运行 python init_database.py")
|
||||
sys.exit(1)
|
||||
|
||||
existing_columns = [col['name'] for col in inspector.get_columns('users')]
|
||||
print(f"users表字段: {existing_columns}")
|
||||
|
||||
# 检查是否存在admin用户
|
||||
admin_user = session.query(User).filter(User.username == 'admin').first()
|
||||
|
||||
if admin_user:
|
||||
print(f"\n找到admin用户 (ID: {admin_user.id})")
|
||||
print(f" 邮箱: {admin_user.email}")
|
||||
print(f" 角色: {admin_user.role}")
|
||||
print(f" 激活状态: {admin_user.is_active}")
|
||||
|
||||
# 验证密码
|
||||
password_ok = check_password_hash(admin_user.password_hash, 'admin123')
|
||||
print(f" 密码验证: {'正确' if password_ok else '错误'}")
|
||||
|
||||
if not password_ok:
|
||||
print("\n密码不匹配，正在更新密码...")
|
||||
admin_user.password_hash = generate_password_hash('admin123')
|
||||
admin_user.is_active = True
|
||||
if hasattr(admin_user, 'region'):
|
||||
admin_user.region = None
|
||||
session.commit()
|
||||
print("密码已更新为: admin123")
|
||||
|
||||
if not admin_user.is_active:
|
||||
print("用户未激活，正在激活...")
|
||||
admin_user.is_active = True
|
||||
session.commit()
|
||||
print("用户已激活")
|
||||
|
||||
# 最终验证
|
||||
test_password = check_password_hash(admin_user.password_hash, 'admin123')
|
||||
if test_password and admin_user.is_active:
|
||||
print("\n管理员用户已就绪！")
|
||||
print(" 用户名: admin")
|
||||
print(" 密码: admin123")
|
||||
print(" 状态: 已激活")
|
||||
else:
|
||||
print("\n警告: 用户状态异常")
|
||||
print(f" 密码正确: {test_password}")
|
||||
print(f" 已激活: {admin_user.is_active}")
|
||||
else:
|
||||
print("\n未找到admin用户，正在创建...")
|
||||
|
||||
# 准备密码哈希
|
||||
password_hash = generate_password_hash('admin123')
|
||||
|
||||
# 检查表结构，使用SQL直接插入避免字段不匹配
|
||||
try:
|
||||
# 先尝试使用模型创建
|
||||
new_admin = User(
|
||||
username='admin',
|
||||
email='admin@tsp.com',
|
||||
password_hash=password_hash,
|
||||
role='admin',
|
||||
is_active=True
|
||||
)
|
||||
if 'region' in existing_columns:
|
||||
new_admin.region = None
|
||||
session.add(new_admin)
|
||||
session.commit()
|
||||
print("使用模型创建成功")
|
||||
except Exception as model_error:
|
||||
print(f"模型创建失败: {model_error}")
|
||||
print("尝试使用SQL直接插入...")
|
||||
session.rollback()
|
||||
|
||||
# 使用SQL直接插入
|
||||
insert_fields = ['username', 'email', 'password_hash', 'role']
|
||||
insert_values = {
|
||||
'username': 'admin',
|
||||
'email': 'admin@tsp.com',
|
||||
'password_hash': password_hash,
|
||||
'role': 'admin'
|
||||
}
|
||||
|
||||
if 'is_active' in existing_columns:
|
||||
insert_fields.append('is_active')
|
||||
insert_values['is_active'] = True
|
||||
|
||||
if 'region' in existing_columns:
|
||||
insert_fields.append('region')
|
||||
insert_values['region'] = None
|
||||
|
||||
fields_str = ', '.join(insert_fields)
|
||||
values_str = ', '.join([f":{k}" for k in insert_fields])
|
||||
|
||||
sql = f"""
|
||||
INSERT INTO users ({fields_str})
|
||||
VALUES ({values_str})
|
||||
"""
|
||||
|
||||
session.execute(text(sql), insert_values)
|
||||
session.commit()
|
||||
print("使用SQL创建成功")
|
||||
|
||||
# 验证创建结果
|
||||
verify_user = session.query(User).filter(User.username == 'admin').first()
|
||||
if verify_user:
|
||||
test_password = check_password_hash(verify_user.password_hash, 'admin123')
|
||||
print(f"\n验证结果:")
|
||||
print(f" 用户ID: {verify_user.id}")
|
||||
print(f" 密码正确: {test_password}")
|
||||
print(f" 已激活: {verify_user.is_active}")
|
||||
|
||||
if test_password and verify_user.is_active:
|
||||
print("\n管理员用户创建成功！")
|
||||
print(" 用户名: admin")
|
||||
print(" 密码: admin123")
|
||||
else:
|
||||
print("\n警告: 用户创建成功但状态异常")
|
||||
|
||||
# 创建其他示例用户
|
||||
for username, email, password, role, region in [
|
||||
('overseas_ops', 'overseas@tsp.com', 'ops123', 'overseas_ops', 'overseas'),
|
||||
('domestic_ops', 'domestic@tsp.com', 'ops123', 'domestic_ops', 'domestic')
|
||||
]:
|
||||
ops_user = session.query(User).filter(User.username == username).first()
|
||||
if not ops_user:
|
||||
print(f"\n创建{username}用户...")
|
||||
try:
|
||||
new_user = User(
|
||||
username=username,
|
||||
email=email,
|
||||
password_hash=generate_password_hash(password),
|
||||
role=role,
|
||||
is_active=True
|
||||
)
|
||||
if 'region' in existing_columns:
|
||||
new_user.region = region
|
||||
session.add(new_user)
|
||||
session.commit()
|
||||
print(f" {username}用户创建成功")
|
||||
except Exception as e:
|
||||
print(f" {username}用户创建失败: {e}")
|
||||
session.rollback()
|
||||
|
||||
print("\n" + "=" * 60)
|
||||
print("操作完成！")
|
||||
print("=" * 60)
|
||||
|
||||
except Exception as e:
|
||||
print(f"错误: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user