Added replace_custom_metadata method to AtlanClient

Author: ErnestoLoma

pyatlan/client/atlan.py

@@ -20,6 +20,7 @@
 from urllib3.util.retry import Retry
 
 from pyatlan.client.constants import (
+    ADD_BUSINESS_ATTRIBUTE_BY_ID,
     BULK_UPDATE,
     CREATE_TYPE_DEFS,
     DELETE_ENTITY_BY_ATTRIBUTE,
@@ -57,6 +58,8 @@
     Classification,
     ClassificationName,
     Classifications,
+    CustomMetadata,
+    CustomMetadataReqest,
 )
 from pyatlan.model.enums import AtlanDeleteType, AtlanTypeCategory, CertificateStatus
 from pyatlan.model.response import AssetMutationResponse
@@ -588,3 +591,14 @@ def remove_announcement(
         asset.name = name
         asset.remove_announcement()
         return self._update_asset_by_attribute(asset, asset_type, qualified_name)
+
+    def replace_custom_metadata(self, guid: str, custom_metadata: CustomMetadata):
+        # TODO: This endpoint is not currently functioning correctly on the server
+        custom_metadata_request = CustomMetadataReqest(__root__=custom_metadata)
+        self._call_api(
+            ADD_BUSINESS_ATTRIBUTE_BY_ID.format_path(
+                {"entity_guid": guid, "bm_id": custom_metadata._meta_data_type_id}
+            ),
+            None,
+            custom_metadata_request,
+        )

pyatlan/client/constants.py

@@ -122,7 +122,12 @@
 
 # Business Attributes APIs
 ADD_BUSINESS_ATTRIBUTE = API(
-    ENTITY_API + "guid/{entity_guid}/businessmetadata",
+    ENTITY_API + "guid/{entity_guid}/businessmetadata/",
+    HTTPMethod.POST,
+    HTTPStatus.NO_CONTENT,
+)
+ADD_BUSINESS_ATTRIBUTE_BY_ID = API(
+    ENTITY_API + "guid/{entity_guid}/businessmetadata/{bm_id}",
     HTTPMethod.POST,
     HTTPStatus.NO_CONTENT,
 )

pyatlan/model/core.py

@@ -197,3 +197,7 @@ def __setattr__(self, key, value):
         if not hasattr(self, key):
             raise AttributeError(f"Attribute {key} does not exist")
         super().__setattr__(key, value)
+
+
+class CustomMetadataReqest(AtlanObject):
+    __root__: CustomMetadata