strepsiades
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
|
|
class _NodeLogFilter(logging.Filter):
|
|
def filter(self, record: logging.LogRecord) -> bool:
|
|
message = record.getMessage()
|
|
if "node=" in message:
|
|
return False
|
|
if "节点 " in message and "状态" in message:
|
|
return False
|
|
return True
|
|
|
|
|
|
def init_pipeline_logger(
|
|
*,
|
|
log_file_path: Optional[str] = None,
|
|
level: int = logging.INFO,
|
|
mirror_console: bool = True,
|
|
replace_handlers: bool = True,
|
|
suppress_node_logs: bool = True,
|
|
) -> logging.Logger:
|
|
logger = logging.getLogger("sync_state_machine")
|
|
logger.setLevel(level)
|
|
logger.propagate = False
|
|
|
|
if replace_handlers:
|
|
clear_pipeline_logger_handlers(logger)
|
|
|
|
formatter = logging.Formatter("%(message)s")
|
|
node_filter = _NodeLogFilter() if suppress_node_logs else None
|
|
|
|
if mirror_console:
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(level)
|
|
console_handler.setFormatter(formatter)
|
|
if node_filter is not None:
|
|
console_handler.addFilter(node_filter)
|
|
logger.addHandler(console_handler)
|
|
|
|
if log_file_path:
|
|
path = Path(log_file_path)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
file_handler = logging.FileHandler(path, mode="w", encoding="utf-8")
|
|
file_handler.setLevel(level)
|
|
file_handler.setFormatter(formatter)
|
|
if node_filter is not None:
|
|
file_handler.addFilter(node_filter)
|
|
logger.addHandler(file_handler)
|
|
|
|
return logger
|
|
|
|
|
|
def clear_pipeline_logger_handlers(logger: Optional[logging.Logger] = None) -> None:
|
|
target = logger or logging.getLogger("sync_state_machine")
|
|
handlers = list(target.handlers)
|
|
for handler in handlers:
|
|
target.removeHandler(handler)
|
|
try:
|
|
handler.close()
|
|
except Exception:
|
|
pass
|