77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
'''
|
|
This module forwards everything from logging, with the addition of a level
|
|
LOUD=1 and all loggers are given the `loud` method.
|
|
'''
|
|
from logging import *
|
|
|
|
_getLogger = getLogger
|
|
|
|
LOUD = 1
|
|
|
|
def getLogger(name=None, main_fallback=None):
|
|
# Normally it is best practice to use getLogger(__name__), but when running
|
|
# a script directly you'll see "__main__" in the output, which I think is
|
|
# ugly and unexpected for anyone who doesn't know what's going on behind
|
|
# the scenes. But hardcoding your logger name is not good either.
|
|
# So, main_fallback is used to present your preferred name in case of main.
|
|
if name == '__main__' and main_fallback is not None:
|
|
name = main_fallback
|
|
log = _getLogger(name)
|
|
add_loud(log)
|
|
return log
|
|
|
|
def add_loud(log):
|
|
def loud(self, message, *args, **kwargs):
|
|
if self.isEnabledFor(LOUD):
|
|
self._log(LOUD, message, args, **kwargs)
|
|
|
|
addLevelName(LOUD, 'LOUD')
|
|
log.loud = loud.__get__(log, log.__class__)
|
|
|
|
def get_level_by_argv(argv):
|
|
'''
|
|
If any of the following arguments are present in argv, return the
|
|
corresponding log level along with a new copy of argv that has had the
|
|
argument string removed.
|
|
|
|
Since we are removing the argument, your argparser should not have options
|
|
with these same names.
|
|
|
|
--loud: LOUD
|
|
--debug: DEBUG
|
|
--quiet: ERROR
|
|
--silent: 99999999999
|
|
none of the above: INFO
|
|
'''
|
|
argv = argv[:]
|
|
|
|
if '--loud' in argv:
|
|
level = LOUD
|
|
argv.remove('--loud')
|
|
elif '--debug' in argv:
|
|
level = DEBUG
|
|
argv.remove('--debug')
|
|
elif '--quiet' in argv:
|
|
level = ERROR
|
|
argv.remove('--quiet')
|
|
elif '--silent' in argv:
|
|
level = 99999999999
|
|
argv.remove('--silent')
|
|
else:
|
|
level = INFO
|
|
|
|
return (level, argv)
|
|
|
|
def set_level_by_argv(log, argv):
|
|
'''
|
|
This function is helpful for single-file scripts where you instantiate the
|
|
logger in the global scope, and use this function to set its level
|
|
according to the "--debug" flags in argv, then pass the rest of argv to
|
|
your argparser.
|
|
'''
|
|
basicConfig()
|
|
|
|
(level, argv) = get_level_by_argv(argv)
|
|
log.setLevel(level)
|
|
|
|
return argv
|