import argparse from src.clear_cid_delivery_history import clear_delivery_history_isrcs ISRC_MAX_NUM = 1000 def main(): print("Running script to clear cid delivery history") parser = argparse.ArgumentParser(prefix_chars="+") parser.add_argument("partner_id", type=int, help="Please enter the partner_id") parser.add_argument("isrc_list", nargs="+", help="Please enter the isrc(s) to be cleared") args = parser.parse_args() if len(args.isrc_list) > ISRC_MAX_NUM: parser.error(f"Total number of ISRCs must not be more than {ISRC_MAX_NUM}") clear_delivery_history_isrcs(args.partner_id, args.isrc_list) if __name__ == "__main__": main()