Delete utils.py
Browse files
utils.py
DELETED
|
@@ -1,131 +0,0 @@
|
|
| 1 |
-
"""
|
| 2 |
-
工具函数模块
|
| 3 |
-
包含通用的工具函数和辅助方法
|
| 4 |
-
"""
|
| 5 |
-
import uuid
|
| 6 |
-
import time
|
| 7 |
-
import logging
|
| 8 |
-
from typing import Optional
|
| 9 |
-
from loguru import logger
|
| 10 |
-
from config import Config
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
class Utils:
|
| 14 |
-
"""工具函数集合"""
|
| 15 |
-
|
| 16 |
-
@staticmethod
|
| 17 |
-
def generate_request_id() -> str:
|
| 18 |
-
"""生成请求ID"""
|
| 19 |
-
return str(uuid.uuid4())
|
| 20 |
-
|
| 21 |
-
@staticmethod
|
| 22 |
-
def generate_token_id(refresh_token: str) -> str:
|
| 23 |
-
"""生成token的唯一标识符(用于日志显示)"""
|
| 24 |
-
# 使用token的前8位和后8位作为标识符
|
| 25 |
-
if len(refresh_token) >= 16:
|
| 26 |
-
return f"TOKEN_{refresh_token[:8]}...{refresh_token[-8:]}"
|
| 27 |
-
else:
|
| 28 |
-
return f"TOKEN_{refresh_token[:8]}..."
|
| 29 |
-
|
| 30 |
-
@staticmethod
|
| 31 |
-
def get_current_timestamp() -> int:
|
| 32 |
-
"""获取当前时间戳"""
|
| 33 |
-
return int(time.time())
|
| 34 |
-
|
| 35 |
-
@staticmethod
|
| 36 |
-
def validate_api_key(api_key: str) -> bool:
|
| 37 |
-
"""验证API密钥"""
|
| 38 |
-
config = Config()
|
| 39 |
-
return api_key in config.VALID_API_KEYS
|
| 40 |
-
|
| 41 |
-
@staticmethod
|
| 42 |
-
def extract_bearer_token(auth_header: str) -> Optional[str]:
|
| 43 |
-
"""从Authorization头中提取Bearer token"""
|
| 44 |
-
if not auth_header:
|
| 45 |
-
return None
|
| 46 |
-
|
| 47 |
-
parts = auth_header.split(" ")
|
| 48 |
-
if len(parts) != 2 or parts[0].lower() != "bearer":
|
| 49 |
-
return None
|
| 50 |
-
|
| 51 |
-
return parts[1]
|
| 52 |
-
|
| 53 |
-
@staticmethod
|
| 54 |
-
def setup_logging():
|
| 55 |
-
"""设置loguru日志配置"""
|
| 56 |
-
# 移除默认的日志处理器
|
| 57 |
-
logger.remove()
|
| 58 |
-
|
| 59 |
-
# 添加控制台日志输出,带颜色和格式
|
| 60 |
-
logger.add(
|
| 61 |
-
sink=lambda x: print(x, end=""),
|
| 62 |
-
format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
|
| 63 |
-
level="INFO",
|
| 64 |
-
colorize=True
|
| 65 |
-
)
|
| 66 |
-
|
| 67 |
-
# 根据环境变量决定是否启用文件日志
|
| 68 |
-
if Config.enable_file_logging():
|
| 69 |
-
logger.add(
|
| 70 |
-
"warp_api_server.log",
|
| 71 |
-
rotation="10 MB",
|
| 72 |
-
retention="7 days",
|
| 73 |
-
level="DEBUG",
|
| 74 |
-
format="{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {name}:{function}:{line} - {message}",
|
| 75 |
-
encoding="utf-8"
|
| 76 |
-
)
|
| 77 |
-
logger.info("📝 文件日志已启用: warp_api_server.log")
|
| 78 |
-
else:
|
| 79 |
-
logger.info("📝 文件日志已禁用 (可通过环境变量 ENABLE_FILE_LOGGING=true 启用)")
|
| 80 |
-
|
| 81 |
-
# 配置标准logging的loguru处理器
|
| 82 |
-
class InterceptHandler(logging.Handler):
|
| 83 |
-
def emit(self, record):
|
| 84 |
-
# 获取对应的loguru级别
|
| 85 |
-
try:
|
| 86 |
-
level = logger.level(record.levelname).name
|
| 87 |
-
except ValueError:
|
| 88 |
-
level = record.levelno
|
| 89 |
-
|
| 90 |
-
# 查找调用者位置
|
| 91 |
-
frame, depth = logging.currentframe(), 2
|
| 92 |
-
while frame.f_code.co_filename == logging.__file__:
|
| 93 |
-
frame = frame.f_back
|
| 94 |
-
depth += 1
|
| 95 |
-
|
| 96 |
-
logger.opt(depth=depth, exception=record.exc_info).log(
|
| 97 |
-
level, record.getMessage()
|
| 98 |
-
)
|
| 99 |
-
|
| 100 |
-
# 替换所有标准logging处理器
|
| 101 |
-
logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True)
|
| 102 |
-
logging.getLogger("urllib3").setLevel(logging.WARNING)
|
| 103 |
-
|
| 104 |
-
logger.info("🚀 Loguru日志系统初始化完成")
|
| 105 |
-
|
| 106 |
-
@staticmethod
|
| 107 |
-
def deduplicate_tokens(tokens: list) -> list:
|
| 108 |
-
"""去重token列表,保持顺序"""
|
| 109 |
-
seen = set()
|
| 110 |
-
result = []
|
| 111 |
-
for token in tokens:
|
| 112 |
-
if token and token not in seen:
|
| 113 |
-
seen.add(token)
|
| 114 |
-
result.append(token)
|
| 115 |
-
return result
|
| 116 |
-
|
| 117 |
-
@staticmethod
|
| 118 |
-
def validate_refresh_token_format(token: str) -> bool:
|
| 119 |
-
"""验证refresh token格式是否有效"""
|
| 120 |
-
if not token or not isinstance(token, str):
|
| 121 |
-
return False
|
| 122 |
-
|
| 123 |
-
# 基本长度检查
|
| 124 |
-
if len(token) < 100:
|
| 125 |
-
return False
|
| 126 |
-
|
| 127 |
-
# 检查是否包含必要的字符
|
| 128 |
-
if not any(c.isalnum() for c in token):
|
| 129 |
-
return False
|
| 130 |
-
|
| 131 |
-
return True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|