pandajedi/jeditest/resubmitDataCarouselRequest.py

0001 import sys
0002
0003 from pandajedi.jedicore.JediTaskBufferInterface import JediTaskBufferInterface
0004 from pandajedi.jediddm.DDMInterface import DDMInterface
0005 from pandajedi.jedirefine import RefinerUtils
0006 from pandaserver.taskbuffer.DataCarousel import DataCarouselInterface
0007
0008 vo = "atlas"
0009 request_id = int(sys.argv[1])
0010
0011 print("set tbIF")
0012 tbIF = JediTaskBufferInterface()
0013 tbIF.setupInterface(max_size=1)
0014
0015 print("set ddmIF")
0016 ddmIF = DDMInterface()
0017 ddmIF.setupInterface()
0018
0019 print("set DCIF")
0020 data_carousel_interface = DataCarouselInterface(tbIF, ddmIF.getInterface(vo))
0021 if data_carousel_interface is None:
0022     # data carousel interface is undefined
0023     errStr = f"data carousel interface is undefined for vo={vo}"
0024     print(errStr)
0025     sys.exit(1)
0026
0027 print(f"resubmit for request_id={request_id}")
0028 dc_req_spec_resubmitted, _ = data_carousel_interface.resubmit_request(request_id)
0029
0030 if not dc_req_spec_resubmitted:
0031     # failed to resubmit
0032     print(f"failed; got {dc_req_spec_resubmitted}")
0033 else:
0034     # resubmitted
0035     print(f"resubmitted; new request_id={dc_req_spec_resubmitted.request_id}")