idds/tests/test_get_request_info.py

0001 from idds.common.constants import RequestStatus       # noqa F401
0002 from idds.common.utils import json_loads              # noqa F401
0003
0004 from lsst.ctrl.bps import BPS_DEFAULTS, BPS_SEARCH_ORDER, DEFAULT_MEM_FMT, DEFAULT_MEM_UNIT, BpsConfig    # noqa F401
0005 from lsst.ctrl.bps.panda.utils import (               # noqa F401
0006     add_final_idds_work,                              # noqa F401
0007     add_idds_work,                                    # noqa F401
0008     copy_files_for_distribution,                      # noqa F401
0009     get_idds_client,                                  # noqa F401
0010     get_idds_result,                                  # noqa F401
0011 )
0012
0013 default_config = BpsConfig(BPS_DEFAULTS)
0014
0015 idds_client = get_idds_client(default_config)
0016
0017 wms_workflow_id = 4112
0018 # only check the request status
0019 ret = idds_client.get_requests(request_id=wms_workflow_id)
0020 print(ret)
0021 # note: good to check the ret at first to make sure it's successful (see ctrl_bps_panda)
0022 print(ret[1][1][0]['status'])
0023
0024 # to show the status of different tasks
0025 ret = idds_client.get_requests(request_id=wms_workflow_id, with_detail=True)
0026 print(ret)
0027
0028 workloads = []
0029 for workload in ret[1][1]:
0030     workloads.append(workload['transform_workload_id'])
0031 print(workloads)
0032
0033 # show one workload file information
0034 workload_1 = workloads[0]
0035 ret = idds_client.get_contents_output_ext(request_id=wms_workflow_id, workload_id=workload_1)
0036 print(ret)