letterm commited on
Commit
6928332
·
verified ·
1 Parent(s): 1c6ea51

Delete utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -131
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