394 lines
14 KiB
Python
394 lines
14 KiB
Python
"""
|
|
서버/CLI 친화 Logger 모듈 - 파일/콘솔 분리 로깅과 자동 정리
|
|
|
|
권장 로그 레벨:
|
|
- 개발: logging.DEBUG
|
|
- 운영: logging.INFO 또는 logging.WARNING
|
|
|
|
사용 예시:
|
|
```python
|
|
# 기본 사용
|
|
logger = Logger(
|
|
log_file="logs/app.log",
|
|
file_log_level=logging.DEBUG
|
|
)
|
|
|
|
logger.info("서버 시작")
|
|
|
|
# FastAPI에서 사용 (예)
|
|
# from fastapi import FastAPI
|
|
# app = FastAPI()
|
|
# log = Logger(log_file="logs/api.log")
|
|
# @app.get("/")
|
|
# def root():
|
|
# log.info("요청 수신")
|
|
# return {"ok": True}
|
|
```
|
|
"""
|
|
import re
|
|
import logging
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
import os
|
|
import time
|
|
import threading
|
|
from datetime import datetime, timedelta
|
|
from pathlib import Path
|
|
import traceback
|
|
|
|
# 로그 레벨 이름 매핑 (안전하고 호환성 좋은 방법)
|
|
def get_level_name(level):
|
|
"""로그 레벨을 이름으로 변환 (호환성 보장)"""
|
|
level_names = {
|
|
logging.NOTSET: "NOTSET",
|
|
logging.DEBUG: "DEBUG",
|
|
logging.INFO: "INFO",
|
|
logging.WARNING: "WARNING",
|
|
logging.ERROR: "ERROR",
|
|
logging.CRITICAL: "CRITICAL"
|
|
}
|
|
return level_names.get(level, f"Level {level}")
|
|
|
|
class Logger:
|
|
|
|
def __init__(self, gui_logger=None, log_file="Edit_PartTimer_log.log", logger_name="Edit_PartTimer_log",
|
|
file_log_level=logging.DEBUG, gui_log_level=logging.INFO,
|
|
max_days=3, cleanup_interval=3600):
|
|
"""
|
|
개선된 Logger 초기화 (서버/CLI용)
|
|
:param gui_logger: 선택적 콜백 함수(메시지 수신용)
|
|
:param log_file: 로그 파일 이름
|
|
:param logger_name: 로거 이름
|
|
:param file_log_level: 파일 로거의 로그 레벨
|
|
:param gui_log_level: 콜백 호출 로그 레벨(기본 INFO)
|
|
:param max_days: 로그 파일 보관 일수 (기본 3일)
|
|
:param cleanup_interval: 정리 작업 간격(초, 기본 1시간)
|
|
"""
|
|
self.gui_logger = gui_logger
|
|
self.file_log_level = file_log_level
|
|
self.gui_log_level = gui_log_level
|
|
self.max_days = max_days
|
|
self.cleanup_interval = cleanup_interval
|
|
self.log_dir = Path(log_file).parent
|
|
self.log_base_name = Path(log_file).stem
|
|
|
|
# 로그 디렉토리 생성
|
|
self.log_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
# 로그 설정
|
|
self.logger = logging.getLogger(logger_name)
|
|
self.logger.setLevel(file_log_level)
|
|
# 상위 로거로의 전파 방지(중복 출력 방지)
|
|
self.logger.propagate = False
|
|
|
|
# 기존 핸들러 제거 (중복 방지)
|
|
for handler in self.logger.handlers[:]:
|
|
self.logger.removeHandler(handler)
|
|
|
|
# 포맷 설정
|
|
self.simple_format = "[%(asctime)s] [%(levelname)s] %(message)s"
|
|
self.detailed_format = (
|
|
"[%(asctime)s] [%(threadName)s] [%(levelname)s] "
|
|
"[%(filename)s:%(funcName)s:%(lineno)d] %(message)s"
|
|
)
|
|
# 콜백 전용 간결한 포맷 (시간:분:초 + 메시지만)
|
|
# 필요 시 log()에서 Formatter를 생성하여 사용
|
|
|
|
# 핸들러 추가
|
|
self._add_console_handler(file_log_level)
|
|
self._add_file_handler(log_file, file_log_level)
|
|
|
|
# 자동 정리 스레드 시작
|
|
self._start_cleanup_thread()
|
|
|
|
def _add_console_handler(self, level):
|
|
"""콘솔 핸들러 추가"""
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(level)
|
|
formatter = logging.Formatter(
|
|
self.detailed_format if level <= logging.DEBUG else self.simple_format
|
|
)
|
|
console_handler.setFormatter(formatter)
|
|
self.logger.addHandler(console_handler)
|
|
|
|
def _add_file_handler(self, log_file, level):
|
|
"""파일 크기 기반 로테이팅 + 수명 관리"""
|
|
from logging.handlers import RotatingFileHandler
|
|
# 확장자가 .log가 아니면 .log로 변경
|
|
if not log_file.endswith('.log'):
|
|
base_name, _ = os.path.splitext(log_file)
|
|
log_file = base_name + '.log'
|
|
|
|
# 크기 기반 로테이션: 10MB, 최대 50개(정리 스레드가 3일/일5개로 실제 보존 제한)
|
|
file_handler = RotatingFileHandler(
|
|
log_file,
|
|
maxBytes=10 * 1024 * 1024,
|
|
backupCount=50,
|
|
encoding="utf-8",
|
|
)
|
|
|
|
file_handler.setLevel(level)
|
|
formatter = logging.Formatter(
|
|
self.detailed_format if level <= logging.DEBUG else self.simple_format
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
self.logger.addHandler(file_handler)
|
|
|
|
# 핸들러 참조 저장 (정리 작업용)
|
|
self.file_handler = file_handler
|
|
|
|
def _start_cleanup_thread(self):
|
|
"""자동 정리 스레드 시작"""
|
|
def cleanup_worker():
|
|
while True:
|
|
try:
|
|
self._cleanup_old_logs()
|
|
time.sleep(self.cleanup_interval)
|
|
except Exception as e:
|
|
# 정리 작업 실패 시에도 계속 동작
|
|
pass
|
|
|
|
cleanup_thread = threading.Thread(target=cleanup_worker, daemon=True)
|
|
cleanup_thread.start()
|
|
|
|
def _cleanup_old_logs(self):
|
|
"""오래된 로그(> max_days) 및 날짜별 최대 10개 초과분 정리"""
|
|
cutoff_date = datetime.now() - timedelta(days=self.max_days)
|
|
log_pattern = f"{self.log_base_name}*.log*" # .log, .log.1 등 모두 포함
|
|
|
|
files = []
|
|
for log_file in self.log_dir.glob(log_pattern):
|
|
try:
|
|
if log_file.name == f"{self.log_base_name}.log":
|
|
continue
|
|
file_mtime = datetime.fromtimestamp(log_file.stat().st_mtime)
|
|
files.append((log_file, file_mtime))
|
|
except Exception:
|
|
continue
|
|
|
|
# 1) 보존 기간 초과 파일 삭제
|
|
for log_file, mtime in files:
|
|
if mtime < cutoff_date:
|
|
try:
|
|
log_file.unlink()
|
|
except Exception:
|
|
pass
|
|
|
|
# 2) 날짜별 최대 5개 유지 (최신순 보존)
|
|
from collections import defaultdict
|
|
by_day = defaultdict(list)
|
|
for log_file, mtime in files:
|
|
day_key = mtime.strftime('%Y-%m-%d')
|
|
by_day[day_key].append((log_file, mtime))
|
|
|
|
for day_key, items in by_day.items():
|
|
# 최신순으로 정렬
|
|
items.sort(key=lambda x: x[1], reverse=True)
|
|
for (log_file, _mtime) in items[5:]:
|
|
try:
|
|
if log_file.exists():
|
|
log_file.unlink()
|
|
except Exception:
|
|
pass
|
|
|
|
def log(self, message, level=logging.INFO, exc_info=False):
|
|
"""로그 메시지 기록"""
|
|
if exc_info:
|
|
message = f"{message}\n{traceback.format_exc()}"
|
|
|
|
# 호출 위치 정보를 동적으로 추출
|
|
caller_frame = logging.currentframe().f_back
|
|
record = self.logger.makeRecord(
|
|
self.logger.name, level, caller_frame.f_code.co_filename,
|
|
caller_frame.f_lineno, message, None, None, caller_frame.f_code.co_name
|
|
)
|
|
|
|
# 파일/콘솔 핸들러에 메시지 전달
|
|
if level >= self.file_log_level:
|
|
self.logger.handle(record)
|
|
|
|
# 선택적 콜백으로 전달 (간결 포맷 적용)
|
|
if self.gui_logger and level >= self.gui_log_level:
|
|
gui_formatter = logging.Formatter(
|
|
fmt="[%(asctime)s] %(message)s",
|
|
datefmt="%H:%M:%S"
|
|
)
|
|
formatted_message = gui_formatter.format(record)
|
|
try:
|
|
self.gui_logger(formatted_message)
|
|
except Exception:
|
|
# 콜백 오류는 로깅 실패로 간주하지 않음
|
|
pass
|
|
|
|
def set_gui_logger(self, gui_logger, gui_log_level=None):
|
|
"""
|
|
로그 콜백 함수를 설정합니다.
|
|
선택적으로 콜백 로그 레벨도 변경할 수 있습니다.
|
|
"""
|
|
self.gui_logger = gui_logger
|
|
|
|
if gui_log_level is not None:
|
|
self.gui_log_level = gui_log_level
|
|
|
|
def set_gui_log_level(self, level):
|
|
"""콜백 로그 레벨을 동적으로 변경합니다"""
|
|
self.gui_log_level = level
|
|
level_name = get_level_name(level)
|
|
self.logger.info(f"콜백 로그 레벨이 {level_name}로 변경되었습니다")
|
|
|
|
def set_file_log_level(self, level):
|
|
"""파일 로그 레벨을 동적으로 변경합니다"""
|
|
self.file_log_level = level
|
|
self.logger.setLevel(level)
|
|
for handler in self.logger.handlers:
|
|
if isinstance(handler, (logging.FileHandler, TimedRotatingFileHandler)):
|
|
handler.setLevel(level)
|
|
level_name = get_level_name(level)
|
|
self.logger.info(f"파일 로그 레벨이 {level_name}로 변경되었습니다")
|
|
|
|
def get_log_levels(self):
|
|
"""현재 로그 레벨 정보 반환"""
|
|
return {
|
|
"file_level": get_level_name(self.file_log_level),
|
|
"gui_level": get_level_name(self.gui_log_level),
|
|
"logger_level": get_level_name(self.logger.level)
|
|
}
|
|
|
|
def get_log_info(self):
|
|
"""로그 파일 정보 반환"""
|
|
try:
|
|
log_files = list(self.log_dir.glob(f"{self.log_base_name}*.log"))
|
|
total_size = sum(f.stat().st_size for f in log_files)
|
|
|
|
return {
|
|
"log_dir": str(self.log_dir),
|
|
"total_files": len(log_files),
|
|
"total_size_mb": total_size / (1024 * 1024),
|
|
"max_days": self.max_days,
|
|
"files": [f.name for f in log_files]
|
|
}
|
|
except Exception:
|
|
return {"error": "로그 정보 조회 실패"}
|
|
|
|
def force_cleanup(self):
|
|
"""수동 정리 실행"""
|
|
self._cleanup_old_logs()
|
|
|
|
# 파이썬 표준 로깅 인터페이스 지원
|
|
def debug(self, message, *args, **kwargs):
|
|
"""DEBUG 레벨 로그"""
|
|
if args:
|
|
message = message % args
|
|
self.log(message, level=logging.DEBUG, exc_info=kwargs.get('exc_info', False))
|
|
|
|
def info(self, message, *args, **kwargs):
|
|
"""INFO 레벨 로그"""
|
|
if args:
|
|
message = message % args
|
|
self.log(message, level=logging.INFO, exc_info=kwargs.get('exc_info', False))
|
|
|
|
def warning(self, message, *args, **kwargs):
|
|
"""WARNING 레벨 로그"""
|
|
if args:
|
|
message = message % args
|
|
self.log(message, level=logging.WARNING, exc_info=kwargs.get('exc_info', False))
|
|
|
|
def error(self, message, *args, **kwargs):
|
|
"""ERROR 레벨 로그"""
|
|
if args:
|
|
message = message % args
|
|
self.log(message, level=logging.ERROR, exc_info=kwargs.get('exc_info', False))
|
|
|
|
def critical(self, message, *args, **kwargs):
|
|
"""CRITICAL 레벨 로그"""
|
|
if args:
|
|
message = message % args
|
|
self.log(message, level=logging.CRITICAL, exc_info=kwargs.get('exc_info', False))
|
|
|
|
def exception(self, message, *args, **kwargs):
|
|
"""ERROR 레벨로 예외 정보와 함께 로그"""
|
|
if args:
|
|
message = message % args
|
|
self.log(message, level=logging.ERROR, exc_info=True)
|
|
|
|
# 전역 기본 로거 인스턴스
|
|
_default_logger = None
|
|
|
|
def get_default_logger():
|
|
"""기본 로거 인스턴스 반환"""
|
|
global _default_logger
|
|
if _default_logger is None:
|
|
_default_logger = Logger(log_file="logs/default.log")
|
|
return _default_logger
|
|
|
|
def set_default_logger(logger):
|
|
"""기본 로거 설정"""
|
|
global _default_logger
|
|
_default_logger = logger
|
|
|
|
# 편의 함수들 - 실제 구현
|
|
def debug(msg, *args, **kwargs):
|
|
"""편의 함수 - DEBUG 레벨 로그"""
|
|
get_default_logger().debug(msg, *args, **kwargs)
|
|
|
|
def info(msg, *args, **kwargs):
|
|
"""편의 함수 - INFO 레벨 로그"""
|
|
get_default_logger().info(msg, *args, **kwargs)
|
|
|
|
def warning(msg, *args, **kwargs):
|
|
"""편의 함수 - WARNING 레벨 로그"""
|
|
get_default_logger().warning(msg, *args, **kwargs)
|
|
|
|
def error(msg, *args, **kwargs):
|
|
"""편의 함수 - ERROR 레벨 로그"""
|
|
get_default_logger().error(msg, *args, **kwargs)
|
|
|
|
def critical(msg, *args, **kwargs):
|
|
"""편의 함수 - CRITICAL 레벨 로그"""
|
|
get_default_logger().critical(msg, *args, **kwargs)
|
|
|
|
def exception(msg, *args, **kwargs):
|
|
"""편의 함수 - 예외 정보와 함께 로그"""
|
|
get_default_logger().exception(msg, *args, **kwargs)
|
|
|
|
|
|
# 구조화된 로깅을 위한 추가 클래스
|
|
class StructuredLogger(Logger):
|
|
"""JSON 형태의 구조화된 로깅을 지원하는 로거"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def log_structured(self, event, level=logging.INFO, **context):
|
|
"""구조화된 로그 기록"""
|
|
import json
|
|
|
|
log_data = {
|
|
"timestamp": datetime.now().isoformat(),
|
|
"event": event,
|
|
"level": get_level_name(level),
|
|
**context
|
|
}
|
|
|
|
message = json.dumps(log_data, ensure_ascii=False, separators=(',', ':'))
|
|
self.log(message, level)
|
|
|
|
def log_performance(self, operation, duration, **context):
|
|
"""성능 로그 기록"""
|
|
self.log_structured(
|
|
"performance",
|
|
level=logging.INFO,
|
|
operation=operation,
|
|
duration_ms=round(duration * 1000, 2),
|
|
**context
|
|
)
|
|
|
|
def log_error_with_context(self, error, **context):
|
|
"""에러와 컨텍스트를 함께 로그"""
|
|
self.log_structured(
|
|
"error",
|
|
level=logging.ERROR,
|
|
error_type=type(error).__name__,
|
|
error_message=str(error),
|
|
**context
|
|
)
|