[generator] Regenerated latest typedef models

Author: Aryamanz29

pyatlan/generator/class_generator.py

@@ -213,6 +213,14 @@ class AssetInfo:
     entity_defs_by_name: Dict[str, EntityDef] = {}
     sub_type_names_to_ignore: Set[str] = set()
     is_core_asset: bool = False
+    _ASSETS_REQUIRE_CLIENT = {
+        "Badge",
+        "Collection",
+        "Connection",
+        "DataProduct",
+        "Referenceable",
+        "Purpose",
+    }
     _CORE_ASSETS = {
         "Referenceable",
         "Asset",

pyatlan/generator/templates/imports.jinja2

@@ -127,5 +127,3 @@ from pyatlan.model.data_mesh import DataProductsAssetsDSL
 from pyatlan.model.contract import DataContractSpec
 from pyatlan.model.lineage_ref import LineageRef
 from pyatlan.model.utils import construct_object_key
-if TYPE_CHECKING:
-    from pyatlan.client.atlan import AtlanClient

pyatlan/generator/templates/methods/asset/data_product.jinja2

@@ -89,7 +89,7 @@
         """
         Reterieves list of all assets linked to the provided data product.
 
-        :param client: connectivity to an Atlan tenant.
+        :param client: connectivity to an Atlan tenant
         :raises InvalidRequestError: if DataProduct asset DSL cannot be found (does not exist) in Atlan
         :raises AtlanError: if there is an issue interacting with the API
         :returns: instance of `IndexSearchResults` with list of all assets linked to the provided data product

pyatlan/generator/templates/module.jinja2

@@ -6,7 +6,7 @@
 
 {% include 'imports.jinja2' %}
 
-{% if asset_info.name != 'DataProduct' %}
+{% if asset_info.name not in asset_info._ASSETS_REQUIRE_CLIENT %}
 from pyatlan.client.atlan import AtlanClient
 {% endif %}
 
@@ -32,7 +32,7 @@ from .asset import SelfAsset
 
 {{ asset_info.import_super_class }}
 
-{% if asset_info.name == 'DataProduct' %}
+{% if asset_info.name in asset_info._ASSETS_REQUIRE_CLIENT %}
 if TYPE_CHECKING:
     from pyatlan.client.atlan import AtlanClient
 {% endif %}

pyatlan/model/assets/collection.py

@@ -4,19 +4,21 @@
 
 from __future__ import annotations
 
-from typing import ClassVar, List, Optional
+from typing import TYPE_CHECKING, ClassVar, List, Optional
 from uuid import uuid4
 
 from pydantic.v1 import Field, validator
 
-from pyatlan.client.atlan import AtlanClient
 from pyatlan.errors import AtlanError, ErrorCode
 from pyatlan.model.enums import IconType
 from pyatlan.model.fields.atlan_fields import KeywordField, TextField
 from pyatlan.utils import init_guid, validate_required_fields
 
 from .core.namespace import Namespace
 
+if TYPE_CHECKING:
+    from pyatlan.client.atlan import AtlanClient
+
 
 class Collection(Namespace):
     """Description"""

pyatlan/model/assets/core/access_control.py

@@ -40,6 +40,12 @@ def __setattr__(self, name, value):
     """
     TBC
     """
+    DENY_SIDEBAR_TABS: ClassVar[KeywordField] = KeywordField(
+        "denySidebarTabs", "denySidebarTabs"
+    )
+    """
+    TBC
+    """
     DENY_CUSTOM_METADATA_GUIDS: ClassVar[KeywordField] = KeywordField(
         "denyCustomMetadataGuids", "denyCustomMetadataGuids"
     )
@@ -100,6 +106,7 @@ def __setattr__(self, name, value):
 
     _convenience_properties: ClassVar[List[str]] = [
         "is_access_control_enabled",
+        "deny_sidebar_tabs",
         "deny_custom_metadata_guids",
         "deny_asset_metadata_types",
         "deny_asset_tabs",
@@ -126,6 +133,16 @@ def is_access_control_enabled(self, is_access_control_enabled: Optional[bool]):
             self.attributes = self.Attributes()
         self.attributes.is_access_control_enabled = is_access_control_enabled
 
+    @property
+    def deny_sidebar_tabs(self) -> Optional[Set[str]]:
+        return None if self.attributes is None else self.attributes.deny_sidebar_tabs
+
+    @deny_sidebar_tabs.setter
+    def deny_sidebar_tabs(self, deny_sidebar_tabs: Optional[Set[str]]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.deny_sidebar_tabs = deny_sidebar_tabs
+
     @property
     def deny_custom_metadata_guids(self) -> Optional[Set[str]]:
         return (
@@ -240,6 +257,7 @@ def policies(self, policies: Optional[List[AuthPolicy]]):
 
     class Attributes(Asset.Attributes):
         is_access_control_enabled: Optional[bool] = Field(default=None, description="")
+        deny_sidebar_tabs: Optional[Set[str]] = Field(default=None, description="")
         deny_custom_metadata_guids: Optional[Set[str]] = Field(
             default=None, description=""
         )

pyatlan/model/structs.py

@@ -45,19 +45,6 @@ def flatten_structs_attributes(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         return values
 
 
-class MCRuleSchedule(AtlanObject):
-    """Description"""
-
-    mc_rule_schedule_type: Optional[str] = Field(default=None, description="")
-    mc_rule_schedule_interval_in_minutes: Optional[int] = Field(
-        default=None, description=""
-    )
-    mc_rule_schedule_start_time: Optional[datetime] = Field(
-        default=None, description=""
-    )
-    mc_rule_schedule_crontab: Optional[str] = Field(default=None, description="")
-
-
 class DbtJobRun(AtlanObject):
     """Description"""
 
@@ -74,6 +61,19 @@ class DbtJobRun(AtlanObject):
     dbt_compiled_code: Optional[str] = Field(default=None, description="")
 
 
+class MCRuleSchedule(AtlanObject):
+    """Description"""
+
+    mc_rule_schedule_type: Optional[str] = Field(default=None, description="")
+    mc_rule_schedule_interval_in_minutes: Optional[int] = Field(
+        default=None, description=""
+    )
+    mc_rule_schedule_start_time: Optional[datetime] = Field(
+        default=None, description=""
+    )
+    mc_rule_schedule_crontab: Optional[str] = Field(default=None, description="")
+
+
 class AwsCloudWatchMetric(AtlanObject):
     """Description"""
 
@@ -104,6 +104,13 @@ class ColumnValueFrequencyMap(AtlanObject):
     column_value_frequency: Optional[int] = Field(default=None, description="")
 
 
+class SourceTagAttachmentValue(AtlanObject):
+    """Description"""
+
+    tag_attachment_key: Optional[str] = Field(default=None, description="")
+    tag_attachment_value: Optional[str] = Field(default=None, description="")
+
+
 class BadgeCondition(AtlanObject):
     """Description"""
 
@@ -136,13 +143,6 @@ def create(
     badge_condition_colorhex: Optional[str] = Field(default=None, description="")
 
 
-class SourceTagAttachmentValue(AtlanObject):
-    """Description"""
-
-    tag_attachment_key: Optional[str] = Field(default=None, description="")
-    tag_attachment_value: Optional[str] = Field(default=None, description="")
-
-
 class StarredDetails(AtlanObject):
     """Description"""
 
@@ -444,10 +444,10 @@ class SourceTagAttribute(AtlanObject):
     )
 
 
-MCRuleSchedule.update_forward_refs()
-
 DbtJobRun.update_forward_refs()
 
+MCRuleSchedule.update_forward_refs()
+
 AwsCloudWatchMetric.update_forward_refs()
 
 Histogram.update_forward_refs()
@@ -456,10 +456,10 @@ class SourceTagAttribute(AtlanObject):
 
 ColumnValueFrequencyMap.update_forward_refs()
 
-BadgeCondition.update_forward_refs()
-
 SourceTagAttachmentValue.update_forward_refs()
 
+BadgeCondition.update_forward_refs()
+
 StarredDetails.update_forward_refs()
 
 AwsTag.update_forward_refs()