Merge pull request #501 from atlanhq/APP-4979

APP-4979 :  Added support for `AnaplanSystemDimension.creator()`

Author: Aryamanz29

docs/asset/anaplansystemdimension.rst

@@ -0,0 +1,10 @@
+.. _anaplansystemdimension:
+
+AnaplanSystemDimension
+======================
+
+.. module:: pyatlan.model.assets
+    :no-index:
+
+.. autoclass:: AnaplanSystemDimension
+    :members:

docs/asset/sourcetag.rst

@@ -0,0 +1,10 @@
+.. _sourcetag:
+
+sourceTag
+=========
+
+.. module:: pyatlan.model.assets
+    :no-index:
+
+.. autoclass:: sourceTag
+    :members:

docs/assets.rst

@@ -37,6 +37,7 @@ You can interact with all of the following different kinds of assets:
     asset/anaplanmodel
     asset/anaplanmodule
     asset/anaplanpage
+    asset/anaplansystemdimension
     asset/anaplanview
     asset/anaplanworkspace
     asset/anomalo
@@ -317,3 +318,4 @@ You can interact with all of the following different kinds of assets:
     asset/view
     asset/workflow
     asset/workflowrun
+    asset/sourcetag

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

@@ -0,0 +1,18 @@
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls, *, name: str, connection_qualified_name: str
+    ) -> AnaplanSystemDimension:
+        validate_required_fields(
+            ["name", "connection_qualified_name"], [name, connection_qualified_name]
+        )
+        attributes = AnaplanSystemDimension.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
+            ),
+        )
+        return cls(attributes=attributes)

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

@@ -9,6 +9,8 @@
         admin_users: Optional[List[str]] = None,
         admin_groups: Optional[List[str]] = None,
         admin_roles: Optional[List[str]] = None,
+        host: Optional[str] = None,
+        port: Optional[int] = None,
     ) -> Connection:
         validate_required_fields(["name", "connector_type"], [name, connector_type])
         if not admin_users and not admin_groups and not admin_roles:
@@ -24,6 +26,8 @@
         attr.admin_users = set() if admin_users is None else set(admin_users)
         attr.admin_groups = set() if admin_groups is None else set(admin_groups)
         attr.admin_roles = set() if admin_roles is None else set(admin_roles)
+        attr.host = host
+        attr.port = port
         return cls(attributes=attr)
 
     @classmethod

pyatlan/model/assets/__init__.py

@@ -67,8 +67,8 @@
         "Column",
         "DatabricksUnityCatalogTag",
         "SnowflakeStream",
-        "Database",
         "CalculationView",
+        "Database",
         "Procedure",
         "SnowflakeTag",
         "CosmosMongoDB",
@@ -214,6 +214,7 @@
     "anaplan_module": ["AnaplanModule"],
     "anaplan_model": ["AnaplanModel"],
     "anaplan_app": ["AnaplanApp"],
+    "anaplan_system_dimension": ["AnaplanSystemDimension"],
     "anaplan_dimension": ["AnaplanDimension"],
     "anaplan_view": ["AnaplanView"],
     "tableau_workbook": ["TableauWorkbook"],

pyatlan/model/assets/__init__.pyi

@@ -64,8 +64,8 @@ __all__ = [
     "Column",
     "DatabricksUnityCatalogTag",
     "SnowflakeStream",
-    "Database",
     "CalculationView",
+    "Database",
     "Procedure",
     "SnowflakeTag",
     "CosmosMongoDB",
@@ -211,6 +211,7 @@ __all__ = [
     "AnaplanModule",
     "AnaplanModel",
     "AnaplanApp",
+    "AnaplanSystemDimension",
     "AnaplanDimension",
     "AnaplanView",
     "TableauWorkbook",
@@ -332,6 +333,7 @@ from .anaplan_list import AnaplanList
 from .anaplan_model import AnaplanModel
 from .anaplan_module import AnaplanModule
 from .anaplan_page import AnaplanPage
+from .anaplan_system_dimension import AnaplanSystemDimension
 from .anaplan_view import AnaplanView
 from .anaplan_workspace import AnaplanWorkspace
 from .auth_service import AuthService

pyatlan/model/assets/anaplan_system_dimension.py

@@ -0,0 +1,54 @@
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2022 Atlan Pte. Ltd.
+
+
+from __future__ import annotations
+
+from typing import ClassVar, List
+
+from pydantic.v1 import Field, validator
+
+from pyatlan.model.enums import AtlanConnectorType
+from pyatlan.utils import init_guid, validate_required_fields
+
+from .anaplan import Anaplan
+
+
+class AnaplanSystemDimension(Anaplan):
+    """Description"""
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls, *, name: str, connection_qualified_name: str
+    ) -> AnaplanSystemDimension:
+        validate_required_fields(
+            ["name", "connection_qualified_name"], [name, connection_qualified_name]
+        )
+        attributes = AnaplanSystemDimension.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
+            ),
+        )
+        return cls(attributes=attributes)
+
+    type_name: str = Field(default="AnaplanSystemDimension", allow_mutation=False)
+
+    @validator("type_name")
+    def validate_type_name(cls, v):
+        if v != "AnaplanSystemDimension":
+            raise ValueError("must be AnaplanSystemDimension")
+        return v
+
+    def __setattr__(self, name, value):
+        if name in AnaplanSystemDimension._convenience_properties:
+            return object.__setattr__(self, name, value)
+        super().__setattr__(name, value)
+
+    _convenience_properties: ClassVar[List[str]] = []
+
+
+AnaplanSystemDimension.Attributes.update_forward_refs()

pyatlan/model/assets/core/__init__.py

@@ -176,8 +176,8 @@
 Column.Attributes.update_forward_refs(**localns)
 DatabricksUnityCatalogTag.Attributes.update_forward_refs(**localns)
 SnowflakeStream.Attributes.update_forward_refs(**localns)
-Database.Attributes.update_forward_refs(**localns)
 CalculationView.Attributes.update_forward_refs(**localns)
+Database.Attributes.update_forward_refs(**localns)
 Procedure.Attributes.update_forward_refs(**localns)
 SnowflakeTag.Attributes.update_forward_refs(**localns)
 CosmosMongoDB.Attributes.update_forward_refs(**localns)

pyatlan/model/structs.py

@@ -87,13 +87,6 @@ class Action(AtlanObject):
     task_action_display_text: Optional[str] = Field(default=None, description="")
 
 
-class Histogram(AtlanObject):
-    """Description"""
-
-    boundaries: Set[float] = Field(description="")
-    frequencies: Set[float] = Field(description="")
-
-
 class KafkaTopicConsumption(AtlanObject):
     """Description"""
 
@@ -103,45 +96,18 @@ class KafkaTopicConsumption(AtlanObject):
     topic_current_offset: Optional[int] = Field(default=None, description="")
 
 
-class ColumnValueFrequencyMap(AtlanObject):
+class Histogram(AtlanObject):
     """Description"""
 
-    column_value: Optional[str] = Field(default=None, description="")
-    column_value_frequency: Optional[int] = Field(default=None, description="")
+    boundaries: Set[float] = Field(description="")
+    frequencies: Set[float] = Field(description="")
 
 
-class BadgeCondition(AtlanObject):
+class ColumnValueFrequencyMap(AtlanObject):
     """Description"""
 
-    @classmethod
-    def create(
-        cls,
-        *,
-        badge_condition_operator: BadgeComparisonOperator,
-        badge_condition_value: str,
-        badge_condition_colorhex: Union[BadgeConditionColor, str],
-    ) -> "BadgeCondition":
-        validate_required_fields(
-            [
-                "badge_condition_operator",
-                "badge_condition_value",
-                "badge_condition_colorhex",
-            ],
-            [badge_condition_operator, badge_condition_value, badge_condition_colorhex],
-        )
-        return cls(
-            badge_condition_operator=badge_condition_operator.value,
-            badge_condition_value=badge_condition_value,
-            badge_condition_colorhex=(
-                badge_condition_colorhex.value
-                if isinstance(badge_condition_colorhex, BadgeConditionColor)
-                else badge_condition_colorhex
-            ),
-        )
-
-    badge_condition_operator: Optional[str] = Field(default=None, description="")
-    badge_condition_value: Optional[str] = Field(default=None, description="")
-    badge_condition_colorhex: Optional[str] = Field(default=None, description="")
+    column_value: Optional[str] = Field(default=None, description="")
+    column_value_frequency: Optional[int] = Field(default=None, description="")
 
 
 class SourceTagAttachmentValue(AtlanObject):
@@ -295,11 +261,38 @@ def of(
         )
 
 
-class AzureTag(AtlanObject):
+class BadgeCondition(AtlanObject):
     """Description"""
 
-    azure_tag_key: str = Field(description="")
-    azure_tag_value: str = Field(description="")
+    @classmethod
+    def create(
+        cls,
+        *,
+        badge_condition_operator: BadgeComparisonOperator,
+        badge_condition_value: str,
+        badge_condition_colorhex: Union[BadgeConditionColor, str],
+    ) -> "BadgeCondition":
+        validate_required_fields(
+            [
+                "badge_condition_operator",
+                "badge_condition_value",
+                "badge_condition_colorhex",
+            ],
+            [badge_condition_operator, badge_condition_value, badge_condition_colorhex],
+        )
+        return cls(
+            badge_condition_operator=badge_condition_operator.value,
+            badge_condition_value=badge_condition_value,
+            badge_condition_colorhex=(
+                badge_condition_colorhex.value
+                if isinstance(badge_condition_colorhex, BadgeConditionColor)
+                else badge_condition_colorhex
+            ),
+        )
+
+    badge_condition_operator: Optional[str] = Field(default=None, description="")
+    badge_condition_value: Optional[str] = Field(default=None, description="")
+    badge_condition_colorhex: Optional[str] = Field(default=None, description="")
 
 
 class StarredDetails(AtlanObject):
@@ -309,6 +302,13 @@ class StarredDetails(AtlanObject):
     asset_starred_at: Optional[datetime] = Field(default=None, description="")
 
 
+class AzureTag(AtlanObject):
+    """Description"""
+
+    azure_tag_key: str = Field(description="")
+    azure_tag_value: str = Field(description="")
+
+
 class AuthPolicyCondition(AtlanObject):
     """Description"""
 
@@ -323,13 +323,6 @@ class AwsTag(AtlanObject):
     aws_tag_value: str = Field(description="")
 
 
-class GoogleTag(AtlanObject):
-    """Description"""
-
-    google_tag_key: str = Field(description="")
-    google_tag_value: str = Field(description="")
-
-
 class DbtMetricFilter(AtlanObject):
     """Description"""
 
@@ -341,6 +334,13 @@ class DbtMetricFilter(AtlanObject):
     dbt_metric_filter_value: Optional[str] = Field(default=None, description="")
 
 
+class GoogleTag(AtlanObject):
+    """Description"""
+
+    google_tag_key: str = Field(description="")
+    google_tag_value: str = Field(description="")
+
+
 class BusinessPolicyRule(AtlanObject):
     """Description"""
 
@@ -416,30 +416,30 @@ class SourceTagAttribute(AtlanObject):
 
 Action.update_forward_refs()
 
-Histogram.update_forward_refs()
-
 KafkaTopicConsumption.update_forward_refs()
 
-ColumnValueFrequencyMap.update_forward_refs()
+Histogram.update_forward_refs()
 
-BadgeCondition.update_forward_refs()
+ColumnValueFrequencyMap.update_forward_refs()
 
 SourceTagAttachmentValue.update_forward_refs()
 
 SourceTagAttachment.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()
 
-GoogleTag.update_forward_refs()
-
 DbtMetricFilter.update_forward_refs()
 
+GoogleTag.update_forward_refs()
+
 BusinessPolicyRule.update_forward_refs()
 
 AuthPolicyValiditySchedule.update_forward_refs()