Skip to content

Commit

Permalink
fix(logging): fix success level message
Browse files Browse the repository at this point in the history
  • Loading branch information
Xiong Ma committed May 6, 2020
1 parent 0d8d892 commit bbbb9e7
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions jina/logging/base.py
Expand Up @@ -112,7 +112,7 @@ def info(self, msg: str, **kwargs):
sys.stdout.write('I:%s:%s' % (self.context, self._planify(msg)))

def critical(self, msg: str, **kwargs):
"""log info-level message"""
"""log critical-level message"""
if self.log_level <= LogVerbosity.CRITICAL:
sys.stdout.write('C:%s:%s' % (self.context, self._planify(msg)))

Expand All @@ -132,7 +132,7 @@ def warning(self, msg: str, **kwargs):
sys.stdout.write('W:%s:%s' % (self.context, self._planify(msg)))

def success(self, msg: str, **kwargs):
"""log warn-level message"""
"""log success-level message"""
if self.log_level <= LogVerbosity.SUCCESS:
sys.stdout.write('W:%s:%s' % (self.context, self._planify(msg)))

Expand Down Expand Up @@ -224,8 +224,8 @@ def get_logger(context: str, context_len: int = 15,
console_handler.setFormatter(ColorFormatter(fmt_str))
logger.addHandler(console_handler)

logging.SUCCESS = 25 # between WARNING and INFO
logging.addLevelName(logging.SUCCESS, 'SUCCESS')
setattr(logger, 'success', lambda message, *args: logger._log(logging.SUCCESS, message, args))
success_level = 25 # between WARNING and INFO
logging.addLevelName(success_level, 'SUCCESS')
setattr(logger, 'success', lambda message: logger.log(success_level, message))

return logger

0 comments on commit bbbb9e7

Please sign in to comment.