import argparse from generate import generate if __name__=='__main__': parser = argparse.ArgumentParser(description="Generate gear orders", prog='PROG') subparsers = parser.add_subparsers(help="Sub-command help", dest="command") parser_immediate = subparsers.add_parser('immediate', help='Immediately generate a command') args = parser.parse_args() if args.command == 'immediate': orders_info = generate() if "count" in orders_info and orders_info['count'] > 1: print(f"These are the same orders from the last {orders_info['count']} days") print("\n".join(orders_info['orders']))