import logging from logging.handlers import SysLogHandler import sys class Logger: def __init__( self, name: str, debug: bool = False, write_to_syslog: bool = False, systemd: bool = False, do_test_logs: bool = True, ): self.__name = name self.__debug = debug self.__write_to_syslog = write_to_syslog self.__systemd = systemd self.__do_test_logs = do_test_logs self._init_logger() def _init_logger(self): self.__logger = logging.getLogger(self.__name) if self.__debug: level = logging.DEBUG else: level = logging.INFO self.__logger.setLevel(level) formatter = logging.Formatter( fmt="[{name}][{levelname:<7}] {message}", style='{' ) formatter_full = logging.Formatter( fmt="[{asctime}][{name}][{levelname:<7}] {message}", style='{' ) # Console output / stream handler (STDOUT) handler = logging.StreamHandler( stream=sys.stdout ) handler.setLevel(level) handler.addFilter(lambda entry: entry.levelno <= logging.INFO) handler.setFormatter( formatter if self.__systemd else formatter_full ) self.__logger.addHandler(handler) # Console output / stream handler (STDERR) handler = logging.StreamHandler( stream=sys.stderr ) handler.setLevel(logging.WARNING) handler.setFormatter( formatter if self.__systemd else formatter_full ) self.__logger.addHandler(handler) # Syslog handler if self.__write_to_syslog: handler = SysLogHandler( address="/dev/log" ) handler.setLevel(level) handler.setFormatter(formatter) self.__logger.addHandler(handler) # This is annoying inside cron if self.__do_test_logs: self.debug("Test debug log") self.info("Test info log") self.warn("Test warn log") self.error("Test error log") def debug(self, s): self.__logger.debug(s) def info(self, s): self.__logger.info(s) def warn(self, s): self.__logger.warning(s) def warning(self, s): self.__logger.warning(s) def error(self, s): self.__logger.error(s)