import argparse from src.clear_delivery_history import clear_delivery_history_upcs UPC_MAX_NUM = 1000 def main(): print("Running script to clear delivery history") parser = argparse.ArgumentParser(prefix_chars="+") parser.add_argument("partner_id", help="Please enter the partner_id", type=int) parser.add_argument("upc_list", nargs="+", help="Please enter the upc(s) to be cleared") args = parser.parse_args() if len(args.upc_list) > UPC_MAX_NUM: parser.error(f"Total number of UPCs should not be more than {UPC_MAX_NUM}") clear_delivery_history_upcs(args.partner_id, args.upc_list) if __name__ == "__main__": main()