84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
重置管理员密码脚本
|
||
用于解决密码哈希兼容性问题
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
|
||
# 添加项目根目录到Python路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
# 尝试加载.env文件（在导入app之前，确保环境变量生效）
|
||
try:
|
||
from dotenv import load_dotenv
|
||
env_loaded = load_dotenv()
|
||
if env_loaded:
|
||
print("成功加载.env文件")
|
||
else:
|
||
print("未找到或无法加载.env文件，将使用默认配置")
|
||
except ImportError:
|
||
print("python-dotenv未安装，跳过.env文件加载")
|
||
|
||
from app import create_app, db
|
||
from app.models.admin import Admin
|
||
from werkzeug.security import generate_password_hash
|
||
|
||
def reset_admin_password(username='admin', new_password='admin123'):
|
||
"""重置管理员密码"""
|
||
# 检测环境配置
|
||
config_name = os.environ.get('FLASK_CONFIG', 'production')
|
||
|
||
# 检查数据库配置
|
||
database_url = os.environ.get('DATABASE_URL', '')
|
||
if database_url:
|
||
print(f"使用数据库: {database_url.split('@')[-1] if '@' in database_url else database_url[:50]}...")
|
||
else:
|
||
print("警告: 未设置 DATABASE_URL 环境变量，将使用默认配置")
|
||
|
||
# 使用生产环境配置
|
||
app = create_app(config_name)
|
||
|
||
with app.app_context():
|
||
# 查找管理员用户（排除已删除的）
|
||
admin = Admin.query.filter_by(username=username, is_deleted=0).first()
|
||
|
||
if not admin:
|
||
print(f"错误: 用户 '{username}' 不存在或已被删除")
|
||
return False
|
||
|
||
try:
|
||
# 重新设置密码（使用新格式）
|
||
admin.set_password(new_password)
|
||
db.session.commit()
|
||
print(f"成功: 用户 '{username}' 的密码已重置为: {new_password}")
|
||
print(f"新的密码哈希格式: {admin.password_hash[:50]}...")
|
||
return True
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
print(f"错误: 重置密码失败 - {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
if __name__ == '__main__':
|
||
# 支持命令行参数或交互式输入
|
||
if len(sys.argv) > 1:
|
||
# 命令行参数方式
|
||
username = sys.argv[1]
|
||
new_password = sys.argv[2] if len(sys.argv) > 2 else 'admin123'
|
||
else:
|
||
# 交互式输入方式
|
||
username = input("请输入要重置的用户名 (默认: admin): ").strip() or 'admin'
|
||
new_password = input("请输入新密码 (默认: admin123): ").strip() or 'admin123'
|
||
|
||
print(f"正在重置用户 '{username}' 的密码...")
|
||
success = reset_admin_password(username, new_password)
|
||
|
||
if success:
|
||
print("密码重置完成！现在可以使用新密码登录了。")
|
||
else:
|
||
print("密码重置失败，请检查错误信息。")
|
||
|
||
sys.exit(0 if success else 1) |