wdg1122/Snap.Server
1
0
Fork 0
mirror of https://github.com/wangdage12/Snap.Server.git synced 2026-02-17 08:52:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加类型注释、修复抽卡id问题

Browse Source
This commit is contained in:
fanbook-wangdage
2026-02-03 12:12:45 +08:00
parent aa82a19ac7
commit 6b82806931
8 changed files with 74 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/extensions.py
View File

@@ -26,10 +26,10 @@ def init_mongo(uri: str, test_mode=False):
logger.error(f"MongoDB connection failed: {e}")
raise
def generate_code(length=6):
def generate_code(length=6) -> str:
"""生成数字验证码"""
return ''.join(secrets.choice('0123456789') for _ in range(length))
def generate_numeric_id(length=8):
def generate_numeric_id(length=8) -> str:
"""生成数字ID"""
return ''.join(secrets.choice(string.digits) for _ in range(length))