diff --git a/main.py b/main.py index 72c6bf5..a214ea5 100644 --- a/main.py +++ b/main.py @@ -1,12 +1,20 @@ from hotprices_au import woolies, coles import argparse +import logging def main(): parser = argparse.ArgumentParser() parser.add_argument('--quick', action='store_true', default=False) + parser.add_argument('--debug', action='store_true', default=False) parser.add_argument('store', choices=['woolies', 'coles']) args = parser.parse_args() + + if args.debug: + logging.basicConfig(level=logging.DEBUG) + else: + logging.basicConfig(level=logging.INFO) + match args.store: case 'woolies': woolies.main(args.quick)