Add flexibility to log_handler.

This commit is contained in:
Paul Kronenwetter 2022-05-31 20:34:22 -04:00
parent 2be89b5b20
commit b44c60fa32

View file

@ -4,13 +4,13 @@ import structlog
# https://www.structlog.org/en/stable/standard-library.html # https://www.structlog.org/en/stable/standard-library.html
def setup_logging(filename): def setup_logging(filename: str = "", level: str = "DEBUG"):
""" """
Args: Args:
filename: filename:
level:str: Log level to output, possible values are:
Returns: "CRITICAL", "FATAL", "ERROR", "WARNING", "WARN", "INFO", "DEBUG"
""" """
@ -40,12 +40,12 @@ def setup_logging(filename):
}, },
"handlers": { "handlers": {
"default": { "default": {
"level": "DEBUG", "level": level,
"class": "logging.StreamHandler", "class": "logging.StreamHandler",
"formatter": "colored", "formatter": "colored",
}, },
"file": { "file": {
"level": "DEBUG", "level": level,
"class": "logging.handlers.WatchedFileHandler", "class": "logging.handlers.WatchedFileHandler",
"filename": f"{filename}.log", "filename": f"{filename}.log",
"formatter": "plain", "formatter": "plain",
@ -53,8 +53,8 @@ def setup_logging(filename):
}, },
"loggers": { "loggers": {
"": { "": {
"handlers": ["default", "file"], "handlers": ["default", "file"] if filename else ["default"],
"level": "DEBUG", "level": level,
"propagate": True, "propagate": True,
}, },
}, },