Update types for log adapter

log_task_context
Tyler Goodlet 2019-12-09 22:10:15 -05:00
parent cf73283586
commit 14bfef0df7
1 changed files with 11 additions and 6 deletions

View File

@ -48,7 +48,9 @@ BOLD_PALETTE = {
} }
def get_logger(name: str = None) -> logging.Logger: def get_logger(
name: str = None
) -> logging.LoggerAdapter:
'''Return the package log or a sub-log for `name` if provided. '''Return the package log or a sub-log for `name` if provided.
''' '''
log = rlog = logging.getLogger(_proj_name) log = rlog = logging.getLogger(_proj_name)
@ -58,18 +60,21 @@ def get_logger(name: str = None) -> logging.Logger:
# add our actor-task aware adapter which will dynamically look up # add our actor-task aware adapter which will dynamically look up
# the actor and task names at each log emit # the actor and task names at each log emit
log = logging.LoggerAdapter(log, ActorContextInfo()) logger = logging.LoggerAdapter(log, ActorContextInfo())
# additional levels # additional levels
for name, val in LEVELS.items(): for name, val in LEVELS.items():
logging.addLevelName(val, name) logging.addLevelName(val, name)
# ex. create ``log.trace()`` # ex. create ``logger.trace()``
setattr(log, name.lower(), partial(log.log, val)) setattr(logger, name.lower(), partial(logger.log, val))
return log return logger
def get_console_log(level: str = None, name: str = None) -> logging.Logger: def get_console_log(
level: str = None,
name: str = None
) -> logging.LoggerAdapter:
'''Get the package logger and enable a handler which writes to stderr. '''Get the package logger and enable a handler which writes to stderr.
Yeah yeah, i know we can use ``DictConfig``. You do it. Yeah yeah, i know we can use ``DictConfig``. You do it.