Merge pull request #439 from atlanhq/FT-527

Aryamanz29 · web-flow · commit 1c13f3e541b8 · 2024-12-10T16:45:34.000+05:30
FT-527: Added `update_only`, `case_insensitive` and `table_agnostic` options to `Batch`
diff --git a/docs/assets.rst b/docs/assets.rst
@@ -52,8 +52,6 @@ You can interact with all of the following different kinds of assets:
     asset/businesspolicyexception
     asset/businesspolicyincident
     asset/businesspolicylog
-    asset/businessprocessmodel
-    asset/businessprocessmodelentity
     asset/calculationview
     asset/catalog
     asset/cloud
diff --git a/pyatlan/client/asset.py b/pyatlan/client/asset.py
diff --git a/pyatlan/generator/templates/methods/asset/asset.jinja2 b/pyatlan/generator/templates/methods/asset/asset.jinja2
@@ -43,6 +43,7 @@
         return cls.creator(*args, **kwargs)
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset], qualified_name: str = "", name: str = ""
     ) -> SelfAsset:
diff --git a/pyatlan/generator/templates/methods/asset/atlas_glossary_category.jinja2 b/pyatlan/generator/templates/methods/asset/atlas_glossary_category.jinja2
@@ -69,6 +69,7 @@
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/methods/asset/atlas_glossary_term.jinja2 b/pyatlan/generator/templates/methods/asset/atlas_glossary_term.jinja2
@@ -73,6 +73,7 @@
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/methods/asset/auth_policy.jinja2 b/pyatlan/generator/templates/methods/asset/auth_policy.jinja2
@@ -6,6 +6,7 @@
         return cls(attributes=attributes)
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/methods/asset/data_domain.jinja2 b/pyatlan/generator/templates/methods/asset/data_domain.jinja2
@@ -36,6 +36,7 @@
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/methods/asset/data_product.jinja2 b/pyatlan/generator/templates/methods/asset/data_product.jinja2
@@ -39,6 +39,7 @@
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls,
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/methods/asset/persona.jinja2 b/pyatlan/generator/templates/methods/asset/persona.jinja2
@@ -131,6 +131,7 @@
         return policy
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/methods/asset/purpose.jinja2 b/pyatlan/generator/templates/methods/asset/purpose.jinja2
@@ -137,6 +137,7 @@
             raise ValueError("No user or group specified for the policy.")
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/generator/templates/module.jinja2 b/pyatlan/generator/templates/module.jinja2
@@ -44,13 +44,22 @@ class {{ entity_def.name }}({{super_classes[0]}} {%- if "Asset" in super_classes
         {% include file_name %}
     {% endif %}
 
+    {% if entity_def.name in ("Table", "View", "MaterialisedView")%}
+    type_name: str = Field(default="{{ entity_def.name }}", allow_mutation=True)
+
+    @validator('type_name')
+    def validate_type_name(cls, v):
+        return v
+    {% else %}
+
     type_name: str = Field(default="{{ entity_def.name }}", allow_mutation=False)
 
     @validator('type_name')
     def validate_type_name(cls, v):
         if v != "{{ entity_def.name }}":
             raise ValueError('must be {{ entity_def.name }}')
         return v
+    {% endif %}
 
     {% include 'properties.jinja2' %}
     {%- if entity_def.attribute_defs or entity_def.relationship_attribute_defs %}
diff --git a/pyatlan/model/assets/__init__.py b/pyatlan/model/assets/__init__.py
@@ -136,7 +136,6 @@
     "event_store": ["EventStore"],
     "insight": ["Insight"],
     "a_p_i": ["API"],
-    "business_process_model": ["BusinessProcessModel"],
     "google": ["Google"],
     "azure": ["Azure"],
     "a_w_s": ["AWS"],
@@ -178,7 +177,6 @@
     "a_p_i_object": ["APIObject"],
     "a_p_i_path": ["APIPath"],
     "a_p_i_field": ["APIField"],
-    "business_process_model_entity": ["BusinessProcessModelEntity"],
     "data_studio_asset": ["DataStudioAsset"],
     "s3_bucket": ["S3Bucket"],
     "s3_object": ["S3Object"],
diff --git a/pyatlan/model/assets/__init__.pyi b/pyatlan/model/assets/__init__.pyi
@@ -133,7 +133,6 @@ __all__ = [
     "NoSQL",
     "Insight",
     "API",
-    "BusinessProcessModel",
     "Google",
     "Azure",
     "AWS",
@@ -175,7 +174,6 @@ __all__ = [
     "APIObject",
     "APIPath",
     "APIField",
-    "BusinessProcessModelEntity",
     "DataStudioAsset",
     "S3Bucket",
     "S3Object",
@@ -319,8 +317,6 @@ from .business_policy import BusinessPolicy
 from .business_policy_exception import BusinessPolicyException
 from .business_policy_incident import BusinessPolicyIncident
 from .business_policy_log import BusinessPolicyLog
-from .business_process_model import BusinessProcessModel
-from .business_process_model_entity import BusinessProcessModelEntity
 from .cloud import Cloud
 from .cognite import Cognite
 from .cognite3_d_model import Cognite3DModel
diff --git a/pyatlan/model/assets/business_process_model.py b/pyatlan/model/assets/business_process_model.py
diff --git a/pyatlan/model/assets/business_process_model_entity.py b/pyatlan/model/assets/business_process_model_entity.py
diff --git a/pyatlan/model/assets/core/asset.py b/pyatlan/model/assets/core/asset.py
@@ -87,6 +87,7 @@ def create(cls: Type[SelfAsset], *args, **kwargs) -> SelfAsset:
         return cls.creator(*args, **kwargs)
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset], qualified_name: str = "", name: str = ""
     ) -> SelfAsset:
diff --git a/pyatlan/model/assets/core/atlas_glossary_category.py b/pyatlan/model/assets/core/atlas_glossary_category.py
@@ -89,6 +89,7 @@ def trim_to_required(self) -> AtlasGlossaryCategory:
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/model/assets/core/atlas_glossary_term.py b/pyatlan/model/assets/core/atlas_glossary_term.py
@@ -98,6 +98,7 @@ def trim_to_required(self) -> AtlasGlossaryTerm:
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/model/assets/core/auth_policy.py b/pyatlan/model/assets/core/auth_policy.py
@@ -17,7 +17,7 @@
     RelationField,
 )
 from pyatlan.model.structs import AuthPolicyCondition, AuthPolicyValiditySchedule
-from pyatlan.utils import validate_required_fields
+from pyatlan.utils import init_guid, validate_required_fields
 
 from .asset import Asset, SelfAsset
 
@@ -32,6 +32,7 @@ def __create(cls, *, name: str) -> AuthPolicy:
         return cls(attributes=attributes)
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/model/assets/core/data_domain.py b/pyatlan/model/assets/core/data_domain.py
@@ -56,6 +56,7 @@ def create(
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/model/assets/core/data_product.py b/pyatlan/model/assets/core/data_product.py
@@ -73,6 +73,7 @@ def create(
         )
 
     @classmethod
+    @init_guid
     def updater(
         cls,
         qualified_name: str = "",
diff --git a/pyatlan/model/assets/core/materialised_view.py b/pyatlan/model/assets/core/materialised_view.py
@@ -86,12 +86,10 @@ def create(cls, *, name: str, schema_qualified_name: str) -> MaterialisedView:
         )
         return cls.creator(name=name, schema_qualified_name=schema_qualified_name)
 
-    type_name: str = Field(default="MaterialisedView", allow_mutation=False)
+    type_name: str = Field(default="MaterialisedView", allow_mutation=True)
 
     @validator("type_name")
     def validate_type_name(cls, v):
-        if v != "MaterialisedView":
-            raise ValueError("must be MaterialisedView")
         return v
 
     def __setattr__(self, name, value):
diff --git a/pyatlan/model/assets/core/persona.py b/pyatlan/model/assets/core/persona.py
@@ -161,6 +161,7 @@ def create_domain_policy(
         return policy
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/model/assets/core/table.py b/pyatlan/model/assets/core/table.py
@@ -88,12 +88,10 @@ def create(cls, *, name: str, schema_qualified_name: str) -> Table:
             schema_qualified_name=schema_qualified_name,
         )
 
-    type_name: str = Field(default="Table", allow_mutation=False)
+    type_name: str = Field(default="Table", allow_mutation=True)
 
     @validator("type_name")
     def validate_type_name(cls, v):
-        if v != "Table":
-            raise ValueError("must be Table")
         return v
 
     def __setattr__(self, name, value):
diff --git a/pyatlan/model/assets/core/view.py b/pyatlan/model/assets/core/view.py
@@ -88,12 +88,10 @@ def create(cls, *, name: str, schema_qualified_name: str) -> View:
             schema_qualified_name=schema_qualified_name,
         )
 
-    type_name: str = Field(default="View", allow_mutation=False)
+    type_name: str = Field(default="View", allow_mutation=True)
 
     @validator("type_name")
     def validate_type_name(cls, v):
-        if v != "View":
-            raise ValueError("must be View")
         return v
 
     def __setattr__(self, name, value):
diff --git a/pyatlan/model/assets/purpose.py b/pyatlan/model/assets/purpose.py
@@ -166,6 +166,7 @@ def create_data_policy(
             raise ValueError("No user or group specified for the policy.")
 
     @classmethod
+    @init_guid
     def updater(
         cls: type[SelfAsset],
         qualified_name: str = "",
diff --git a/pyatlan/model/enums.py b/pyatlan/model/enums.py
@@ -2227,6 +2227,12 @@ class AssetInputHandling(str, Enum):
     UPDATE = "update"
 
 
+class AssetCreationHandling(str, Enum):
+    FULL = "upsert"
+    PARTIAL = "partial"
+    NONE = "update"
+
+
 class AssetDeltaHandling(str, Enum):
     FULL_REPLACEMENT = "full"
     INCREMENTAL = "delta"
diff --git a/pyatlan/model/structs.py b/pyatlan/model/structs.py
diff --git a/tests/integration/test_asset_batch.py b/tests/integration/test_asset_batch.py
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,7 @@`
`69`	`69`	`)`
`70`	`70`
`71`	`71`	`@classmethod`
	`72`	`+ @init_guid`
`72`	`73`	`def updater(`
`73`	`74`	`cls: type[SelfAsset],`
`74`	`75`	`qualified_name: str = "",`
Original file line number	Diff line number	Diff line change
`@@ -73,6 +73,7 @@`
`73`	`73`	`)`
`74`	`74`
`75`	`75`	`@classmethod`
	`76`	`+ @init_guid`
`76`	`77`	`def updater(`
`77`	`78`	`cls: type[SelfAsset],`
`78`	`79`	`qualified_name: str = "",`
Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@`
`36`	`36`	`)`
`37`	`37`
`38`	`38`	`@classmethod`
	`39`	`+ @init_guid`
`39`	`40`	`def updater(`
`40`	`41`	`cls: type[SelfAsset],`
`41`	`42`	`qualified_name: str = "",`
Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,7 @@`
`39`	`39`	`)`
`40`	`40`
`41`	`41`	`@classmethod`
	`42`	`+ @init_guid`
`42`	`43`	`def updater(`
`43`	`44`	`cls,`
`44`	`45`	`qualified_name: str = "",`