80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""创建唯一后台用户或重置其密码，不提供 HTTP 注册入口。"""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import asyncio
|
|
import getpass
|
|
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import delete, select
|
|
|
|
from app.auth import hash_password
|
|
from app.config import settings
|
|
from app.db import async_session, engine
|
|
from app.models import AuthSession, User
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="创建后台用户或重置密码")
|
|
parser.add_argument("--username", default="admin", help="登录用户名，默认 admin")
|
|
return parser.parse_args()
|
|
|
|
|
|
async def upsert_user(username: str, password: str) -> str:
|
|
now = datetime.now(timezone.utc)
|
|
async with async_session() as db:
|
|
user = (await db.execute(select(User).where(User.username == username))).scalar_one_or_none()
|
|
if user is None:
|
|
user = User(
|
|
username=username,
|
|
password_hash=hash_password(password),
|
|
password_algo="argon2id",
|
|
is_active=True,
|
|
failed_login_count=0,
|
|
password_changed_at=now,
|
|
created_at=now,
|
|
updated_at=now,
|
|
)
|
|
db.add(user)
|
|
action = "created"
|
|
else:
|
|
user.password_hash = hash_password(password)
|
|
user.password_algo = "argon2id"
|
|
user.is_active = True
|
|
user.failed_login_count = 0
|
|
user.locked_until = None
|
|
user.password_changed_at = now
|
|
user.updated_at = now
|
|
await db.execute(delete(AuthSession).where(AuthSession.user_id == user.id))
|
|
action = "updated"
|
|
await db.commit()
|
|
return action
|
|
|
|
|
|
async def async_main() -> None:
|
|
args = parse_args()
|
|
username = args.username.strip()
|
|
if not username or len(username) > 64:
|
|
raise SystemExit("用户名长度必须为 1-64 个字符")
|
|
|
|
password = getpass.getpass("Password: ")
|
|
confirm = getpass.getpass("Confirm password: ")
|
|
if password != confirm:
|
|
raise SystemExit("两次密码输入不一致")
|
|
if len(password) < settings.auth_min_password_length:
|
|
raise SystemExit(f"密码至少需要 {settings.auth_min_password_length} 个字符")
|
|
|
|
try:
|
|
action = await upsert_user(username, password)
|
|
print(f"User {username!r} {action}. All previous sessions were revoked.")
|
|
finally:
|
|
await engine.dispose()
|
|
|
|
|
|
def main() -> None:
|
|
asyncio.run(async_main())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|