Merge pull request #336 from HSF/dev

add site in transforms
HSF · Aug 28, 2024 · 48ab214 · 48ab214
2 parents d34447f + e133e98
commit 48ab214
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git 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
@@ -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,