From 7ec1415428fda38ec9b8726044addd45d8023c45 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Tue, 15 Apr 2025 11:12:26 +0800
Subject: [PATCH] update

---
 app/api/v1/auth.py  | 3 ++-
 app/api/v1/tryon.py | 4 ++++
 2 files changed, 6 insertions(+), 1 deletion(-)

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)