def __init__(self, stdout=None, stderr=None, no_color=False):
self.stdout = OutputWrapper(stdout or sys.stdout)
self.stderr = OutputWrapper(stderr or sys.stderr)
if no_color:
self.style = no_style()
else:
self.style = color_style()
self.stderr.style_func = self.style.ERROR
def __init__(self, app_config, stdout=None, stderr=None, style=None): super(BaseAction, self).__init__() # Sets app_config self.app = app_config # Sets stdout and stderr, which were supposed to passed from command self.stdout = stdout or sys.stdout self.stderr = stderr or sys.stderr self.style = style or color_style() # Imports stats class self.stats = import_string(self.stats_class)(self) # String to record and display after all works done self.output = []
def main_help_text(self, commands_only=False):
"""
Returns the script's main help text, as a string.
"""
if commands_only:
usage = sorted(get_commands().keys())
else:
usage = [
"",
"Type '%s help <subcommand>' for help on a specific subcommand."
% self.prog_name,
"",
"Available subcommands:",
]
commands_dict = defaultdict(lambda: [])
for name, app in get_commands().items():
if app == 'moose.core':
app = 'moose'
else:
app = app.rpartition('.')[-1]
commands_dict[app].append(name)
style = color_style()
for app in sorted(commands_dict.keys()):
usage.append("")
usage.append(style.NOTICE("[%s]" % app))
for name in sorted(commands_dict[app]):
usage.append(" %s" % name)
# Output an extra note if settings are not properly configured
if self.settings_exception is not None:
usage.append(
style.NOTICE(
"Note that only Moose core commands are listed "
"as settings are not properly configured (error: %s)."
% self.settings_exception))
return '\n'.join(usage)
def __init__(self, out, style_func=None, ending='\n'):
self._out = out
self.style_func = style_func
self.ending = ending
self.style = color_style()
def __init__(self, *args, **kwargs):
self.style = color_style()
super(ConsoleFormatter, self).__init__(*args, **kwargs)
# -*- coding: utf-8 -*-
# A standard output wrapper, defines whether (according to the verbose level)
# and what style (according to colors defined) uses to output
import sys
from moose.utils.encoding import force_str
from moose.core.management.color import color_style
from moose.core.exceptions import ImproperlyConfigured
style = color_style()
class OutputWrapper(object):
@property
def verbose(self):
return self._verbose
@verbose.setter
def verbose(self, verbose):
if verbose in [0, 1, 2, 3]:
self._verbose = verbose
elif verbose == None:
self._verbose = 1
else:
raise ImproperlyConfigured(
"Verbose level was ought to be set from "
"0 to 3, but got %s." % verbose)
@property
def style_func(self):