Merge pull request #768 from atlanhq/APP-9295

App 9295 - Used KeywordField for AtlanTask.GUID

Author: Aryamanz29

docs/asset/databricksmetricview.rst

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

docs/asset/partialv01.rst

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

docs/asset/partialv01field.rst

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

docs/asset/partialv01object.rst

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

docs/assets.rst

@@ -130,6 +130,7 @@ You can interact with all of the following different kinds of assets:
     asset/databricksaimodelversion
     asset/databricksexternallocation
     asset/databricksexternallocationpath
+    asset/databricksmetricview
     asset/databricksnotebook
     asset/databricksunitycatalogtag
     asset/databricksvolume
@@ -264,6 +265,9 @@ You can interact with all of the following different kinds of assets:
     asset/nosql
     asset/notebook
     asset/objectstore
+    asset/partialv01
+    asset/partialv01field
+    asset/partialv01object
     asset/persona
     asset/powerbi
     asset/powerbiapp

pyatlan/model/assets/__init__.py

@@ -30,6 +30,7 @@
         "Fivetran",
         "DataContract",
         "DataQuality",
+        "PartialV01",
         "AI",
         "Resource",
         "FlowField",
@@ -71,6 +72,7 @@
         "Metric",
         "DataQualityRule",
         "Soda",
+        "PartialV01Field",
         "AIApplication",
         "AIModelVersion",
         "AIModel",
@@ -79,7 +81,6 @@
         "Link",
         "DataDomain",
         "DataProduct",
-        "Dremio",
         "Table",
         "Query",
         "Schema",
@@ -90,8 +91,8 @@
         "TablePartition",
         "Column",
         "SnowflakeStage",
-        "DatabricksUnityCatalogTag",
         "SnowflakeStream",
+        "DatabricksUnityCatalogTag",
         "Database",
         "CalculationView",
         "Procedure",
@@ -151,6 +152,7 @@
         "SnowflakeAIModelContext",
         "DatabricksAIModelContext",
         "SnowflakeDynamicTable",
+        "DatabricksMetricView",
         "BigqueryRoutine",
         "DatabricksVolume",
         "DatabricksVolumePath",
@@ -229,6 +231,7 @@
     "a_d_l_s": ["ADLS"],
     "g_c_s": ["GCS"],
     "databricks_notebook": ["DatabricksNotebook"],
+    "partial_v01_object": ["PartialV01Object"],
     "dataverse": ["Dataverse"],
     "cognite": ["Cognite"],
     "salesforce": ["Salesforce"],
@@ -238,6 +241,7 @@
     "cube_dimension": ["CubeDimension"],
     "cube_field": ["CubeField"],
     "custom_entity": ["CustomEntity"],
+    "dremio": ["Dremio"],
     "bigquery_tag": ["BigqueryTag"],
     "snowflake": ["Snowflake"],
     "a_p_i_spec": ["APISpec"],

pyatlan/model/assets/__init__.pyi

@@ -27,6 +27,7 @@ __all__ = [
     "Fivetran",
     "DataContract",
     "DataQuality",
+    "PartialV01",
     "AI",
     "Resource",
     "FlowField",
@@ -68,6 +69,7 @@ __all__ = [
     "Metric",
     "DataQualityRule",
     "Soda",
+    "PartialV01Field",
     "AIApplication",
     "AIModelVersion",
     "AIModel",
@@ -76,7 +78,6 @@ __all__ = [
     "Link",
     "DataDomain",
     "DataProduct",
-    "Dremio",
     "Table",
     "Query",
     "Schema",
@@ -87,8 +88,8 @@ __all__ = [
     "TablePartition",
     "Column",
     "SnowflakeStage",
-    "DatabricksUnityCatalogTag",
     "SnowflakeStream",
+    "DatabricksUnityCatalogTag",
     "Database",
     "CalculationView",
     "Procedure",
@@ -148,6 +149,7 @@ __all__ = [
     "SnowflakeAIModelContext",
     "DatabricksAIModelContext",
     "SnowflakeDynamicTable",
+    "DatabricksMetricView",
     "BigqueryRoutine",
     "DatabricksVolume",
     "DatabricksVolumePath",
@@ -225,6 +227,7 @@ __all__ = [
     "ADLS",
     "GCS",
     "DatabricksNotebook",
+    "PartialV01Object",
     "Dataverse",
     "Cognite",
     "Salesforce",
@@ -234,6 +237,7 @@ __all__ = [
     "CubeDimension",
     "CubeField",
     "CustomEntity",
+    "Dremio",
     "BigqueryTag",
     "Snowflake",
     "APISpec",
@@ -507,6 +511,7 @@ from .core.database import Database
 from .core.databricks import Databricks
 from .core.databricks_a_i_model_context import DatabricksAIModelContext
 from .core.databricks_a_i_model_version import DatabricksAIModelVersion
+from .core.databricks_metric_view import DatabricksMetricView
 from .core.databricks_unity_catalog_tag import DatabricksUnityCatalogTag
 from .core.databricks_volume import DatabricksVolume
 from .core.databricks_volume_path import DatabricksVolumePath
@@ -520,7 +525,6 @@ from .core.dbt_test import DbtTest
 from .core.document_d_b import DocumentDB
 from .core.document_d_b_collection import DocumentDBCollection
 from .core.document_d_b_database import DocumentDBDatabase
-from .core.dremio import Dremio
 from .core.dynamo_d_b_secondary_index import DynamoDBSecondaryIndex
 from .core.fabric import Fabric
 from .core.fabric_activity import FabricActivity
@@ -570,6 +574,8 @@ from .core.mongo_d_b_database import MongoDBDatabase
 from .core.monte_carlo import MonteCarlo
 from .core.namespace import Namespace
 from .core.no_s_q_l import NoSQL
+from .core.partial_v01 import PartialV01
+from .core.partial_v01_field import PartialV01Field
 from .core.persona import Persona
 from .core.power_b_i import PowerBI
 from .core.power_b_i_app import PowerBIApp
@@ -635,6 +641,7 @@ from .domo_card import DomoCard
 from .domo_dashboard import DomoDashboard
 from .domo_dataset import DomoDataset
 from .domo_dataset_column import DomoDatasetColumn
+from .dremio import Dremio
 from .dremio_column import DremioColumn
 from .dremio_folder import DremioFolder
 from .dremio_physical_dataset import DremioPhysicalDataset
@@ -695,6 +702,7 @@ from .mongo_d_b import MongoDB
 from .multi_dimensional_dataset import MultiDimensionalDataset
 from .notebook import Notebook
 from .object_store import ObjectStore
+from .partial_v01_object import PartialV01Object
 from .preset import Preset
 from .preset_chart import PresetChart
 from .preset_dashboard import PresetDashboard

pyatlan/model/assets/a_d_l_s.py

@@ -8,7 +8,11 @@
 
 from pydantic.v1 import Field, validator
 
-from pyatlan.model.fields.atlan_fields import KeywordField, KeywordTextField
+from pyatlan.model.fields.atlan_fields import (
+    BooleanField,
+    KeywordField,
+    KeywordTextField,
+)
 from pyatlan.model.structs import AzureTag
 
 from .object_store import ObjectStore
@@ -44,6 +48,12 @@ def __setattr__(self, name, value):
     """
     Name of the account for this ADLS asset.
     """
+    CATALOG_HAS_PARTIAL_FIELDS: ClassVar[BooleanField] = BooleanField(
+        "catalogHasPartialFields", "catalogHasPartialFields"
+    )
+    """
+    Indicates this catalog asset has partial fields, if true.
+    """
     AZURE_RESOURCE_ID: ClassVar[KeywordTextField] = KeywordTextField(
         "azureResourceId", "azureResourceId", "azureResourceId.text"
     )
@@ -70,6 +80,7 @@ def __setattr__(self, name, value):
     _convenience_properties: ClassVar[List[str]] = [
         "adls_account_qualified_name",
         "adls_account_name",
+        "catalog_has_partial_fields",
         "azure_resource_id",
         "azure_location",
         "adls_account_secondary_location",
@@ -100,6 +111,20 @@ def adls_account_name(self, adls_account_name: Optional[str]):
             self.attributes = self.Attributes()
         self.attributes.adls_account_name = adls_account_name
 
+    @property
+    def catalog_has_partial_fields(self) -> Optional[bool]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.catalog_has_partial_fields
+        )
+
+    @catalog_has_partial_fields.setter
+    def catalog_has_partial_fields(self, catalog_has_partial_fields: Optional[bool]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.catalog_has_partial_fields = catalog_has_partial_fields
+
     @property
     def azure_resource_id(self) -> Optional[str]:
         return None if self.attributes is None else self.attributes.azure_resource_id
@@ -151,6 +176,7 @@ def azure_tags(self, azure_tags: Optional[List[AzureTag]]):
     class Attributes(ObjectStore.Attributes):
         adls_account_qualified_name: Optional[str] = Field(default=None, description="")
         adls_account_name: Optional[str] = Field(default=None, description="")
+        catalog_has_partial_fields: Optional[bool] = Field(default=None, description="")
         azure_resource_id: Optional[str] = Field(default=None, description="")
         azure_location: Optional[str] = Field(default=None, description="")
         adls_account_secondary_location: Optional[str] = Field(

pyatlan/model/assets/bigquery_tag.py

@@ -77,6 +77,12 @@ def __setattr__(self, name, value):
     """
     Name of the classification in Atlan that is mapped to this tag.
     """
+    CATALOG_HAS_PARTIAL_FIELDS: ClassVar[BooleanField] = BooleanField(
+        "catalogHasPartialFields", "catalogHasPartialFields"
+    )
+    """
+    Indicates this catalog asset has partial fields, if true.
+    """
     QUERY_COUNT: ClassVar[NumericField] = NumericField("queryCount", "queryCount")
     """
     Number of times this asset has been queried.
@@ -213,6 +219,7 @@ def __setattr__(self, name, value):
         "tag_attributes",
         "tag_allowed_values",
         "mapped_atlan_tag_name",
+        "catalog_has_partial_fields",
         "query_count",
         "query_user_count",
         "query_user_map",
@@ -323,6 +330,20 @@ def mapped_atlan_tag_name(self, mapped_atlan_tag_name: Optional[str]):
             self.attributes = self.Attributes()
         self.attributes.mapped_atlan_tag_name = mapped_atlan_tag_name
 
+    @property
+    def catalog_has_partial_fields(self) -> Optional[bool]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.catalog_has_partial_fields
+        )
+
+    @catalog_has_partial_fields.setter
+    def catalog_has_partial_fields(self, catalog_has_partial_fields: Optional[bool]):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.catalog_has_partial_fields = catalog_has_partial_fields
+
     @property
     def query_count(self) -> Optional[int]:
         return None if self.attributes is None else self.attributes.query_count
@@ -601,6 +622,7 @@ class Attributes(Tag.Attributes):
         )
         tag_allowed_values: Optional[Set[str]] = Field(default=None, description="")
         mapped_atlan_tag_name: Optional[str] = Field(default=None, description="")
+        catalog_has_partial_fields: Optional[bool] = Field(default=None, description="")
         query_count: Optional[int] = Field(default=None, description="")
         query_user_count: Optional[int] = Field(default=None, description="")
         query_user_map: Optional[Dict[str, int]] = Field(default=None, description="")

pyatlan/model/assets/core/__init__.py

@@ -48,6 +48,7 @@
 from .databricks import Databricks
 from .databricks_a_i_model_context import DatabricksAIModelContext
 from .databricks_a_i_model_version import DatabricksAIModelVersion
+from .databricks_metric_view import DatabricksMetricView
 from .databricks_unity_catalog_tag import DatabricksUnityCatalogTag
 from .databricks_volume import DatabricksVolume
 from .databricks_volume_path import DatabricksVolumePath
@@ -61,7 +62,6 @@
 from .document_d_b import DocumentDB
 from .document_d_b_collection import DocumentDBCollection
 from .document_d_b_database import DocumentDBDatabase
-from .dremio import Dremio
 from .dynamo_d_b_secondary_index import DynamoDBSecondaryIndex
 from .fabric import Fabric
 from .fabric_activity import FabricActivity
@@ -111,6 +111,8 @@
 from .monte_carlo import MonteCarlo
 from .namespace import Namespace
 from .no_s_q_l import NoSQL
+from .partial_v01 import PartialV01
+from .partial_v01_field import PartialV01Field
 from .persona import Persona
 from .power_b_i import PowerBI
 from .power_b_i_app import PowerBIApp
@@ -181,6 +183,7 @@
 Fivetran.Attributes.update_forward_refs(**localns)
 DataContract.Attributes.update_forward_refs(**localns)
 DataQuality.Attributes.update_forward_refs(**localns)
+PartialV01.Attributes.update_forward_refs(**localns)
 AI.Attributes.update_forward_refs(**localns)
 Resource.Attributes.update_forward_refs(**localns)
 FlowField.Attributes.update_forward_refs(**localns)
@@ -222,6 +225,7 @@
 Metric.Attributes.update_forward_refs(**localns)
 DataQualityRule.Attributes.update_forward_refs(**localns)
 Soda.Attributes.update_forward_refs(**localns)
+PartialV01Field.Attributes.update_forward_refs(**localns)
 AIApplication.Attributes.update_forward_refs(**localns)
 AIModelVersion.Attributes.update_forward_refs(**localns)
 AIModel.Attributes.update_forward_refs(**localns)
@@ -230,7 +234,6 @@
 Link.Attributes.update_forward_refs(**localns)
 DataDomain.Attributes.update_forward_refs(**localns)
 DataProduct.Attributes.update_forward_refs(**localns)
-Dremio.Attributes.update_forward_refs(**localns)
 Table.Attributes.update_forward_refs(**localns)
 Query.Attributes.update_forward_refs(**localns)
 Schema.Attributes.update_forward_refs(**localns)
@@ -241,8 +244,8 @@
 TablePartition.Attributes.update_forward_refs(**localns)
 Column.Attributes.update_forward_refs(**localns)
 SnowflakeStage.Attributes.update_forward_refs(**localns)
-DatabricksUnityCatalogTag.Attributes.update_forward_refs(**localns)
 SnowflakeStream.Attributes.update_forward_refs(**localns)
+DatabricksUnityCatalogTag.Attributes.update_forward_refs(**localns)
 Database.Attributes.update_forward_refs(**localns)
 CalculationView.Attributes.update_forward_refs(**localns)
 Procedure.Attributes.update_forward_refs(**localns)
@@ -302,6 +305,7 @@
 SnowflakeAIModelContext.Attributes.update_forward_refs(**localns)
 DatabricksAIModelContext.Attributes.update_forward_refs(**localns)
 SnowflakeDynamicTable.Attributes.update_forward_refs(**localns)
+DatabricksMetricView.Attributes.update_forward_refs(**localns)
 BigqueryRoutine.Attributes.update_forward_refs(**localns)
 DatabricksVolume.Attributes.update_forward_refs(**localns)
 DatabricksVolumePath.Attributes.update_forward_refs(**localns)