Merge pull request #28 from atlanhq/sales_246

Add functions to client to Update and Remove Announcements

Author: ErnestoLoma

pyatlan/client/atlan.py

@@ -49,6 +49,7 @@
     View,
 )
 from pyatlan.model.core import (
+    Announcement,
     AssetRequest,
     AssetResponse,
     AtlanObject,
@@ -543,3 +544,27 @@ def remove_certificate(
         asset.name = name
         asset.remove_certificate()
         return self._update_asset_by_attribute(asset, asset_type, qualified_name)
+
+    @validate_arguments()
+    def update_announcement(
+        self,
+        asset_type: Type[A],
+        qualified_name: str,
+        name: str,
+        announcement: Announcement,
+    ) -> Optional[A]:
+        asset = asset_type()
+        asset.qualified_name = qualified_name
+        asset.set_announcement(announcement)
+        asset.name = name
+        return self._update_asset_by_attribute(asset, asset_type, qualified_name)
+
+    @validate_arguments()
+    def remove_announcement(
+        self, asset_type: Type[A], qualified_name: str, name: str
+    ) -> Optional[A]:
+        asset = asset_type()
+        asset.qualified_name = qualified_name
+        asset.name = name
+        asset.remove_announcement()
+        return self._update_asset_by_attribute(asset, asset_type, qualified_name)

setup.py

@@ -29,7 +29,7 @@
 
 setup(
     name="pyatlan",
-    version="0.0.13",
+    version="0.0.14",
     author="Atlan Pte, Ltd",
     author_email="[email protected]",
     description="Atlan Python Client",

tests/integration/test_entity_model.py

@@ -670,3 +670,24 @@ def test_update_remove_certificate(client: AtlanClient):
     assert asset is not None
     assert asset.certificate_status is None
     assert asset.certificate_status_message is None
+
+
+def test_update_remove_announcement(client: AtlanClient, announcement: Announcement):
+    glossary = AtlasGlossary.create("Integration Announcement Test")
+    glossary.attributes.user_description = "This is a description of the glossary"
+    glossary = client.upsert(glossary).assets_created(AtlasGlossary)[0]
+    asset = client.update_announcement(
+        asset_type=AtlasGlossary,
+        qualified_name=glossary.qualified_name,
+        name=glossary.name,
+        announcement=announcement,
+    )
+    assert asset is not None
+    assert asset.get_announcment() == announcement
+    asset = client.remove_announcement(
+        asset_type=AtlasGlossary,
+        qualified_name=glossary.qualified_name,
+        name=glossary.name,
+    )
+    assert asset is not None
+    assert asset.get_announcment() is None