Pine
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""默认头像池：新用户未提供头像时，从头像池随机分配一个默认头像。
|
||
|
||
头像池 = 对象存储桶 avatar/images/images/ 目录下的 1321 张头像
|
||
（对象 key 清单见 ``serverdata/avatar_pool.json``，由 code/images/images_data/images
|
||
本地目录与桶内对象一一对应生成）。
|
||
|
||
数据库只存对象相对路径 ``/oss/avatar/images/images/<file>``，
|
||
读取时经 ``resolve_url`` 补全为 CDN 直链（见 app/infrastructure/oss.py）。
|
||
|
||
语义约定：
|
||
- 新用户创建时若未显式提供头像（avatar 为空）→ 随机分配一个默认头像；
|
||
- 用户一旦自行更换头像（上传真实头像）→ 走 update 流程，不再回退/重随机。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import random
|
||
from pathlib import Path
|
||
from typing import List
|
||
|
||
from .. import config
|
||
|
||
_POOL_CACHE: List[str] | None = None
|
||
_POOL_PATH: Path = config.SERVERDATA_DIR / "avatar_pool.json"
|
||
|
||
|
||
def _load_pool() -> List[str]:
|
||
"""读取头像池对象 key 清单（内存缓存）。文件缺失/损坏时返回空池。"""
|
||
global _POOL_CACHE
|
||
if _POOL_CACHE is not None:
|
||
return _POOL_CACHE
|
||
try:
|
||
raw = json.loads(_POOL_PATH.read_text(encoding="utf-8"))
|
||
keys = [
|
||
str(k).strip()
|
||
for k in raw
|
||
if isinstance(k, str) and k.strip()
|
||
]
|
||
except Exception: # noqa: BLE001
|
||
keys = []
|
||
_POOL_CACHE = keys
|
||
return keys
|
||
|
||
|
||
def random_default_avatar() -> str:
|
||
"""从默认头像池随机返回一个对象路径（``/oss/...``）；池为空时返回空串。"""
|
||
pool = _load_pool()
|
||
if not pool:
|
||
return ""
|
||
return "/oss/" + random.choice(pool)
|