Spaces:
Paused
Paused
File size: 1,810 Bytes
f35f208 8f95297 f35f208 8f95297 f35f208 8f95297 f35f208 |
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 59 60 61 62 |
import logging
import sys
from pathlib import Path
class StreamToLogger:
"""
Fake file-like stream object that redirects writes to a logger instance.
"""
def __init__(self, logger, log_level=logging.INFO):
self.logger = logger
self.log_level = log_level
self.linebuf = ''
self.enabled = True
def write(self, buf):
if self.enabled:
for line in buf.rstrip().splitlines():
self.logger.log(self.log_level, line.rstrip())
def flush(self):
pass
def enable(self):
self.enabled = True
def disable(self):
self.enabled = False
def setup_logger(config: dict, name: str = None) -> logging.Logger:
"""Set up logger with configuration from config file."""
logger = logging.getLogger(name or __name__)
# Set level from config
level = getattr(logging, config["logging"]["level"].upper())
logger.setLevel(level)
# Create logs directory if it doesn't exist
log_path = Path(config["folders"]["logs"])
log_path.mkdir(exist_ok=True)
# Create handlers
file_handler = logging.FileHandler(log_path / config["logging"]["file"])
console_handler = logging.StreamHandler()
# Create formatter
formatter = logging.Formatter(config["logging"]["format"])
file_handler.setFormatter(formatter)
console_handler.setFormatter(formatter)
# Add handlers
logger.addHandler(file_handler)
logger.addHandler(console_handler)
# Redirect stdout to logger
stream_to_logger = StreamToLogger(logger, logging.INFO)
sys.stdout = stream_to_logger
# Add methods to enable/disable StreamToLogger
logger.enable_stream_to_logger = stream_to_logger.enable
logger.disable_stream_to_logger = stream_to_logger.disable
return logger |