94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
"""自定义日志 Handler：按日期分文件，兼容 Windows 文件锁。
|
||
|
||
替代 TimedRotatingFileHandler，避免 Windows 上因 rename 操作遇到文件锁
|
||
（PermissionError: [WinError 32]）导致日志丢失的问题。
|
||
|
||
文件命名规则：{prefix}-YYYY-MM-DD.log
|
||
日期切换时自动打开新文件，无需 rename 旧文件。
|
||
"""
|
||
|
||
import logging
|
||
import time
|
||
from pathlib import Path
|
||
|
||
|
||
class DailyFileHandler(logging.Handler):
|
||
"""按日期自动分文件的日志 Handler。
|
||
|
||
与 TimedRotatingFileHandler 的关键区别：
|
||
- 文件直接以日期命名，日期切换时打开新文件，**不 rename 旧文件**
|
||
- 多进程（dev server + 测试 / 管理命令）可同时写入，互不阻塞
|
||
- 自动清理超过 backup_count 天的旧文件
|
||
|
||
dictConfig 用法::
|
||
|
||
'audit_file': {
|
||
'class': 'config.logging_handlers.DailyFileHandler',
|
||
'dir_path': '/path/to/logs',
|
||
'prefix': 'audit',
|
||
'backup_count': 30,
|
||
'formatter': 'verbose',
|
||
}
|
||
"""
|
||
|
||
def __init__(self, dir_path, prefix='audit', backup_count=30, encoding='utf-8'):
|
||
super().__init__()
|
||
self.dir_path = Path(dir_path)
|
||
self.dir_path.mkdir(parents=True, exist_ok=True)
|
||
self.prefix = prefix
|
||
self.backup_count = backup_count
|
||
self.encoding = encoding
|
||
self._current_date = None
|
||
self._stream = None
|
||
self._open_today()
|
||
|
||
def _today(self):
|
||
return time.strftime('%Y-%m-%d')
|
||
|
||
def _open_today(self):
|
||
"""打开当天的日志文件（追加模式）。"""
|
||
if self._stream:
|
||
try:
|
||
self._stream.close()
|
||
except OSError:
|
||
pass
|
||
self._current_date = self._today()
|
||
filepath = self.dir_path / f'{self.prefix}-{self._current_date}.log'
|
||
self._stream = open(filepath, 'a', encoding=self.encoding)
|
||
|
||
def emit(self, record):
|
||
try:
|
||
today = self._today()
|
||
if today != self._current_date:
|
||
self._open_today()
|
||
self._cleanup_old_files()
|
||
msg = self.format(record)
|
||
self._stream.write(msg + '\n')
|
||
self._stream.flush()
|
||
except Exception:
|
||
self.handleError(record)
|
||
|
||
def close(self):
|
||
self.acquire()
|
||
try:
|
||
if self._stream:
|
||
try:
|
||
self._stream.close()
|
||
except OSError:
|
||
pass
|
||
self._stream = None
|
||
finally:
|
||
self.release()
|
||
super().close()
|
||
|
||
def _cleanup_old_files(self):
|
||
"""删除超过 backup_count 天的旧日志文件。"""
|
||
if self.backup_count <= 0:
|
||
return
|
||
try:
|
||
files = sorted(self.dir_path.glob(f'{self.prefix}-*.log'))
|
||
for f in files[:-self.backup_count]:
|
||
f.unlink(missing_ok=True)
|
||
except OSError:
|
||
pass
|