50 lines
969 B
Python
50 lines
969 B
Python
from __future__ import print_function
|
|
|
|
import progressbar
|
|
import sys
|
|
|
|
from .flags import FLAGS
|
|
|
|
|
|
# Logging functions
|
|
# =================
|
|
|
|
def prefix_print(prefix, message):
|
|
print(prefix + ('\n' + prefix).join(message.split('\n')))
|
|
|
|
|
|
def log_debug(message):
|
|
if FLAGS.log_level == 0:
|
|
prefix_print('D ', message)
|
|
|
|
|
|
def log_info(message):
|
|
if FLAGS.log_level <= 1:
|
|
prefix_print('I ', message)
|
|
|
|
|
|
def log_warn(message):
|
|
if FLAGS.log_level <= 2:
|
|
prefix_print('W ', message)
|
|
|
|
|
|
def log_error(message):
|
|
if FLAGS.log_level <= 3:
|
|
prefix_print('E ', message)
|
|
|
|
|
|
def create_progressbar(*args, **kwargs):
|
|
# Progress bars in stdout by default
|
|
if 'fd' not in kwargs:
|
|
kwargs['fd'] = sys.stdout
|
|
|
|
if FLAGS.show_progressbar:
|
|
return progressbar.ProgressBar(*args, **kwargs)
|
|
|
|
return progressbar.NullBar(*args, **kwargs)
|
|
|
|
|
|
def log_progress(message):
|
|
if not FLAGS.show_progressbar:
|
|
log_info(message)
|