Pine
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""存量用户默认头像同步脚本。
|
||
|
||
给「头像为空」的存量用户随机分配一个默认头像（来自 avatar_pool.json 池）。
|
||
已有真实头像（非空、非默认池前缀）的用户不会被覆盖。
|
||
|
||
用法：
|
||
python scripts/sync_default_avatars.py # 实际执行
|
||
python scripts/sync_default_avatars.py --dry-run # 只预览不写库
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
import asyncio
|
||
from pathlib import Path
|
||
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv(Path(__file__).resolve().parent.parent / ".env")
|
||
|
||
from sqlalchemy import text # noqa: E402
|
||
|
||
from app.infrastructure.db import AsyncSessionLocal # noqa: E402
|
||
from app.infrastructure.repositories import utcnow_iso # noqa: E402
|
||
from app.services.avatar_pool import random_default_avatar # noqa: E402
|
||
|
||
DEFAULT_POOL_PREFIX = "/oss/avatar/images/images/"
|
||
|
||
|
||
async def main(dry_run: bool) -> int:
|
||
async with AsyncSessionLocal() as session:
|
||
# 仅处理头像为空（或 NULL）的存量用户；已设置真实头像的不动
|
||
rows = (
|
||
await session.execute(
|
||
text(
|
||
"SELECT id, nickname, username FROM users "
|
||
"WHERE avatar IS NULL OR avatar = ''"
|
||
)
|
||
)
|
||
).all()
|
||
total = len(rows)
|
||
print(f"待同步用户数（头像为空）: {total}")
|
||
if total == 0:
|
||
print("无需同步。")
|
||
return 0
|
||
|
||
updated = 0
|
||
for row in rows:
|
||
avatar = random_default_avatar()
|
||
if not avatar:
|
||
print("警告：默认头像池为空，跳过（请检查 serverdata/avatar_pool.json）")
|
||
return 1
|
||
print(
|
||
f" {'[DRY] ' if dry_run else '[set] '}{row.id} "
|
||
f"({row.nickname or row.username or ''}) -> {avatar}"
|
||
)
|
||
if not dry_run:
|
||
await session.execute(
|
||
text("UPDATE users SET avatar = :av, updated_at = :now WHERE id = :uid"),
|
||
{"av": avatar, "uid": row.id, "now": utcnow_iso()},
|
||
)
|
||
updated += 1
|
||
|
||
if not dry_run:
|
||
await session.commit()
|
||
print(f"完成：{'（dry-run 未写库）' if dry_run else ''}计划/已更新 {updated} 人。")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
parser = argparse.ArgumentParser(description="存量用户默认头像同步")
|
||
parser.add_argument("--dry-run", action="store_true", help="只预览不写库")
|
||
args = parser.parse_args()
|
||
raise SystemExit(asyncio.run(main(args.dry_run)))
|