Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,316 Bytes
17cd746 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
#
# Toyota Motor Europe NV/SA and its affiliated companies retain all intellectual
# property and proprietary rights in and to this software and related documentation.
# Any commercial use, reproduction, disclosure or distribution of this software and
# related documentation without an express license agreement from Toyota Motor Europe NV/SA
# is strictly prohibited.
#
import logging
import sys
from datetime import datetime
import atexit
from pathlib import Path
def _colored(msg, color):
colors = {'red': '\033[91m', 'green': '\033[92m', 'yellow': '\033[93m', 'normal': '\033[0m'}
return colors[color] + msg + colors["normal"]
class ColorFormatter(logging.Formatter):
"""
Class to make command line log entries more appealing
Inspired by https://github.com/facebookresearch/detectron2
"""
def formatMessage(self, record):
"""
Print warnings yellow and errors red
:param record:
:return:
"""
log = super().formatMessage(record)
if record.levelno == logging.WARNING:
prefix = _colored("WARNING", "yellow")
elif record.levelno == logging.ERROR or record.levelno == logging.CRITICAL:
prefix = _colored("ERROR", "red")
else:
return log
return prefix + " " + log
def get_logger(name, level=logging.DEBUG, root=False, log_dir=None):
"""
Replaces the standard library logging.getLogger call in order to make some configuration
for all loggers.
:param name: pass the __name__ variable
:param level: the desired log level
:param root: call only once in the program
:param log_dir: if root is set to True, this defines the directory where a log file is going
to be created that contains all logging output
:return: the logger object
"""
logger = logging.getLogger(name)
logger.setLevel(level)
if root:
# create handler for console
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setLevel(level)
formatter = ColorFormatter(_colored("[%(asctime)s %(name)s]: ", "green") + "%(message)s",
datefmt="%m/%d %H:%M:%S")
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
logger.propagate = False # otherwise root logger prints things again
if log_dir is not None:
# add handler to log to a file
log_dir = Path(log_dir)
if not log_dir.exists():
logger.info(f"Logging directory {log_dir} does not exist and will be created")
log_dir.mkdir(parents=True)
timestamp = datetime.now().strftime("%d-%m-%Y_%H-%M-%S")
log_file = log_dir / f"{timestamp}.log"
# open stream and make sure it will be closed
stream = log_file.open(mode="w")
atexit.register(stream.close)
formatter = logging.Formatter("[%(asctime)s] %(name)s %(levelname)s: %(message)s",
datefmt="%m/%d %H:%M:%S")
file_handler = logging.StreamHandler(stream)
file_handler.setLevel(level)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
return logger
|