初始提交

app/extensions.py

@@ -0,0 +1,35 @@
+import logging
+import coloredlogs
+import secrets
+import string
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+from app.config_loader import config_loader
+
+logger = logging.getLogger("app")
+coloredlogs.install(level=config_loader.LOGGING_LEVEL, logger=logger, fmt=config_loader.LOGGING_FORMAT)
+
+client = None
+
+def init_mongo(uri: str, test_mode=False):
+    global client
+    if test_mode:
+        logger.info("Running in test mode, skipping MongoDB connection")
+        return
+    
+    client = MongoClient(uri, server_api=ServerApi('1'))
+
+    try:
+        client.admin.command('ping')
+        logger.info("MongoDB connected successfully")
+    except Exception as e:
+        logger.error(f"MongoDB connection failed: {e}")
+        raise
+
+def generate_code(length=6):
+    """生成数字验证码"""
+    return ''.join(secrets.choice('0123456789') for _ in range(length))
+
+def generate_numeric_id(length=8):
+    """生成数字ID"""
+    return ''.join(secrets.choice(string.digits) for _ in range(length))