idds/tests/test_get_request_info_panda.py

0001 import json
0002 # from idds.common.constants import RequestStatus       # noqa F401
0003 # from idds.common.utils import json_loads              # noqa F401
0004
0005 import idds.common.utils as idds_utils                # noqa F401
0006 import pandaclient.idds_api                           # noqa F401
0007
0008
0009 idds_client = pandaclient.idds_api.get_api(idds_utils.json_dumps, idds_host=None, compress=True, manager=True)
0010
0011 # wms_workflow_id = 4112
0012 wms_workflow_id = 2154
0013 # only check the request status
0014 ret = idds_client.get_requests(request_id=wms_workflow_id)
0015 print(ret)
0016 # note: good to check the ret at first to make sure it's successful (see ctrl_bps_panda)
0017 print(ret[1][1][0]['status'])
0018
0019 # to show the status of different tasks
0020 ret = idds_client.get_requests(request_id=wms_workflow_id, with_detail=True)
0021 print(ret)
0022
0023 workloads = []
0024 transforms = []
0025 for workload in ret[1][1]:
0026     workloads.append(workload['transform_workload_id'])
0027     transforms.append(workload['transform_id'])
0028 print(workloads)
0029 print(transforms)
0030
0031 for transform_id in transforms:
0032     ret = idds_client.get_transform(request_id=wms_workflow_id, transform_id=transform_id)
0033     print(json.dumps(ret, indent=4, sort_keys=True))
0034
0035 # show one workload file information
0036 for workload_id in workloads:
0037     ret = idds_client.get_contents_output_ext(request_id=wms_workflow_id, workload_id=workload_id)
0038     print(ret)