62 lines
1.5 KiB
Python
62 lines
1.5 KiB
Python
"""Logging module"""
|
|
|
|
import logging
|
|
|
|
|
|
def get_logger(name, level=30):
|
|
"""Get logger"""
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(level)
|
|
|
|
return logger
|
|
|
|
|
|
def log_format(level):
|
|
if level > 10:
|
|
return '%(levelname)s:%(message)s'
|
|
|
|
return '%(levelname)s:%(name)s:%(message)s'
|
|
|
|
|
|
def set_formatter(handler, level):
|
|
"""create formatter and add it to the handlers"""
|
|
formatter = logging.Formatter(log_format(level))
|
|
handler.setFormatter(formatter)
|
|
|
|
|
|
def console(logger, level=30):
|
|
"""create console handler with a higher log level"""
|
|
logger.setLevel(level)
|
|
handler = logging.StreamHandler()
|
|
handler.setLevel(level)
|
|
set_formatter(handler, level)
|
|
|
|
# add the handlers to logger
|
|
logger.addHandler(handler)
|
|
|
|
|
|
def file_logger(logger, file, level=30):
|
|
"""create file handler that logs debug and higher level messages"""
|
|
logger.setLevel(level)
|
|
handler = logging.FileHandler(file)
|
|
handler.setLevel(level)
|
|
set_formatter(handler, log_format(level))
|
|
|
|
# add the handlers to logger
|
|
logger.addHandler(handler)
|
|
|
|
|
|
def get_level(quiet=False, verbose=False, debug=False, num=None):
|
|
"""Return int logging level from command line args"""
|
|
if num and num.isnumeric():
|
|
return int(verbose)
|
|
|
|
if debug:
|
|
return int(logging.getLevelName('DEBUG'))
|
|
if verbose:
|
|
return int(logging.getLevelName('INFO'))
|
|
if quiet:
|
|
return int(logging.getLevelName('ERROR'))
|
|
|
|
return int(logging.getLevelName('WARNING'))
|