def test_colourise_output_cyan(self):
with tempfile.TemporaryFile() as fd:
save = sys.stdout
sys.stdout = fd
assert_not_equal('foo', utils.colourise_output('cyan', 'foo'))
sys.stdout = save
def main():
"""
Reads command line arguments, calls into other modules for parsing,
validation and building output. Exits with appropriate return code.
:returns: boolean
"""
parser = argparse.ArgumentParser()
output_group = parser.add_mutually_exclusive_group()
output_group.add_argument("-v", "--verbose",
help="set loglevel to verbose",
default=False,
action="store_true")
output_group.add_argument("-d", "--debug",
help="set loglevel to debug",
default=False,
action="store_true")
output_group.add_argument("-q", "--quiet",
help="set loglevel to quiet",
default=False,
action="store_true")
parser.add_argument("-e", "--environment",
default='dev',
help="run config for this environment")
parser.add_argument("-p", "--profile",
default='default',
help="use config profile")
parser.add_argument("-P", "--policyfile",
default='/etc/pmcf/policy.json',
help="alternate policy file")
parser.add_argument("-c", "--configfile",
default='/etc/pmcf/pmcf.conf',
help="alternate config file")
parser.add_argument("-a", "--action",
default='create',
choices=['create', 'update', 'trigger', 'delete'],
help="action to take on stack")
parser.add_argument("--poll",
default=False,
action="store_true",
help="poll until completion")
parser.add_argument("stackfile",
help="path to stack (farm) definition file")
args = parser.parse_args()
# Log everything, and send it to stderr.
fmt = "[%(asctime)-15s] [%(levelname)s] %(name)s: %(message)s"
# Simple coloured output
logging.addLevelName(logging.DEBUG, colourise_output(
'green', " %s " % logging.getLevelName(logging.DEBUG)))
logging.addLevelName(logging.INFO, colourise_output(
'cyan', " %s " % logging.getLevelName(logging.INFO)))
logging.addLevelName(logging.WARNING, colourise_output(
'yellow', "%s" % logging.getLevelName(logging.WARNING)))
logging.addLevelName(logging.ERROR, colourise_output(
'red', " %s " % logging.getLevelName(logging.ERROR)))
if args.debug:
lvl = logging.DEBUG
elif args.verbose:
lvl = logging.INFO
elif args.quiet:
lvl = logging.ERROR
else:
lvl = logging.WARNING
logging.basicConfig(format=fmt, level=lvl)
logging.getLogger('boto').setLevel(logging.CRITICAL)
LOG = logging.getLogger(__name__)
try:
cfg = PMCFConfig(args.configfile, args.profile, args)
options = cfg.get_config()
cli = PMCFCLI(options)
return cli.run()
except PMCFException, exc:
LOG.error(exc.message)
return True
