diff --git a/app/api/v1/auth.py b/app/api/v1/auth.py index e4565f6..2e76327 100644 --- a/app/api/v1/auth.py +++ b/app/api/v1/auth.py @@ -106,7 +106,8 @@ async def wechat_login( user.user_code = generate_user_code() user.nickname = f"搭友{user.user_code}" user.avatar = settings.DEFAULT_AVATAR - await user_service.update_user(db, user_id=user.id, user_update=user) + await db.commit() + await db.refresh(user) # 创建访问令牌 - 使用openid作为标识 diff --git a/app/api/v1/tryon.py b/app/api/v1/tryon.py index 5562de8..3f81c7a 100644 --- a/app/api/v1/tryon.py +++ b/app/api/v1/tryon.py @@ -32,6 +32,10 @@ async def tryon( """ 试穿请求 """ + # 获取当前用户的试穿次数 + tryon_remain_count = current_user.tryon_remain_count + if tryon_remain_count <= 0: + raise BusinessError(code=400, message="试穿次数不足") # 获取当前用户的默认形象 person_image = await person_image_service.get_default_image(db, current_user.id)