From 210f0643df6ff34f9d9afa0ede138cdee57e0d28 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Tue, 15 Apr 2025 11:15:19 +0800
Subject: [PATCH] update

---
 app/api/v1/auth.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/app/api/v1/auth.py b/app/api/v1/auth.py
index 2e76327..1c41128 100644
--- a/app/api/v1/auth.py
+++ b/app/api/v1/auth.py
@@ -44,8 +44,8 @@ async def login(
     return StandardResponse(code=200, message="登录成功", data=User.model_validate(user))
 
 def generate_user_code():
-    """生成个人邀请码，6位数字加字母"""
-    return ''.join(random.choices(string.ascii_letters + string.digits, k=6))
+    """生成个人邀请码，6位数字加大写字母"""
+    return ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
 
 @router.post("/login/wechat", tags=["auth"])
 async def wechat_login(
@@ -106,6 +106,7 @@ async def wechat_login(
                 user.user_code = generate_user_code()
                 user.nickname = f"搭友{user.user_code}"
                 user.avatar = settings.DEFAULT_AVATAR
+                user.tryon_remain_count = settings.TRYON_REMAIN_COUNT
                 await db.commit()
                 await db.refresh(user)