12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394 |
- # Colored log, requires Python 2.3 or up.
- from __future__ import division, absolute_import, print_function
- import sys
- from distutils.log import *
- from distutils.log import Log as old_Log
- from distutils.log import _global_log
- if sys.version_info[0] < 3:
- from .misc_util import (red_text, default_text, cyan_text, green_text,
- is_sequence, is_string)
- else:
- from numpy.distutils.misc_util import (red_text, default_text, cyan_text,
- green_text, is_sequence, is_string)
- def _fix_args(args,flag=1):
- if is_string(args):
- return args.replace('%', '%%')
- if flag and is_sequence(args):
- return tuple([_fix_args(a, flag=0) for a in args])
- return args
- class Log(old_Log):
- def _log(self, level, msg, args):
- if level >= self.threshold:
- if args:
- msg = msg % _fix_args(args)
- if 0:
- if msg.startswith('copying ') and msg.find(' -> ') != -1:
- return
- if msg.startswith('byte-compiling '):
- return
- print(_global_color_map[level](msg))
- sys.stdout.flush()
- def good(self, msg, *args):
- """
- If we log WARN messages, log this message as a 'nice' anti-warn
- message.
- """
- if WARN >= self.threshold:
- if args:
- print(green_text(msg % _fix_args(args)))
- else:
- print(green_text(msg))
- sys.stdout.flush()
- _global_log.__class__ = Log
- good = _global_log.good
- def set_threshold(level, force=False):
- prev_level = _global_log.threshold
- if prev_level > DEBUG or force:
- # If we're running at DEBUG, don't change the threshold, as there's
- # likely a good reason why we're running at this level.
- _global_log.threshold = level
- if level <= DEBUG:
- info('set_threshold: setting threshold to DEBUG level,'
- ' it can be changed only with force argument')
- else:
- info('set_threshold: not changing threshold from DEBUG level'
- ' %s to %s' % (prev_level, level))
- return prev_level
- def set_verbosity(v, force=False):
- prev_level = _global_log.threshold
- if v < 0:
- set_threshold(ERROR, force)
- elif v == 0:
- set_threshold(WARN, force)
- elif v == 1:
- set_threshold(INFO, force)
- elif v >= 2:
- set_threshold(DEBUG, force)
- return {FATAL:-2,ERROR:-1,WARN:0,INFO:1,DEBUG:2}.get(prev_level, 1)
- _global_color_map = {
- DEBUG:cyan_text,
- INFO:default_text,
- WARN:red_text,
- ERROR:red_text,
- FATAL:red_text
- }
- # don't use INFO,.. flags in set_verbosity, these flags are for set_threshold.
- set_verbosity(0, force=True)
|