From 3cd2d5d5988c807837825fbd7b7ab79192d2d831 Mon Sep 17 00:00:00 2001 From: Ryan Deivert Date: Thu, 12 Mar 2020 17:29:06 -0700 Subject: [PATCH] changing logger code slightly --- streamalert/shared/logger.py | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/streamalert/shared/logger.py b/streamalert/shared/logger.py index 8379ccd4b..2c91bcc35 100644 --- a/streamalert/shared/logger.py +++ b/streamalert/shared/logger.py @@ -18,6 +18,8 @@ LOCAL_LOGGER_FMT = '[%(levelname)s %(asctime)s (%(name)s:%(lineno)d)]: %(message)s' +logging.basicConfig(level=logging.INFO, format=LOCAL_LOGGER_FMT) + class LogFormatter(logging.Formatter): @@ -41,20 +43,15 @@ def set_formatter(logger): Args: logger (logging.Logger): An instance of a logger for which to update the formatter """ - # Update the LambdaLoggerHandler formatter - if logger.hasHandlers(): - for handler in logger.handlers + logger.parent.handlers: - # pylint: disable=protected-access - # Retain the handlers format spec if it has one - fmt = handler.formatter._fmt if handler.formatter else None - handler.setFormatter(LogFormatter(fmt=fmt)) + # Update the LambdaLoggerHandler formatter if there is one + if not logger.hasHandlers(): return - # Otherwise, create a handler with the desired formatter - formatter = LogFormatter(fmt=LOCAL_LOGGER_FMT) - handler = logging.StreamHandler() - handler.setFormatter(formatter) - logger.addHandler(handler) + for handler in logger.handlers + logger.parent.handlers: + # pylint: disable=protected-access + # Retain the handlers format spec if it has one + fmt = handler.formatter._fmt if handler.formatter else None + handler.setFormatter(LogFormatter(fmt=fmt)) def get_logger(name, level=None):