diff --git a/run.py b/run.py index 492c784..31779b9 100644 --- a/run.py +++ b/run.py @@ -1,7 +1,18 @@ +import argparse import logging from src import main if __name__ == '__main__': - logging.basicConfig(format='[%(levelname)s] : %(message)s', level=logging.DEBUG) + parser = argparse.ArgumentParser() + parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true") + + + args = parser.parse_args() + + if args.verbose: + logging.basicConfig(format='[%(levelname)s] : %(message)s', level=logging.DEBUG) + else: + logging.basicConfig(format='[%(levelname)s] : %(message)s', level=logging.INFO) + main.run()