deltafi 2.0rc1722614194270__py3-none-any.whl → 2.0rc1723500522958__py3-none-any.whl

deltafi/action.py

@@ -27,6 +27,16 @@ from deltafi.result import *
 from pydantic import BaseModel
 
 
+class Join(ABC):
+    def join(self, transform_inputs: List[TransformInput]):
+        all_content = []
+        all_metadata = {}
+        for transform_input in transform_inputs:
+            all_content += transform_input.content
+            all_metadata.update(transform_input.metadata)
+        return TransformInput(content=all_content, metadata=all_metadata)
+
+
 class Action(ABC):
     def __init__(self, action_type: ActionType, description: str, valid_result_types: tuple):
         self.action_type = action_type
@@ -38,7 +48,7 @@ class Action(ABC):
     def build_input(self, context: Context, delta_file_message: DeltaFileMessage):
         pass
 
-    def join(self, action_inputs: List[Any]):
+    def execute_join_action(self, event):
         raise RuntimeError(f"Join is not supported for {self.__class__.__name__}")
 
     @abstractmethod
@@ -49,11 +59,7 @@ class Action(ABC):
         if event.delta_file_messages is None or not len(event.delta_file_messages):
             raise RuntimeError(f"Received event with no delta file messages for did {event.context.did}")
         if event.context.join is not None:
-            result = self.execute(
-                event.context,
-                self.join([self.build_input(event.context, delta_file_message)
-                              for delta_file_message in event.delta_file_messages]),
-                self.param_class().model_validate(event.params))
+            result = self.execute_join_action(event)
         else:
             result = self.execute(
                 event.context,
@@ -123,13 +129,15 @@ class TransformAction(Action, ABC):
     def build_input(self, context: Context, delta_file_message: DeltaFileMessage):
         return TransformInput(content=delta_file_message.content_list, metadata=delta_file_message.metadata)
 
-    def join(self, transform_inputs: List[TransformInput]):
-        all_content = []
-        all_metadata = {}
-        for transform_input in transform_inputs:
-            all_content += transform_input.content
-            all_metadata.update(transform_input.metadata)
-        return TransformInput(content=all_content, metadata=all_metadata)
+    def execute_join_action(self, event):
+        if (isinstance(self, Join)):
+            return self.execute(
+                event.context,
+                self.join([self.build_input(event.context, delta_file_message)
+                           for delta_file_message in event.delta_file_messages]),
+                self.param_class().model_validate(event.params))
+        else:
+            super().execute_join_action(event)
 
     @abstractmethod
     def transform(self, context: Context, params: BaseModel, transform_input: TransformInput):

deltafi/plugin.py

@@ -38,7 +38,7 @@ from deltafi.exception import ExpectedContentException, MissingMetadataException
 from deltafi.logger import get_logger
 from deltafi.result import ErrorResult
 from deltafi.storage import ContentService
-from deltafi.action import Action
+from deltafi.action import Action, Join
 
 
 def _coordinates():
@@ -164,6 +164,7 @@ class Plugin(object):
             'name': self.action_name(action),
             'description': action.description,
             'type': action.action_type.name,
+            'supportsJoin': isinstance(action, Join),
             'schema': action.param_class().model_json_schema()
         }
 

{deltafi-2.0rc1722614194270.dist-info → deltafi-2.0rc1723500522958.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: deltafi
-Version: 2.0rc1722614194270
+Version: 2.0rc1723500522958
 Summary: SDK for DeltaFi plugins and actions
 License: Apache License, Version 2.0
 Keywords: deltafi

{deltafi-2.0rc1722614194270.dist-info → deltafi-2.0rc1723500522958.dist-info}/RECORD

@@ -1,5 +1,5 @@
 deltafi/__init__.py,sha256=qv6y7PpBG0ekTN9EuD9Lj8JYOLVqZA6tvHwgjplncAM,709
-deltafi/action.py,sha256=eK18D7DCCELwEDouq3ssTv-pHWj_qWzkc00rvyKc8ho,5552
+deltafi/action.py,sha256=ejORyFe1cFCC04fdNt1VPWHqhNqZGjpLeD9tTQAYs1w,5755
 deltafi/actioneventqueue.py,sha256=mCRE1PFXy_KvaeezTeMze5N400CO1V06zEF8FFD6xZk,2847
 deltafi/actiontype.py,sha256=V7JbTgHQIaWQ2N36pG-XV7kAXhq8qvj954z71iHw_wI,913
 deltafi/domain.py,sha256=98295Cpm-y6BlbI2b7LvHj4OPD6LXOw1bqK9E8Pm5no,12030
@@ -8,7 +8,7 @@ deltafi/genericmodel.py,sha256=LBIhU1rLU4HjoD8ooMfpbEhcRkrTIeGXCP1-0palKic,1152
 deltafi/input.py,sha256=2qQcOWqz7lOXZIZipTz5XtcGESpjdnUAsZ7b9YqHw7M,1656
 deltafi/logger.py,sha256=b6vTKSJjfXfzJIKB7bLrefIvlaIQAxWkeDJYzT1EXes,2141
 deltafi/metric.py,sha256=eIDjZQVO53KuAFoEtjLNFwqMrp_7BC0Td9GLD1tb6sE,967
-deltafi/plugin.py,sha256=4PMvA22_mo2JOohvqbX96KeT6MAQQygVNuznp5vGSW4,12462
+deltafi/plugin.py,sha256=BxwesFkZ0Zyq2L05nGRz6lMKDxINTqHkcNu_s7r3O_4,12522
 deltafi/result.py,sha256=AMjGygjvix9PdUbQvSa_gZEkTn9BJh2p-BdirqdjJpM,8212
 deltafi/storage.py,sha256=toq58EPZgzj2TfDF-YpFUmRnsWSjACA0KQAZzkM04xU,2740
 deltafi/test_kit/__init__.py,sha256=qv6y7PpBG0ekTN9EuD9Lj8JYOLVqZA6tvHwgjplncAM,709
@@ -18,6 +18,6 @@ deltafi/test_kit/constants.py,sha256=epz0OS-qILcc8t7iIs5B3m2-wvZx8FpYpyb19ZsImbI
 deltafi/test_kit/egress.py,sha256=nF7YYak-_X35m7h2rSIcju86IJBf8cVaFHs7xw7-_b8,1899
 deltafi/test_kit/framework.py,sha256=mgKR1Yd75p76wIqWO5XpP6eMgC8fvz9MAfS01QV9WuU,14910
 deltafi/test_kit/transform.py,sha256=q9zQdZswTKwU5Om8E9Lwg0G5JI8LEXuRZjT9oam-Zoo,4085
-deltafi-2.0rc1722614194270.dist-info/METADATA,sha256=6aqntH7wKM_Safph-8R8oGqClDlKTP9ASbn8p6F50Hk,1446
-deltafi-2.0rc1722614194270.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-deltafi-2.0rc1722614194270.dist-info/RECORD,,
+deltafi-2.0rc1723500522958.dist-info/METADATA,sha256=qxdr7QUZ0KVJqruGH0F9UhY-OtnYyoaPv6JEfPsAomM,1446
+deltafi-2.0rc1723500522958.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+deltafi-2.0rc1723500522958.dist-info/RECORD,,