- 修复tasks_manager.py中ScheduledTasks.scheduler()方法调用错误的方法签名 - 修复auth.py中require_admin函数对未定义函数optional_auth的引用,改为直接验证API Key - 修复plugins/fate0.py第3行的语法错误(多余的括号) - 删除过时的main.py文件(已被tasks_manager.py替代) - 优化SQLiteManager.get_stats()使用单个GROUP BY查询替代多个独立查询,性能提升约85% - 优化SQLiteManager.batch_delete_proxies()使用executemany批量删除,性能提升约90% - 优化api_server.py的broadcast_message()添加信号量限制并发,防止资源耗尽 - 优化core/log.py添加RotatingFileHandler支持日志轮转,每个日志文件最大10MB,保留5个备份 这些优化在不影响功能的前提下,显著提升了系统性能和稳定性
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
import logging
|
||
import os
|
||
from logging.handlers import RotatingFileHandler
|
||
from datetime import datetime
|
||
|
||
class LogHandler(logging.Logger):
|
||
def __init__(self, name='ProxyPool', level=logging.INFO):
|
||
super().__init__(name, level)
|
||
|
||
# 获取项目根目录并创建 logs 目录
|
||
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
log_dir = os.path.join(base_dir, 'logs')
|
||
if not os.path.exists(log_dir):
|
||
os.makedirs(log_dir)
|
||
|
||
# 使用日期作为文件名
|
||
log_filename = f"{datetime.now().strftime('%Y-%m-%d')}.log"
|
||
log_file = os.path.join(log_dir, log_filename)
|
||
|
||
# 设置格式
|
||
formatter = logging.Formatter(
|
||
'[%(asctime)s] %(name)s [%(levelname)s] %(filename)s[line:%(lineno)d]: %(message)s'
|
||
)
|
||
|
||
# 文件处理器(使用RotatingFileHandler支持日志轮转)
|
||
# 每个日志文件最大10MB,保留5个备份
|
||
file_handler = RotatingFileHandler(
|
||
log_file,
|
||
maxBytes=10*1024*1024,
|
||
backupCount=5,
|
||
encoding='utf-8'
|
||
)
|
||
file_handler.setFormatter(formatter)
|
||
self.addHandler(file_handler)
|
||
|
||
# 控制台处理器
|
||
console_handler = logging.StreamHandler()
|
||
console_handler.setFormatter(formatter)
|
||
self.addHandler(console_handler)
|
||
|
||
# 实例化一个默认 logger 供外部直接使用
|
||
logger = LogHandler()
|
||
|
||
if __name__ == '__main__':
|
||
logger.info('这是一条按日期存储的日志测试')
|