deltafi 2.4.0__py3-none-any.whl → 2.5.0__py3-none-any.whl

deltafi/domain.py

@@ -129,7 +129,7 @@ class Content:
     A Content class that holds information about a piece of content, including its name, segments, mediaType, and service.
     Attributes:
         name (str): The name of the content.
-        segments (List<Segment): The list of segments in storage that make up the Content
+        segments (List<Segment>): The list of segments in storage that make up the Content
         media_type (str): The media type of the content
         content_service (ContentService): A ContentService object used to retrieve the content data.
     """
@@ -138,6 +138,7 @@ class Content:
         self.name = name
         self.segments = segments
         self.media_type = media_type
+        self.tags = set()
         self.content_service = content_service
 
     def json(self):
@@ -150,7 +151,8 @@ class Content:
         return {
             'name': self.name,
             'segments': [segment.json() for segment in self.segments],
-            'mediaType': self.media_type
+            'mediaType': self.media_type,
+            'tags': list(self.tags)
         }
 
     def copy(self):
@@ -160,10 +162,12 @@ class Content:
         Returns:
             Content: A deep copy of the Content object.
         """
-        return Content(name=self.name,
+        new_copy = Content(name=self.name,
                        segments=copy.deepcopy(self.segments),
                        media_type=self.media_type,
                        content_service=self.content_service)
+        new_copy.add_tags(self.tags.copy())
+        return new_copy
 
     def subcontent(self, offset: int, size: int):
         """
@@ -297,11 +301,66 @@ class Content:
             segment_names[seg.id()] = seg
         return segment_names
 
+    def add_tag(self, tag: str):
+        """
+        Adds a tag to the content.
+
+        Args:
+            tag (str): The tag to add.
+        """
+        self.tags.add(tag)
+
+    def add_tags(self, tags: set):
+        """
+        Adds multiple tags to the content.
+
+        Args:
+            tags (set): A set of tags to add.
+        """
+        self.tags.update(tags)
+
+    def remove_tag(self, tag: str):
+        """
+        Removes a tag from the content.
+
+        Args:
+            tag (str): The tag to remove.
+        """
+        self.tags.discard(tag)
+
+    def has_tag(self, tag: str) -> bool:
+        """
+        Checks if the content has a specific tag.
+
+        Args:
+            tag (str): The tag to check.
+
+        Returns:
+            bool: True if the content has the tag, False otherwise.
+        """
+        return tag in self.tags
+
+    def clear_tags(self):
+        """
+        Clears all tags from the content.
+        """
+        self.tags.clear()
+
+    def get_tags(self) -> set:
+        """
+        Returns the tags associated with the content.
+
+        Returns:
+            set: A set of tags.
+        """
+        return self.tags
+
     def __eq__(self, other):
         if isinstance(other, Content):
             return (self.name == other.name and
                     self.segments == other.segments and
                     self.media_type == other.media_type and
+                    self.tags == other.tags and
                     self.content_service == other.content_service)
         return False
 
@@ -323,10 +382,13 @@ class Content:
             name = None
         segments = [Segment.from_dict(segment) for segment in content['segments']]
         media_type = content['mediaType']
-        return Content(name=name,
-                       segments=segments,
-                       media_type=media_type,
-                       content_service=content_service)
+        action_content = Content(name=name,
+                                 segments=segments,
+                                 media_type=media_type,
+                                 content_service=content_service)
+        tags = set(content.get('tags', []))
+        action_content.add_tags(tags)
+        return action_content
 
 
 class DeltaFileMessage(NamedTuple):

deltafi/result.py

@@ -112,16 +112,20 @@ class IngressResultItem:
 
         return self
 
-    def save_string_content(self, string_data: str, name: str, media_type: str):
+    def save_string_content(self, string_data: str, name: str, media_type: str, tags: set = None):
         segment = self.context.content_service.put_str(self._did, string_data)
         c = Content(name=name, segments=[segment], media_type=media_type, content_service=self.context.content_service)
+        if tags is not None:
+            c.add_tags(tags)
         self.content.append(c)
         self.context.saved_content.append(c)
         return self
 
-    def save_byte_content(self, byte_data: bytes, name: str, media_type: str):
+    def save_byte_content(self, byte_data: bytes, name: str, media_type: str, tags: set = None):
         segment = self.context.content_service.put_bytes(self._did, byte_data)
         c = Content(name=name, segments=[segment], media_type=media_type, content_service=self.context.content_service)
+        if tags is not None:
+            c.add_tags(tags)
         self.content.append(c)
         self.context.saved_content.append(c)
         return self
@@ -207,16 +211,20 @@ class TransformResult(Result):
 
         return self
 
-    def save_string_content(self, string_data: str, name: str, media_type: str):
+    def save_string_content(self, string_data: str, name: str, media_type: str, tags: set = None):
         segment = self.context.content_service.put_str(self.context.did, string_data)
         c = Content(name=name, segments=[segment], media_type=media_type, content_service=self.context.content_service)
+        if tags is not None:
+            c.add_tags(tags)
         self.content.append(c)
         self.context.saved_content.append(c)
         return self
 
-    def save_byte_content(self, byte_data: bytes, name: str, media_type: str):
+    def save_byte_content(self, byte_data: bytes, name: str, media_type: str, tags: set = None):
         segment = self.context.content_service.put_bytes(self.context.did, byte_data)
         c = Content(name=name, segments=[segment], media_type=media_type, content_service=self.context.content_service)
+        if tags is not None:
+            c.add_tags(tags)
         self.content.append(c)
         self.context.saved_content.append(c)
         return self

{deltafi-2.4.0.dist-info → deltafi-2.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: deltafi
-Version: 2.4.0
+Version: 2.5.0
 Summary: SDK for DeltaFi plugins and actions
 License: Apache License, Version 2.0
 Keywords: deltafi

{deltafi-2.4.0.dist-info → deltafi-2.5.0.dist-info}/RECORD

@@ -2,14 +2,14 @@ deltafi/__init__.py,sha256=FHwsfe4TgGeVLINN6940urXAzxGvl_Hof0gmOB32N7Y,709
 deltafi/action.py,sha256=7fgFOcbNg_6Hex5KQZ7k54VCs--Tuk6ucS3nQnB-vYk,5753
 deltafi/actioneventqueue.py,sha256=1waxnLAhoxP_nltYE107AdeFuxFdBuQDdtJjxiu2pT8,2847
 deltafi/actiontype.py,sha256=2FnTbryP2zVlJfghrh9IqhIHWarPfSKSKt2UxoIrGLU,913
-deltafi/domain.py,sha256=_tLkM1IYBKCdNQaFGtK3cARq7jVyU5Gw_vuyLuikYmE,12182
+deltafi/domain.py,sha256=XzHXLOZSUHIhlrj8jB7_UN-yYIgKVXFHji330k-TlsM,13670
 deltafi/exception.py,sha256=t-Qr3PBp5vdvUeABMFg2iLl6x6Uy3PJl5uxENsnY4kM,943
 deltafi/genericmodel.py,sha256=0qgii2o-fzHS6lM1C5X7_WTTSYsiZAM0i7RTfw0nw4U,1152
 deltafi/input.py,sha256=_9Np2g7OzcAggaQnGw7DDWleGAoqud52z60YSqPNriM,1656
 deltafi/logger.py,sha256=Lms9wmjyZWe9-SCKj4raNmGC9Cyn3BrfiniPXeM9VDY,2140
 deltafi/metric.py,sha256=5Q40bLpNXk8mwu1wnN5L8GX22-iZxYEFBGt3zSNmYxI,972
 deltafi/plugin.py,sha256=nZ3sGyggzz2w_O7E8DknNunb6mZpShj_np7IhuesJ18,17980
-deltafi/result.py,sha256=1ybUGbEyDOBhD12IyUaO6Wx7ZNQEYvzl8_I5hBZq7_Y,9251
+deltafi/result.py,sha256=j_3icJ74xIQo9KQV738XiqaFOYirjuIgFEtld4pBvXk,9555
 deltafi/storage.py,sha256=zgBkaDTiT4BLEE1lAYjq4mqS4zqZKzuoVmG1CNbkD20,2979
 deltafi/test_kit/__init__.py,sha256=FHwsfe4TgGeVLINN6940urXAzxGvl_Hof0gmOB32N7Y,709
 deltafi/test_kit/assertions.py,sha256=itKLaI22PoiiDLHNrvRAAohcrKsbKuW5-ba0sg7J780,1594
@@ -19,6 +19,6 @@ deltafi/test_kit/egress.py,sha256=MXEsYsgiG1G-1zCuwV23RCn7w5bPObP1J0_7nVN03MU,18
 deltafi/test_kit/framework.py,sha256=pU8tLsMNWUhVtIhLusbQL3uPT4y8xsKWdzzXuxmxRrU,15432
 deltafi/test_kit/timed_ingress.py,sha256=ogk6AzbPZInLOPMYy3MhjEXpITPn9-Ha6YSILGBdpro,4177
 deltafi/test_kit/transform.py,sha256=7KOsNY9OSDiyTFXIx0knZeweXIA421vHS6PAxQAcOc0,4122
-deltafi-2.4.0.dist-info/METADATA,sha256=CrbZOCzTqMlEpCwncWOf2DTzb_RvQzi-71-0r1RrVqQ,1517
-deltafi-2.4.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-deltafi-2.4.0.dist-info/RECORD,,
+deltafi-2.5.0.dist-info/METADATA,sha256=EbC3kuT0FKc434kWSUYaJA5ytVkynG6BaEDGwSnkKaQ,1517
+deltafi-2.5.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+deltafi-2.5.0.dist-info/RECORD,,