import logging
import os
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'
        )
        
        # 文件处理器
        file_handler = logging.FileHandler(log_file, 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('这是一条按日期存储的日志测试')