| |
|
| | import sys |
| | import logging as log |
| | import os |
| |
|
| | from .Common import * |
| |
|
| | logLevelMap = { "trace" : log.DEBUG, |
| | "debug" : log.DEBUG, |
| | "info" : log.INFO, |
| | "warning" : log.WARNING, |
| | "error" : log.ERROR, |
| | } |
| |
|
| | class ExitOnExceptionHandler(log.StreamHandler): |
| | def emit(self, record): |
| | super().emit(record) |
| | if record.levelno == log.CRITICAL: |
| | sys.exit(-1) |
| |
|
| | def DefaultLogTimestamp(): |
| | log_format = os.environ.get('ROLL_LOG_TIMESTAMP') |
| | if log_format is not None and log_format != "": |
| | log_format = int(log_format) |
| | if log_format >= 0 and log_format <= 2: |
| | return log_format |
| | else: |
| | log.warn("Invalid value " + str(log_format) + " of environment variable ROLL_LOG_LEVEL") |
| |
|
| | return 0 |
| |
|
| | def DefaultLogLevel(): |
| |
|
| | log_level = os.environ.get('ROLL_LOG_LEVEL') |
| | if log_level is not None and log_level != "": |
| | if log_level in logLevelMap.keys(): |
| | return log_level |
| | else: |
| | log.warn("Invalid value " + log_level + " of environment variable ROLL_LOG_LEVEL") |
| |
|
| | if IsOsMac(): |
| | return "debug" |
| | else: |
| | return "info" |
| |
|
| | def LogTimestampHelp(): |
| | return "0: no timestamp, 1: [09:31AM], 2: [Sep 14 09:32:02]" |
| |
|
| | def LogLevelHelp(): |
| | return "Log level: debug, info, warning, error" |
| |
|
| | def InitLog(log_level, log_timestamp): |
| |
|
| | log.addLevelName(log.DEBUG, "DEBU") |
| | log.addLevelName(log.INFO, "INFO") |
| | log.addLevelName(log.WARNING, "WARN") |
| | log.addLevelName(log.ERROR, "ERRO") |
| | log.addLevelName(log.FATAL, "FATA") |
| |
|
| | |
| | if log_timestamp == 0: |
| | log.basicConfig(handlers=[ExitOnExceptionHandler()], level=logLevelMap[log_level], format='%(levelname)s %(message)s', datefmt='%I:%M%p') |
| | elif log_timestamp == 1: |
| | log.basicConfig(handlers=[ExitOnExceptionHandler()], level=logLevelMap[log_level], format='%(levelname)s[%(asctime)s] %(message)s', datefmt='%I:%M%p') |
| | else: |
| | log.basicConfig(handlers=[ExitOnExceptionHandler()], level=logLevelMap[log_level], format='%(levelname)s[%(asctime)s] %(message)s', datefmt='%b %d %H:%M:%S') |
| |
|
