From 8875fd09fb3c85d35a7b7b4cdc9981638566e6fe Mon Sep 17 00:00:00 2001 From: Wen Guan Date: Wed, 28 Aug 2024 12:04:43 +0200 Subject: [PATCH] add site in transforms --- main/lib/idds/core/transforms.py | 6 ++++-- main/lib/idds/orm/transforms.py | 8 ++++---- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/main/lib/idds/core/transforms.py b/main/lib/idds/core/transforms.py index e814d719..1a2d8e23 100644 --- a/main/lib/idds/core/transforms.py +++ b/main/lib/idds/core/transforms.py @@ -36,7 +36,7 @@ def add_transform(request_id, workload_id, transform_type, transform_tag=None, p parent_transform_id=None, previous_transform_id=None, current_processing_id=None, internal_id=None, has_previous_conditions=None, loop_index=None, cloned_from=None, triggered_conditions=None, untriggered_conditions=None, - workprogress_id=None, session=None): + site=None, workprogress_id=None, session=None): """ Add a transform. @@ -68,7 +68,9 @@ def add_transform(request_id, workload_id, transform_type, transform_tag=None, p parent_transform_id=parent_transform_id, previous_transform_id=previous_transform_id, current_processing_id=current_processing_id, - expired_at=expired_at, transform_metadata=transform_metadata, + expired_at=expired_at, + transform_metadata=transform_metadata, + site=site, internal_id=internal_id, has_previous_conditions=has_previous_conditions, loop_index=loop_index, cloned_from=cloned_from, diff --git a/main/lib/idds/orm/transforms.py b/main/lib/idds/orm/transforms.py index 995dc8e9..52c89c95 100644 --- a/main/lib/idds/orm/transforms.py +++ b/main/lib/idds/orm/transforms.py @@ -34,7 +34,7 @@ def create_transform(request_id, workload_id, transform_type, transform_tag=None parent_transform_id=None, previous_transform_id=None, current_processing_id=None, internal_id=None, has_previous_conditions=None, loop_index=None, cloned_from=None, triggered_conditions=None, untriggered_conditions=None, - retries=0, expired_at=None, transform_metadata=None): + site=None, retries=0, expired_at=None, transform_metadata=None): """ Create a transform. @@ -60,7 +60,7 @@ def create_transform(request_id, workload_id, transform_type, transform_tag=None parent_transform_id=parent_transform_id, previous_transform_id=previous_transform_id, current_processing_id=current_processing_id, - internal_id=internal_id, + internal_id=internal_id, site=site, has_previous_conditions=has_previous_conditions, loop_index=loop_index, cloned_from=cloned_from, triggered_conditions=triggered_conditions, @@ -83,7 +83,7 @@ def add_transform(request_id, workload_id, transform_type, transform_tag=None, p parent_transform_id=None, previous_transform_id=None, current_processing_id=None, internal_id=None, has_previous_conditions=None, loop_index=None, cloned_from=None, triggered_conditions=None, untriggered_conditions=None, - transform_metadata=None, workprogress_id=None, session=None): + transform_metadata=None, workprogress_id=None, site=None, session=None): """ Add a transform. @@ -115,7 +115,7 @@ def add_transform(request_id, workload_id, transform_type, transform_tag=None, p parent_transform_id=parent_transform_id, previous_transform_id=previous_transform_id, current_processing_id=current_processing_id, - internal_id=internal_id, + internal_id=internal_id, site=site, has_previous_conditions=has_previous_conditions, loop_index=loop_index, cloned_from=cloned_from, triggered_conditions=triggered_conditions,