File size: 1,332 Bytes
0651b21 | 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 | from contextlib import contextmanager
import logging
import os
import sys
from loguru import logger
class HydraLoggerBridge(logging.Handler):
def emit(self, record):
# Get corresponding loguru level
try:
level = logger.level(record.levelname).name
except ValueError:
level = record.levelno
# Find caller from where the logged message originated
frame, depth = logging.currentframe(), 2
while frame and frame.f_code.co_filename == logging.__file__:
frame = frame.f_back
depth += 1
logger.opt(depth=depth, exception=record.exc_info).log(level, record.getMessage())
class LoguruStream:
def write(self, message):
if message.strip(): # Only log non-empty messages
logger.info(message.strip()) # Changed to debug level
def flush(self):
pass
@contextmanager
def capture_stdout_to_loguru():
logger.remove()
logger.add(sys.stdout, level="INFO")
loguru_stream = LoguruStream()
old_stdout = sys.stdout
sys.stdout = loguru_stream
try:
yield
finally:
sys.stdout = old_stdout
logger.remove()
console_log_level = os.environ.get("LOGURU_LEVEL", "INFO").upper()
logger.add(sys.stdout, level=console_log_level, colorize=True)
|