优化个人中心与界面交互

backend/app/api/v1/auth.py

@@ -42,6 +42,12 @@ class ExtendResponse(BaseModel):
 router = APIRouter()
 AVATAR_ROOT = Path(__file__).resolve().parent.parent.parent / "uploads" / "avatars"
 AVATAR_ROOT.mkdir(parents=True, exist_ok=True)
+AVATAR_ALLOWED_CONTENT_TYPES = {
+    "image/png": ".png",
+    "image/jpeg": ".jpg",
+    "image/gif": ".gif",
+    "image/webp": ".webp",
+}
 
 
 def issue_user_token(db_user) -> Token:
@@ -215,10 +221,15 @@ async def upload_avatar(
     current_user=Depends(get_current_user),
     db: AsyncSession = Depends(get_db_session),
 ) -> UserRead:
+    ext = AVATAR_ALLOWED_CONTENT_TYPES.get(file.content_type or "")
+    if not ext:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="头像仅支持图片格式",
+        )
     AVATAR_ROOT.mkdir(parents=True, exist_ok=True)
     user_dir = AVATAR_ROOT / str(current_user.id)
     user_dir.mkdir(parents=True, exist_ok=True)
-    ext = Path(file.filename).suffix or ".png"
     filename = f"{uuid.uuid4()}{ext}"
     dest = user_dir / filename
     content = await file.read()