feat: complete huihui square avatar workflows

This commit is contained in:
stefanfeng
2026-07-24 14:04:21 +08:00
parent 2ef58a44b8
commit e3bda469bb
68 changed files with 8062 additions and 132 deletions

View File

@@ -1,5 +1,8 @@
"""虚拟用户管理接口"""
from typing import Optional
from pathlib import Path
import uuid
import mimetypes
from fastapi import APIRouter, Depends, Query, UploadFile, File, HTTPException
from fastapi.responses import StreamingResponse
import io
@@ -9,6 +12,21 @@ from app.schemas import ApiResponse, UserCreateRequest, UserUpdateRequest, UserB
from app.services.user_service import user_service
router = APIRouter()
_UPLOADS_DIR = Path(__file__).resolve().parents[2] / "uploads" / "avatars"
_UPLOADS_DIR.mkdir(parents=True, exist_ok=True)
async def _save_local_avatar(file_bytes: bytes, filename: str, content_type: str | None) -> str:
ext = Path(filename or "").suffix.lower()
if ext not in {".jpg", ".jpeg", ".png", ".gif", ".webp"}:
guessed_ext = mimetypes.guess_extension(content_type or "") or ".jpg"
ext = ".jpg" if guessed_ext == ".jpe" else guessed_ext
if ext not in {".jpg", ".jpeg", ".png", ".gif", ".webp"}:
ext = ".jpg"
target = _UPLOADS_DIR / f"{uuid.uuid4().hex}{ext}"
target.write_bytes(file_bytes)
return f"/api/uploads/avatars/{target.name}"
@router.get("")
@@ -268,13 +286,20 @@ async def sync_all_profiles(db=Depends(get_db)):
sess = await get_session(uid)
if not sess:
return False
ur = await s.execute(select(_VU).where(_VU.id == uid))
user = ur.scalar_one_or_none()
if not user:
return False
platform_uid = sess.get("platform_uid", "")
# 登录成功时 session 里已存有用户信息
vals = {}
if platform_uid: vals["platform_uid"] = platform_uid
# session 里的字段(登录时写入)
if sess.get("nickname"): vals["nickname"] = sess["nickname"]
if sess.get("real_name"): vals["real_name"] = sess["real_name"]
sync_nickname = sess.get("nickname", "")
sync_real_name = sess.get("real_name", "")
resolved_nickname = news_service._resolve_synced_nickname(user, sync_nickname, sync_real_name)
if resolved_nickname: vals["nickname"] = resolved_nickname
if sync_real_name: vals["real_name"] = sync_real_name
if sess.get("sex"): vals["sex"] = int(sess["sex"])
if sess.get("avatar"): vals["avatar_url"] = sess["avatar"]
if vals:
@@ -327,9 +352,8 @@ async def upload_avatar(
else:
return ApiResponse(code=500, message=f"头像上传到平台失败: {result}")
else:
# 仅本地存储(转 base64 或存储到本地)
import base64
avatar_url = f"data:{file.content_type};base64,{base64.b64encode(file_bytes).decode()}"
# 未登录用户本地落盘,避免 base64 超过 avatar_url 字段长度
avatar_url = await _save_local_avatar(file_bytes, file.filename or "", file.content_type)
# 更新数据库
await db.execute(update(_VU).where(_VU.id == user_id).values(avatar_url=avatar_url))