Pull code out to separate function get_level_by_argv.

master
Ethan Dalool 2020-11-08 20:12:05 -08:00
parent 1468f318f8
commit e3d9336383
1 changed files with 13 additions and 6 deletions

View File

@ -13,20 +13,27 @@ def add_loud(log):
addLevelName(LOUD, 'LOUD') addLevelName(LOUD, 'LOUD')
log.loud = lambda *args, **kwargs: log.log(LOUD, *args, **kwargs) log.loud = lambda *args, **kwargs: log.log(LOUD, *args, **kwargs)
def set_level_by_argv(log, argv): def get_level_by_argv(argv):
basicConfig()
argv = argv[:] argv = argv[:]
if '--loud' in argv: if '--loud' in argv:
log.setLevel(LOUD) level = LOUD
argv.remove('--loud') argv.remove('--loud')
elif '--debug' in argv: elif '--debug' in argv:
log.setLevel(DEBUG) level = DEBUG
argv.remove('--debug') argv.remove('--debug')
elif '--quiet' in argv: elif '--quiet' in argv:
log.setLevel(ERROR) level = ERROR
argv.remove('--quiet') argv.remove('--quiet')
else: else:
log.setLevel(INFO) level = INFO
return (level, argv)
def set_level_by_argv(log, argv):
basicConfig()
(level, argv) = get_level_by_argv(argv)
log.setLevel(level)
return argv return argv