dekun
25 lines
903 B
Python
25 lines
903 B
Python
#!/usr/bin/env python3
|
||
"""从 .env 重置管理员账号（服务器上忘记密码时使用）"""
|
||
import os
|
||
import sys
|
||
|
||
from dotenv import load_dotenv
|
||
from werkzeug.security import generate_password_hash
|
||
|
||
BASE = os.path.dirname(os.path.abspath(__file__))
|
||
load_dotenv(os.path.join(BASE, ".env"))
|
||
|
||
sys.path.insert(0, BASE)
|
||
from app import set_setting, get_setting # noqa: E402
|
||
|
||
username = os.getenv("ADMIN_USERNAME", "admin").strip() or "admin"
|
||
password = os.getenv("ADMIN_PASSWORD", "").strip()
|
||
if not password or password == "change-me-on-first-login":
|
||
print("请在 .env 中设置 ADMIN_PASSWORD 后再运行此脚本")
|
||
sys.exit(1)
|
||
|
||
old_username = get_setting("admin_username")
|
||
set_setting("admin_username", username)
|
||
set_setting("admin_password_hash", generate_password_hash(password))
|
||
print(f"已重置管理员: {username}（原账号: {old_username or '无'}）")
|