Merge pull request #472 from atlanhq/CXD-304

CXD-304 | Support for `Dataverse` typedefs

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

@@ -0,0 +1,38 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        dataverse_entity_qualified_name: str,
+    ) -> DataverseAttribute: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        dataverse_entity_qualified_name: str,
+        connection_qualified_name: str,
+    ) -> DataverseAttribute: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        dataverse_entity_qualified_name: str,
+        connection_qualified_name: Optional[str] = None,
+    ) -> DataverseAttribute:
+        validate_required_fields(
+            ["name", "dataverse_entity_qualified_name"], [name, dataverse_entity_qualified_name],
+        )
+        attributes = DataverseAttribute.Attributes.creator(
+            name=name,
+            dataverse_entity_qualified_name=dataverse_entity_qualified_name,
+            connection_qualified_name=connection_qualified_name,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,12 @@
+
+    @classmethod
+    @init_guid
+    def creator(cls, *, name: str, connection_qualified_name: str) -> DataverseEntity:
+        validate_required_fields(
+            ["name", "connection_qualified_name"], [name, connection_qualified_name],
+        )
+        attributes = DataverseEntity.Attributes.creator(
+            name=name,
+            connection_qualified_name=connection_qualified_name,
+        )
+        return cls(attributes=attributes)

pyatlan/generator/templates/methods/attribute/dataverse_attribute.jinja2

@@ -0,0 +1,35 @@
+
+        @classmethod
+        @init_guid
+        def creator(
+            cls,
+            *,
+            name: str,
+            dataverse_entity_qualified_name: str,
+            connection_qualified_name: Optional[str] = None,
+        ) -> DataverseAttribute.Attributes:
+            validate_required_fields(
+                ["name", "dataverse_entity_qualified_name"],
+                [name, dataverse_entity_qualified_name],
+            )
+            if connection_qualified_name:
+                connector_name = AtlanConnectorType.get_connector_name(
+                    connection_qualified_name
+                )
+            else:
+                connection_qn, connector_name = AtlanConnectorType.get_connector_name(
+                    dataverse_entity_qualified_name,
+                    "dataverse_entity_qualified_name",
+                    4,
+                )
+
+            return DataverseAttribute.Attributes(
+                name=name,
+                dataverse_entity_qualified_name=dataverse_entity_qualified_name,
+                connection_qualified_name=connection_qualified_name or connection_qn,
+                qualified_name=f"{dataverse_entity_qualified_name}/{name}",
+                connector_name=connector_name,
+                dataverse_entity=DataverseEntity.ref_by_qualified_name(
+                    dataverse_entity_qualified_name
+                ),
+            )

pyatlan/generator/templates/methods/attribute/dataverse_entity.jinja2

@@ -0,0 +1,17 @@
+
+        @classmethod
+        @init_guid
+        def creator(
+            cls, *, name: str, connection_qualified_name: str
+        ) -> DataverseEntity.Attributes:
+            validate_required_fields(
+                ["name", "connection_qualified_name"], [name, connection_qualified_name],
+            )
+            return DataverseEntity.Attributes(
+                name=name,
+                qualified_name=f"{connection_qualified_name}/{name}",
+                connection_qualified_name=connection_qualified_name,
+                connector_name=AtlanConnectorType.get_connector_name(
+                    connection_qualified_name
+                ),
+            )

pyatlan/model/assets/dataverse_attribute.py

@@ -4,18 +4,59 @@
 
 from __future__ import annotations
 
-from typing import ClassVar, List, Optional
+from typing import ClassVar, List, Optional, overload
 
 from pydantic.v1 import Field, validator
 
+from pyatlan.model.enums import AtlanConnectorType
 from pyatlan.model.fields.atlan_fields import BooleanField, KeywordField, RelationField
+from pyatlan.utils import init_guid, validate_required_fields
 
 from .dataverse import Dataverse
 
 
 class DataverseAttribute(Dataverse):
     """Description"""
 
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        dataverse_entity_qualified_name: str,
+    ) -> DataverseAttribute: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        dataverse_entity_qualified_name: str,
+        connection_qualified_name: str,
+    ) -> DataverseAttribute: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        dataverse_entity_qualified_name: str,
+        connection_qualified_name: Optional[str] = None,
+    ) -> DataverseAttribute:
+        validate_required_fields(
+            ["name", "dataverse_entity_qualified_name"],
+            [name, dataverse_entity_qualified_name],
+        )
+        attributes = DataverseAttribute.Attributes.creator(
+            name=name,
+            dataverse_entity_qualified_name=dataverse_entity_qualified_name,
+            connection_qualified_name=connection_qualified_name,
+        )
+        return cls(attributes=attributes)
+
     type_name: str = Field(default="DataverseAttribute", allow_mutation=False)
 
     @validator("type_name")
@@ -188,6 +229,41 @@ class Attributes(Dataverse.Attributes):
             default=None, description=""
         )  # relationship
 
+        @classmethod
+        @init_guid
+        def creator(
+            cls,
+            *,
+            name: str,
+            dataverse_entity_qualified_name: str,
+            connection_qualified_name: Optional[str] = None,
+        ) -> DataverseAttribute.Attributes:
+            validate_required_fields(
+                ["name", "dataverse_entity_qualified_name"],
+                [name, dataverse_entity_qualified_name],
+            )
+            if connection_qualified_name:
+                connector_name = AtlanConnectorType.get_connector_name(
+                    connection_qualified_name
+                )
+            else:
+                connection_qn, connector_name = AtlanConnectorType.get_connector_name(
+                    dataverse_entity_qualified_name,
+                    "dataverse_entity_qualified_name",
+                    4,
+                )
+
+            return DataverseAttribute.Attributes(
+                name=name,
+                dataverse_entity_qualified_name=dataverse_entity_qualified_name,
+                connection_qualified_name=connection_qualified_name or connection_qn,
+                qualified_name=f"{dataverse_entity_qualified_name}/{name}",
+                connector_name=connector_name,
+                dataverse_entity=DataverseEntity.ref_by_qualified_name(
+                    dataverse_entity_qualified_name
+                ),
+            )
+
     attributes: DataverseAttribute.Attributes = Field(
         default_factory=lambda: DataverseAttribute.Attributes(),
         description=(

pyatlan/model/assets/dataverse_entity.py

@@ -8,14 +8,29 @@
 
 from pydantic.v1 import Field, validator
 
+from pyatlan.model.enums import AtlanConnectorType
 from pyatlan.model.fields.atlan_fields import KeywordField, RelationField
+from pyatlan.utils import init_guid, validate_required_fields
 
 from .dataverse import Dataverse
 
 
 class DataverseEntity(Dataverse):
     """Description"""
 
+    @classmethod
+    @init_guid
+    def creator(cls, *, name: str, connection_qualified_name: str) -> DataverseEntity:
+        validate_required_fields(
+            ["name", "connection_qualified_name"],
+            [name, connection_qualified_name],
+        )
+        attributes = DataverseEntity.Attributes.creator(
+            name=name,
+            connection_qualified_name=connection_qualified_name,
+        )
+        return cls(attributes=attributes)
+
     type_name: str = Field(default="DataverseEntity", allow_mutation=False)
 
     @validator("type_name")
@@ -102,6 +117,24 @@ class Attributes(Dataverse.Attributes):
             default=None, description=""
         )  # relationship
 
+        @classmethod
+        @init_guid
+        def creator(
+            cls, *, name: str, connection_qualified_name: str
+        ) -> DataverseEntity.Attributes:
+            validate_required_fields(
+                ["name", "connection_qualified_name"],
+                [name, connection_qualified_name],
+            )
+            return DataverseEntity.Attributes(
+                name=name,
+                qualified_name=f"{connection_qualified_name}/{name}",
+                connection_qualified_name=connection_qualified_name,
+                connector_name=AtlanConnectorType.get_connector_name(
+                    connection_qualified_name
+                ),
+            )
+
     attributes: DataverseEntity.Attributes = Field(
         default_factory=lambda: DataverseEntity.Attributes(),
         description=(

pyatlan/model/constants.py

@@ -43,6 +43,8 @@
         "ColumnProcess",
         "Connection",
         "DataStudioAsset",
+        "DataverseAttribute",
+        "DataverseEntity",
         "Database",
         "DbtColumnProcess",
         "DbtMetric",

pyatlan/model/structs.py

@@ -110,13 +110,6 @@ 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 SourceTagAttachment(AtlanObject):
     """Description"""
 
@@ -261,6 +254,13 @@ def of(
         )
 
 
+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"""
 
@@ -295,18 +295,18 @@ def create(
     badge_condition_colorhex: Optional[str] = Field(default=None, description="")
 
 
-class AzureTag(AtlanObject):
+class StarredDetails(AtlanObject):
     """Description"""
 
-    azure_tag_key: str = Field(description="")
-    azure_tag_value: str = Field(description="")
+    asset_starred_by: Optional[str] = Field(default=None, description="")
+    asset_starred_at: Optional[datetime] = Field(default=None, description="")
 
 
-class StarredDetails(AtlanObject):
+class AzureTag(AtlanObject):
     """Description"""
 
-    asset_starred_by: Optional[str] = Field(default=None, description="")
-    asset_starred_at: Optional[datetime] = Field(default=None, description="")
+    azure_tag_key: str = Field(description="")
+    azure_tag_value: str = Field(description="")
 
 
 class AuthPolicyCondition(AtlanObject):
@@ -410,16 +410,16 @@ class SourceTagAttribute(AtlanObject):
 
 ColumnValueFrequencyMap.update_forward_refs()
 
-SourceTagAttachmentValue.update_forward_refs()
-
 SourceTagAttachment.update_forward_refs()
 
-BadgeCondition.update_forward_refs()
+SourceTagAttachmentValue.update_forward_refs()
 
-AzureTag.update_forward_refs()
+BadgeCondition.update_forward_refs()
 
 StarredDetails.update_forward_refs()
 
+AzureTag.update_forward_refs()
+
 AuthPolicyCondition.update_forward_refs()
 
 AwsTag.update_forward_refs()

pyatlan/model/typedef.py

@@ -53,6 +53,8 @@
     "ColumnProcess",
     "Connection",
     "DataStudioAsset",
+    "DataverseAttribute",
+    "DataverseEntity",
     "Database",
     "DbtColumnProcess",
     "DbtMetric",