Move folders
This commit is contained in:
parent
ba2e252759
commit
3cf28379e8
89 changed files with 47 additions and 31 deletions
26
src/great_ai/utilities/logger/get_logger.py
Normal file
26
src/great_ai/utilities/logger/get_logger.py
Normal file
|
|
@ -0,0 +1,26 @@
|
|||
import logging
|
||||
from typing import Dict
|
||||
|
||||
from .custom_formatter import CustomFormatter
|
||||
|
||||
loggers: Dict[str, logging.Logger] = {}
|
||||
|
||||
|
||||
def get_logger(
|
||||
name: str, level: int = logging.INFO, disable_colors: bool = False
|
||||
) -> logging.Logger:
|
||||
if name not in loggers:
|
||||
logger = logging.getLogger(name)
|
||||
logger.setLevel(level)
|
||||
|
||||
fmt = "%(asctime)s | %(levelname)8s | %(message)s"
|
||||
|
||||
stdout_handler = logging.StreamHandler()
|
||||
stdout_handler.setLevel(level)
|
||||
if not disable_colors:
|
||||
stdout_handler.setFormatter(CustomFormatter(fmt))
|
||||
|
||||
logger.addHandler(stdout_handler)
|
||||
loggers[name] = logger
|
||||
|
||||
return loggers[name]
|
||||
Loading…
Add table
Add a link
Reference in a new issue