andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Make colors optional

Browse source
This commit is contained in:
Andras Schmelczer 2022-06-28 18:27:22 +02:00
parent d9e862af6b
commit 1556106cf0
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/great_ai/utilities/logger/get_logger.py
View file

@ -6,7 +6,9 @@ from .custom_formatter import CustomFormatter
loggers: Dict[str, logging.Logger] = {}
def get_logger(name: str, level: int = logging.INFO) -> 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)
@ -14,7 +16,8 @@ def get_logger(name: str, level: int = logging.INFO) -> logging.Logger:
fmt = "%(asctime)s | %(levelname)8s | %(message)s"
stdout_handler = logging.StreamHandler()
stdout_handler.setLevel(logging.DEBUG)
stdout_handler.setLevel(level)
if not disable_colors:
stdout_handler.setFormatter(CustomFormatter(fmt))
logger.addHandler(stdout_handler)