Added additional integration tests for purging and creating glossaries

Author: ErnestoLoma

pyatlan/client/atlan.py

@@ -57,7 +57,7 @@ def call_api(self, api, query_params=None, request_obj=None):
 
         if request_obj is not None:
             if isinstance(request_obj, AtlanObject):
-                params["data"] = request_obj.json(by_alias=True, exclude_unset=True)
+                params["data"] = request_obj.json(by_alias=True)
             else:
                 params["data"] = json.dumps(request_obj)
 

pyatlan/client/entity.py

@@ -25,6 +25,8 @@
     AssetMutationResponse,
     BulkRequest,
 )
+
+from pyatlan.model.enums import AtlanDeleteType
 from pyatlan.utils import (
     API,
     APPLICATION_JSON,
@@ -168,6 +170,10 @@ class EntityClient:
         MULTIPART_FORM_DATA,
         APPLICATION_JSON,
     )
+    # Glossary APIS
+    GLOSSARY_URI = BASE_URI + "glossary"
+
+    GET_ALL_GLOSSARIES = API(GLOSSARY_URI, HTTPMethod.GET, HTTPStatus.OK)
 
     # Labels APIs
     ADD_LABELS = API(
@@ -196,6 +202,11 @@ class EntityClient:
         HTTPMethod.DELETE,
         HTTPStatus.NO_CONTENT,
     )
+    DEFAULT_LIMIT = -1
+    DEFAULT_OFFSET = 0
+    DEFAULT_SORT = "ASC"
+    LIMIT = "limit"
+    OFFSET = "offset"
 
     def __init__(self, client: AtlanClient):
         self.client = client
@@ -228,3 +239,17 @@ def update_entity(self, entity: T) -> AssetMutationResponse[T]:
         raw_json = self.client.call_api(EntityClient.BULK_UPDATE, None, request)
         response: AssetMutationResponse[T] = AssetMutationResponse(**raw_json)
         return response
+
+    def create_entity(self, entity: T) -> AssetMutationResponse[T]:
+        request = BulkRequest[T](entities=[entity])
+        raw_json = self.client.call_api(EntityClient.BULK_UPDATE, None, request)
+        response: AssetMutationResponse[T] = AssetMutationResponse(**raw_json)
+        return response
+
+    def purge_entity_by_guid(self, guid) -> AssetMutationResponse[T]:
+        raw_json = self.client.call_api(
+            EntityClient.DELETE_ENTITY_BY_GUID.format_path_with_params(guid),
+            {"deleteType": AtlanDeleteType.HARD.value},
+        )
+        response: AssetMutationResponse[T] = AssetMutationResponse(**raw_json)
+        return response

pyatlan/generator/templates/entity.jinja2

@@ -5,6 +5,7 @@ from datetime import datetime
 from pyatlan.model.core import AtlanObject, Classification, Announcement
 from pyatlan.model.enums import CertificateStatus, EntityStatus, google_datastudio_asset_type, powerbi_endorsement, \
     IconType, AnnouncementType, SourceCostUnitType
+from pyatlan.utils import next_id
 
 class Internal(AtlanObject):
     """For internal usage"""
@@ -29,7 +30,8 @@ class {{ entity_def.name }}({{super_classes[0]}}):
     class Attributes(AtlanObject):
         {%- for attribute_def in entity_def.attribute_defs %}
         {%- set type = attribute_def.typeName | get_type %}
-        {{attribute_def.name | to_snake_case }}: {% if attribute_def.isOptional %}Optional[{% endif %}{{type}}{% if attribute_def.isOptional %}]{% endif %} = Field(None, description='' , alias='{{attribute_def.name}}')
+        {%- set default_value = "''" if attribute_def.name == "qualifiedName" else "None" %}
+        {{attribute_def.name | to_snake_case }}: {% if attribute_def.isOptional %}Optional[{% endif %}{{type}}{% if attribute_def.isOptional %}]{% endif %} = Field({{ default_value }}, description='' , alias='{{attribute_def.name}}')
         {%- endfor %}
         {%- for attribute_def in entity_def.relationship_attribute_defs %}
         {%- set type = attribute_def.typeName | get_type %}
@@ -56,9 +58,9 @@ class {{ entity_def.name }}({{super_classes[0]}}):
         example=1648852296555,
     )
     guid: Optional[str] = Field(
-        None,
         description='Unique identifier of the entity instance.\n',
         example='917ffec9-fa84-4c59-8e6c-c7b114d04be3',
+        default_factory=next_id,
     )
     is_incomplete: Optional[bool] = Field(False, description='', example=False)
     labels: Optional[List[str]] = Field(None, description='Internal use only.')
@@ -124,7 +126,7 @@ class {{ entity_def.name }}({{super_classes[0]}}):
     {%- if entity_def.attribute_defs %}
         {%- if not entity_def.sub_types %}
 
-    type_name: Literal['{{ entity_def.name }}']
+    type_name: Literal['{{ entity_def.name }}'] = Field("{{ entity_def.name }}")
 
         {%- endif %}
     class Attributes({{super_classes[0]}}.Attributes):