File size: 1,635 Bytes
fec9168 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
"""
Logging utilities for the temporal reasoning dataset pipeline.
"""
import logging
import sys
from pathlib import Path
from typing import Optional
def setup_logger(
name: str,
log_file: Optional[str] = None,
level: str = "INFO",
console_output: bool = True
) -> logging.Logger:
"""
Set up a logger with file and/or console handlers.
Args:
name: Name of the logger
log_file: Path to log file (if None, only console logging)
level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
console_output: Whether to output logs to console
Returns:
Configured logger instance
"""
logger = logging.getLogger(name)
logger.setLevel(getattr(logging, level.upper()))
# Remove existing handlers to avoid duplicates
logger.handlers = []
# Create formatter
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s',
datefmt='%Y-%m-%d %H:%M:%S'
)
# Console handler
if console_output:
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setLevel(getattr(logging, level.upper()))
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
# File handler
if log_file:
log_path = Path(log_file)
log_path.parent.mkdir(parents=True, exist_ok=True)
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(getattr(logging, level.upper()))
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
return logger
|