67 lines
1.5 KiB
Python
67 lines
1.5 KiB
Python
import logging
|
|
|
|
|
|
def valid_logger(func):
|
|
def wrapper(*args, **kwargs):
|
|
logger = args[0]
|
|
assert logger._was_set, \
|
|
"Logger is not set. Please call set_logger() first."
|
|
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
class Logger:
|
|
outfile = None
|
|
|
|
_logger = None
|
|
|
|
_was_set = False
|
|
|
|
def __init__(self, logger_name):
|
|
if self._logger is None:
|
|
self._logger = logging.getLogger(logger_name)
|
|
|
|
def set_logger(self, outfile='run.log', level=logging.INFO):
|
|
self.outfile = outfile
|
|
|
|
if level is not None:
|
|
self._logger.setLevel(level)
|
|
|
|
self._create_handlers()
|
|
|
|
self._was_set = True
|
|
|
|
@valid_logger
|
|
def info(self, message):
|
|
self._logger.info(message)
|
|
|
|
@valid_logger
|
|
def debug(self, message):
|
|
self._logger.debug(message)
|
|
|
|
@valid_logger
|
|
def warning(self, message):
|
|
self._logger.warning(message)
|
|
|
|
@valid_logger
|
|
def error(self, message):
|
|
self._logger.error(message)
|
|
|
|
def _create_handlers(self):
|
|
handlers = []
|
|
if self.outfile is not None:
|
|
handlers.append(logging.FileHandler(self.outfile, mode='a+'))
|
|
else:
|
|
handlers.append(logging.StreamHandler())
|
|
|
|
for handler in handlers:
|
|
handler.setFormatter(logging.Formatter('%(message)s'))
|
|
self._logger.addHandler(handler)
|
|
|
|
def close(self):
|
|
for handler in self._logger.handlers:
|
|
handler.close()
|
|
self._logger.removeHandler(handler)
|