idds/tests/core_tests_dep_id.py

0001 import time                       # noqa F401
0002 from datetime import datetime     # noqa F401
0003
0004 from idds.common.utils import json_dumps, setup_logging                 # noqa F401
0005 from idds.common.constants import ContentStatus, ContentType, ContentRelationType, ContentLocking          # noqa F401
0006 from idds.core.requests import get_requests              # noqa F401
0007 from idds.core.messages import retrieve_messages         # noqa F401
0008 from idds.core.transforms import get_transforms, get_transform          # noqa F401
0009 from idds.core.workprogress import get_workprogresses    # noqa F401
0010 from idds.core.processings import get_processings        # noqa F401
0011 from idds.core import transforms as core_transforms      # noqa F401
0012 from idds.core.transforms import release_inputs_by_collection, release_inputs_by_collection_old     # noqa F401
0013 from idds.workflowv2.workflow import Workflow            # noqa F401
0014 from idds.workflowv2.work import Work                    # noqa F401
0015
0016 from idds.orm import contents as orm_contents       # noqa F401
0017 from idds.core import catalog as core_catalog       # noqa F401
0018
0019
0020 setup_logging(__name__)
0021
0022 request_id = 3347
0023 transform_id = 26788   # 3028, 3029
0024
0025 # ret = core_catalog.update_contents_to_others_by_dep_id(request_id=request_id, transform_id=transform_id)
0026 # print(ret)
0027
0028 # ret = core_catalog.update_contents_from_others_by_dep_id(request_id=request_id, transform_id=transform_id)
0029 # print(ret)
0030
0031 for req_id in [3350, 3407, 3414, 3420]:
0032     start = time.time()
0033     ret = core_catalog.get_updated_transforms_by_content_status(request_id=req_id)
0034     print('request_id: %s, time: %s' % (req_id, time.time() - start))
0035     print(ret)