idds/tests/trigger_release.py

0001
0002 from idds.common.utils import json_dumps                 # noqa F401
0003 from idds.common.constants import ContentStatus, ContentRelationType          # noqa F401
0004 from idds.core.requests import get_requests              # noqa F401
0005 from idds.core.messages import retrieve_messages         # noqa F401
0006 from idds.core.transforms import get_transforms, release_inputs          # noqa F401
0007 from idds.core.workprogress import get_workprogresses    # noqa F401
0008 from idds.core.processings import get_processings        # noqa F401
0009 from idds.core import transforms as core_transforms      # noqa F401
0010 from idds.core.catalog import get_contents, update_contents  # noqa F401
0011 from idds.orm.contents import get_input_contents             # noqa F401
0012
0013
0014 request_ids = [368, 369, 370, 371, 372, 373, 374, 375, 376]
0015 request_ids = [2895]
0016 for request_id in request_ids:
0017     contents = get_contents(request_id=request_id, status=ContentStatus.Available)
0018     ret_contents = {}
0019     for content in contents:
0020         if content['content_relation_type'] == ContentRelationType.Output:   # InputDependency
0021             if content['coll_id'] not in ret_contents:
0022                 ret_contents[content['coll_id']] = []
0023             ret_contents[content['coll_id']].append(content)
0024
0025     for ret_content in ret_contents:
0026         print("coll_id: %s, num_contents: %s" % (ret_content, len(ret_contents[ret_content])))
0027         # break
0028
0029     updated_contents = core_transforms.release_inputs_by_collection(ret_contents)
0030     for update_content in updated_contents:
0031         print(update_content)
0032         # break
0033
0034     update_contents(updated_contents)