242 lines
10 KiB
Python
242 lines
10 KiB
Python
import base64
|
||
from fastapi import Depends
|
||
from fastapi.routing import APIRouter
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy import select
|
||
from database.creator import get_db_session
|
||
from database.user import UserProfile, User
|
||
from provider.loveac.authme import fetch_user_by_token, AuthmeRequest
|
||
from utils.file_manager import file_manager
|
||
from .model import (
|
||
UserProfileResponse,
|
||
GetUserProfileRequest,
|
||
UpdateUserProfileRequest,
|
||
UserProfileData,
|
||
UserSettings,
|
||
)
|
||
|
||
user_router = APIRouter(prefix="/api/v1/user")
|
||
|
||
|
||
@user_router.post("/profile/get", summary="获取用户资料")
|
||
async def get_user_profile(
|
||
data: GetUserProfileRequest,
|
||
asyncsession: AsyncSession = Depends(get_db_session),
|
||
):
|
||
"""
|
||
获取用户资料
|
||
:param data: GetUserProfileRequest
|
||
:return: UserProfileResponse
|
||
"""
|
||
try:
|
||
# 使用token验证获取用户
|
||
authme_request = AuthmeRequest(token=data.token)
|
||
user = await fetch_user_by_token(authme_request, asyncsession)
|
||
|
||
async with asyncsession as session:
|
||
result = await session.execute(
|
||
select(UserProfile).where(UserProfile.userid == user.userid)
|
||
)
|
||
profile = result.scalars().first()
|
||
|
||
if not profile:
|
||
# 如果用户资料不存在，创建默认资料
|
||
profile = UserProfile(
|
||
userid=user.userid,
|
||
avatar_filename=None,
|
||
background_filename=None,
|
||
nickname=None,
|
||
settings_filename=None
|
||
)
|
||
session.add(profile)
|
||
await session.commit()
|
||
|
||
# 获取头像数据
|
||
avatar_data = None
|
||
if profile.avatar_filename:
|
||
avatar_bytes = await file_manager.get_avatar(profile.avatar_filename)
|
||
if avatar_bytes:
|
||
# 转换为base64
|
||
avatar_data = base64.b64encode(avatar_bytes).decode('utf-8')
|
||
# 根据文件扩展名添加data URI前缀
|
||
if profile.avatar_filename.endswith('.png'):
|
||
avatar_data = f"data:image/png;base64,{avatar_data}"
|
||
elif profile.avatar_filename.endswith(('.jpg', '.jpeg')):
|
||
avatar_data = f"data:image/jpeg;base64,{avatar_data}"
|
||
elif profile.avatar_filename.endswith('.gif'):
|
||
avatar_data = f"data:image/gif;base64,{avatar_data}"
|
||
|
||
# 获取背景数据
|
||
background_data = None
|
||
if profile.background_filename:
|
||
background_bytes = await file_manager.get_background(profile.background_filename)
|
||
if background_bytes:
|
||
# 转换为base64
|
||
background_data = base64.b64encode(background_bytes).decode('utf-8')
|
||
# 根据文件扩展名添加data URI前缀
|
||
if profile.background_filename.endswith('.png'):
|
||
background_data = f"data:image/png;base64,{background_data}"
|
||
elif profile.background_filename.endswith(('.jpg', '.jpeg')):
|
||
background_data = f"data:image/jpeg;base64,{background_data}"
|
||
elif profile.background_filename.endswith('.gif'):
|
||
background_data = f"data:image/gif;base64,{background_data}"
|
||
elif profile.background_filename.endswith('.webp'):
|
||
background_data = f"data:image/webp;base64,{background_data}"
|
||
|
||
# 获取设置数据
|
||
settings_data = None
|
||
if profile.settings_filename:
|
||
settings_dict = await file_manager.get_settings(profile.settings_filename)
|
||
if settings_dict:
|
||
settings_data = UserSettings(**settings_dict)
|
||
|
||
profile_data = UserProfileData(
|
||
userid=profile.userid,
|
||
avatar=avatar_data,
|
||
background=background_data,
|
||
nickname=profile.nickname,
|
||
settings=settings_data,
|
||
)
|
||
|
||
return UserProfileResponse(
|
||
code=200,
|
||
message="获取用户资料成功",
|
||
data=profile_data
|
||
)
|
||
|
||
except Exception as e:
|
||
return UserProfileResponse(
|
||
code=500,
|
||
message=f"获取用户资料失败: {str(e)}",
|
||
data=None
|
||
)
|
||
|
||
|
||
@user_router.post("/profile/update", summary="更新用户资料")
|
||
async def update_user_profile(
|
||
data: UpdateUserProfileRequest,
|
||
asyncsession: AsyncSession = Depends(get_db_session),
|
||
):
|
||
"""
|
||
更新用户资料
|
||
:param data: UpdateUserProfileRequest
|
||
:return: UserProfileResponse
|
||
"""
|
||
try:
|
||
# 使用token验证获取用户
|
||
authme_request = AuthmeRequest(token=data.token)
|
||
user = await fetch_user_by_token(authme_request, asyncsession)
|
||
|
||
async with asyncsession as session:
|
||
result = await session.execute(
|
||
select(UserProfile).where(UserProfile.userid == user.userid)
|
||
)
|
||
profile = result.scalars().first()
|
||
|
||
if not profile:
|
||
# 如果用户资料不存在，创建新的
|
||
profile = UserProfile(
|
||
userid=user.userid,
|
||
avatar_filename=None,
|
||
background_filename=None,
|
||
nickname=data.nickname,
|
||
settings_filename=None
|
||
)
|
||
session.add(profile)
|
||
else:
|
||
# 更新昵称
|
||
if data.nickname is not None:
|
||
profile.nickname = data.nickname
|
||
|
||
# 处理头像更新
|
||
if data.avatar is not None:
|
||
if data.avatar: # 如果头像不为空
|
||
new_avatar_filename = await file_manager.save_avatar(user.userid, data.avatar)
|
||
profile.avatar_filename = new_avatar_filename
|
||
else: # 如果头像为空，表示删除头像
|
||
if profile.avatar_filename:
|
||
await file_manager.delete_avatar(profile.avatar_filename)
|
||
profile.avatar_filename = None
|
||
|
||
# 处理背景更新
|
||
if data.background is not None:
|
||
if data.background: # 如果背景不为空
|
||
new_background_filename = await file_manager.save_background(user.userid, data.background)
|
||
profile.background_filename = new_background_filename
|
||
else: # 如果背景为空，表示删除背景
|
||
if profile.background_filename:
|
||
await file_manager.delete_background(profile.background_filename)
|
||
profile.background_filename = None
|
||
|
||
# 处理设置更新
|
||
if data.settings is not None:
|
||
if data.settings: # 如果设置不为空
|
||
# data.settings在model验证时已经被转换为UserSettings对象
|
||
if isinstance(data.settings, UserSettings):
|
||
settings_dict = data.settings.model_dump()
|
||
new_settings_filename = await file_manager.save_settings(user.userid, settings_dict)
|
||
profile.settings_filename = new_settings_filename
|
||
else:
|
||
# 如果不是UserSettings对象，说明验证有问题
|
||
raise ValueError(f"Settings对象类型错误: {type(data.settings)}")
|
||
else: # 如果设置为空，表示删除设置
|
||
if profile.settings_filename:
|
||
await file_manager.delete_settings(profile.settings_filename)
|
||
profile.settings_filename = None
|
||
|
||
await session.commit()
|
||
await session.refresh(profile)
|
||
|
||
# 获取更新后的数据
|
||
avatar_data = None
|
||
if profile.avatar_filename:
|
||
avatar_bytes = await file_manager.get_avatar(profile.avatar_filename)
|
||
if avatar_bytes:
|
||
avatar_data = base64.b64encode(avatar_bytes).decode('utf-8')
|
||
if profile.avatar_filename.endswith('.png'):
|
||
avatar_data = f"data:image/png;base64,{avatar_data}"
|
||
elif profile.avatar_filename.endswith(('.jpg', '.jpeg')):
|
||
avatar_data = f"data:image/jpeg;base64,{avatar_data}"
|
||
elif profile.avatar_filename.endswith('.gif'):
|
||
avatar_data = f"data:image/gif;base64,{avatar_data}"
|
||
|
||
background_data = None
|
||
if profile.background_filename:
|
||
background_bytes = await file_manager.get_background(profile.background_filename)
|
||
if background_bytes:
|
||
background_data = base64.b64encode(background_bytes).decode('utf-8')
|
||
if profile.background_filename.endswith('.png'):
|
||
background_data = f"data:image/png;base64,{background_data}"
|
||
elif profile.background_filename.endswith(('.jpg', '.jpeg')):
|
||
background_data = f"data:image/jpeg;base64,{background_data}"
|
||
elif profile.background_filename.endswith('.gif'):
|
||
background_data = f"data:image/gif;base64,{background_data}"
|
||
elif profile.background_filename.endswith('.webp'):
|
||
background_data = f"data:image/webp;base64,{background_data}"
|
||
|
||
settings_data = None
|
||
if profile.settings_filename:
|
||
settings_dict = await file_manager.get_settings(profile.settings_filename)
|
||
if settings_dict:
|
||
settings_data = UserSettings(**settings_dict)
|
||
|
||
profile_data = UserProfileData(
|
||
userid=profile.userid,
|
||
avatar=avatar_data,
|
||
background=background_data,
|
||
nickname=profile.nickname,
|
||
settings=settings_data,
|
||
)
|
||
|
||
return UserProfileResponse(
|
||
code=200,
|
||
message="更新用户资料成功",
|
||
data=profile_data
|
||
)
|
||
|
||
except Exception as e:
|
||
return UserProfileResponse(
|
||
code=500,
|
||
message=f"更新用户资料失败: {str(e)}",
|
||
data=None
|
||
) |