Change log helper to accept format

This commit is contained in:
Zeva Rose 2022-02-19 17:07:51 -07:00
parent f57d79c01a
commit 5ecab61fdb

View file

@ -2,11 +2,12 @@
import logging
from typing import Optional
NORMAL_FORMAT = "[%(asctime)s] [%(levelname)8s] %(message)s"
DEBUG_FORMAT = "[%(asctime)s] [%(levelname)8s] [%(filename)s:%(lineno)-3.3s] %(message)s"
FORMAT = "[%(asctime)s] [%(name)s] [%(levelname)8s] %(message)s"
handler = logging.StreamHandler()
def get_logger(name: str, fmt: Optional[str] = NORMAL_FORMAT) -> logging.Logger:
def get_logger(name: str, fmt: Optional[str] = FORMAT) -> logging.Logger:
"""
Get a pre-configured logger
@ -15,7 +16,6 @@ def get_logger(name: str, fmt: Optional[str] = NORMAL_FORMAT) -> logging.Logger:
"""
logger = logging.getLogger(name)
formatter = logging.Formatter(fmt)
handler = logging.StreamHandler()
handler.setFormatter(formatter)
logger.addHandler(handler)
return logger