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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|