Add main_fallback name to vlogging.getLogger.
This commit is contained in:
parent
80d66c4117
commit
c1703357e8
1 changed files with 9 additions and 2 deletions
|
@ -8,8 +8,15 @@ _getLogger = getLogger
|
||||||
|
|
||||||
LOUD = 1
|
LOUD = 1
|
||||||
|
|
||||||
def getLogger(*args, **kwargs):
|
def getLogger(name=None, main_fallback=None):
|
||||||
log = _getLogger(*args, **kwargs)
|
# 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)
|
add_loud(log)
|
||||||
return log
|
return log
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue