Add retreive_minimal method to AtlanClient

Author: ErnestoLoma

pyatlan/client/atlan.py

@@ -320,7 +320,7 @@ def get_asset_by_qualified_name(
             return asset
         except AtlanError as ae:
             if ae.status_code == HTTPStatus.NOT_FOUND:
-                raise NotFoundError(message=ae.user_message, code=ae.code)
+                raise NotFoundError(message=ae.user_message, code=ae.code) from ae
             raise ae
 
     @validate_arguments()
@@ -355,9 +355,18 @@ def get_asset_by_guid(
             return asset
         except AtlanError as ae:
             if ae.status_code == HTTPStatus.NOT_FOUND:
-                raise NotFoundError(message=ae.user_message, code=ae.code)
+                raise NotFoundError(message=ae.user_message, code=ae.code) from ae
             raise ae
 
+    @validate_arguments()
+    def retrieve_minimal(self, guid: str, asset_type: Type[A]) -> A:
+        return self.get_asset_by_guid(
+            guid=guid,
+            asset_type=asset_type,
+            min_ext_info=True,
+            ignore_relationships=True,
+        )
+
     def upsert(
         self,
         entity: Union[Asset, list[Asset]],
@@ -419,7 +428,6 @@ def get_typedefs(self, type_category: AtlanTypeCategory) -> TypeDefResponse:
         return TypeDefResponse(**raw_json)
 
     def create_typedef(self, typedef: TypeDef) -> TypeDefResponse:
-        payload = None
         if isinstance(typedef, ClassificationDef):
             # Set up the request payload...
             payload = TypeDefResponse(
@@ -463,15 +471,15 @@ def add_classifications(
         classification_names: list[str],
         propagate: bool = True,
         remove_propagation_on_delete: bool = True,
-        restrict_lineage_propogation: bool = True,
+        restrict_lineage_propagation: bool = True,
     ) -> None:
         classifications = Classifications(
             __root__=[
                 Classification(
                     type_name=ClassificationName(display_text=name),
                     propagate=propagate,
                     remove_propagations_on_entity_delete=remove_propagation_on_delete,
-                    restrict_propagation_through_lineage=restrict_lineage_propogation,
+                    restrict_propagation_through_lineage=restrict_lineage_propagation,
                 )
                 for name in classification_names
             ]
@@ -528,11 +536,9 @@ def _update_asset_by_attribute(self, asset, asset_type, qualified_name: str):
             AssetRequest[Asset](entity=asset),
         )
         response = AssetMutationResponse(**raw_json)
-        assets = response.assets_partially_updated(asset_type=asset_type)
-        if assets:
+        if assets := response.assets_partially_updated(asset_type=asset_type):
             return assets[0]
-        assets = response.assets_updated(asset_type=asset_type)
-        if assets:
+        if assets := response.assets_updated(asset_type=asset_type):
             return assets[0]
         return None
 

pyatlan/generator/templates/entity.jinja2

@@ -233,21 +233,25 @@ class {{ entity_def.name }}({{super_classes[0]}} {%- if "Asset" in super_classes
         if ba_id is None:
             raise ValueError(f"No business attributes with the name: {name} exist")
         for a_type in CustomMetadataCache.types_by_asset[self.type_name]:
-            if hasattr(a_type, "_meta_data_type_name") and a_type._meta_data_type_name == name:
+            if (
+                hasattr(a_type, "_meta_data_type_name")
+                and a_type._meta_data_type_name == name
+            ):
                 break
         else:
             raise ValueError(
                 f"Business attributes {name} are not applicable to {self.type_name}"
             )
-        ba_type = CustomMetadataCache.get_type_for_id(ba_id)
-        if not ba_type:
+        if ba_type := CustomMetadataCache.get_type_for_id(ba_id):
+            return (
+                ba_type(self.business_attributes[ba_id])
+                if self.business_attributes and ba_id in self.business_attributes
+                else ba_type()
+            )
+        else:
             raise ValueError(
                 f"Business attributes {name} are not applicable to {self.type_name}"
             )
-        if self.business_attributes and ba_id in self.business_attributes:
-            return ba_type(self.business_attributes[ba_id])
-        else:
-            return ba_type()
 
     def set_business_attribute(self, business_attributes: BusinessAttributes) -> None:
         from pyatlan.cache.custom_metadata_cache import CustomMetadataCache

pyatlan/model/assets.py

@@ -262,15 +262,16 @@ def get_business_attributes(self, name: str) -> BusinessAttributes:
             raise ValueError(
                 f"Business attributes {name} are not applicable to {self.type_name}"
             )
-        ba_type = CustomMetadataCache.get_type_for_id(ba_id)
-        if not ba_type:
+        if ba_type := CustomMetadataCache.get_type_for_id(ba_id):
+            return (
+                ba_type(self.business_attributes[ba_id])
+                if self.business_attributes and ba_id in self.business_attributes
+                else ba_type()
+            )
+        else:
             raise ValueError(
                 f"Business attributes {name} are not applicable to {self.type_name}"
             )
-        if self.business_attributes and ba_id in self.business_attributes:
-            return ba_type(self.business_attributes[ba_id])
-        else:
-            return ba_type()
 
     def set_business_attribute(self, business_attributes: BusinessAttributes) -> None:
         from pyatlan.cache.custom_metadata_cache import CustomMetadataCache