Skip to content
Snippets Groups Projects
Commit 84eca126 authored by James Graham's avatar James Graham
Browse files

feat: add secret flags for output formatting

parent 28ad67f3
No related branches found
No related tags found
No related merge requests found
...@@ -228,6 +228,14 @@ def parse_arguments(arg_list): ...@@ -228,6 +228,14 @@ def parse_arguments(arg_list):
run_options.add_argument('--log-level', default='INFO', run_options.add_argument('--log-level', default='INFO',
choices=('DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'), choices=('DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'),
help="Which log messages should be shown?") help="Which log messages should be shown?")
# Secret options
secret_options = parser.add_argument_group("secret options")
secret_options.add_argument('--curmudgeon', default=False, action='store_true',
help=argparse.SUPPRESS)
secret_options.add_argument('--cow', default=False, action='store_true',
help=argparse.SUPPRESS)
# yapf: enable # yapf: enable
args = parser.parse_args(arg_list) args = parser.parse_args(arg_list)
...@@ -267,25 +275,39 @@ def main(): ...@@ -267,25 +275,39 @@ def main():
start_time = time.time() start_time = time.time()
args = parse_arguments(sys.argv[1:]) args = parse_arguments(sys.argv[1:])
logging.basicConfig(level=args.log_level, if args.curmudgeon:
format='%(message)s', logging.basicConfig(level=args.log_level)
datefmt='[%X]',
handlers=[RichHandler(rich_tracebacks=True)]) else:
logging.basicConfig(level=args.log_level,
banner = r""" format='%(message)s',
_____ _____ _____ _______ ____ ____ _ datefmt='[%X]',
| __ \ / ____/ ____|__ __/ __ \ / __ \| | handlers=[RichHandler(rich_tracebacks=True)])
| |__) | _| | | | __ | | | | | | | | | |
| ___/ | | | | | | |_ | | | | | | | | | | | banner = r"""
| | | |_| | |___| |__| | | | | |__| | |__| | |____ _____ _____ _____ _______ ____ ____ _
|_| \__, |\_____\_____| |_| \____/ \____/|______| | __ \ / ____/ ____|__ __/ __ \ / __ \| |
__/ | | |__) | _| | | | __ | | | | | | | | | |
|___/ | ___/ | | | | | | |_ | | | | | | | | | | |
""" # noqa | | | |_| | |___| |__| | | | | |__| | |__| | |____
|_| \__, |\_____\_____| |_| \____/ \____/|______|
logger.info('[bold blue]%s[/]', __/ |
textwrap.dedent(banner), |___/
extra={'markup': True}) """ # noqa
banner = textwrap.dedent(banner)
if args.cow:
try:
import cowsay
banner = cowsay.cow('PyCGTOOL')
except ImportError:
pass
else:
logger.info('[bold blue]%s[/]',
textwrap.dedent(banner),
extra={'markup': True})
logger.info(30 * '-') logger.info(30 * '-')
logger.info('Topology:\t%s', args.topology) logger.info('Topology:\t%s', args.topology)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment