Add Notification asset model and related enums

pyatlan/generator/templates/imports.jinja2

@@ -140,6 +140,7 @@ from pyatlan.model.structs import (
     AssetExternalDQTestsByStatus,
     AwsCloudWatchMetric,
     DataQualityRuleThresholdObject,
+    NotificationExternalReference,
     SQLProcedureAdditionalProperties,
     SQLProcedureReturnType,
     SageMakerUnifiedStudioAssetFilter,

pyatlan/generator/templates/structs.jinja2

@@ -17,6 +17,7 @@ from pyatlan.model.enums import (
     DataQualityRuleThresholdUnit,
     AppWorkflowRunStatus,
     AssetSmusMetadataFormStatus,
+    NotificationSurface,
 )
 from pyatlan.model.utils import to_camel_case
 from pyatlan.utils import select_optional_set_fields, validate_required_fields

pyatlan/model/assets/__init__.py

@@ -9,6 +9,7 @@
         "Process",
         "AtlasGlossaryCategory",
         "AccessControl",
+        "Notification",
         "AuthPolicy",
         "StakeholderTitle",
         "Catalog",
@@ -23,6 +24,7 @@
         "V1CoalesceProcess",
         "ColumnProcess",
         "Persona",
+        "WorkflowNotification",
         "App",
         "Airflow",
         "ADF",
@@ -117,8 +119,8 @@
         "Snowflake",
         "DatabricksUnityCatalogTag",
         "SnowflakeStream",
-        "Database",
         "CalculationView",
+        "Database",
         "Procedure",
         "Table",
         "SnowflakePipe",

pyatlan/model/assets/__init__.pyi

@@ -6,6 +6,7 @@ __all__ = [
     "Process",
     "AtlasGlossaryCategory",
     "AccessControl",
+    "Notification",
     "AuthPolicy",
     "StakeholderTitle",
     "Catalog",
@@ -20,6 +21,7 @@ __all__ = [
     "V1CoalesceProcess",
     "ColumnProcess",
     "Persona",
+    "WorkflowNotification",
     "App",
     "Airflow",
     "ADF",
@@ -114,8 +116,8 @@ __all__ = [
     "Snowflake",
     "DatabricksUnityCatalogTag",
     "SnowflakeStream",
-    "Database",
     "CalculationView",
+    "Database",
     "Procedure",
     "Table",
     "SnowflakePipe",
@@ -234,7 +236,6 @@ __all__ = [
     "Unstructured",
     "V1Coalesce",
     "EventStore",
-    "NoSQL",
     "Insight",
     "AssetGrouping",
     "ObjectStore",
@@ -724,6 +725,7 @@ 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.notification import Notification
 from .core.partial import Partial
 from .core.partial_field import PartialField
 from .core.partial_object import PartialObject
@@ -796,6 +798,7 @@ from .core.table_partition import TablePartition
 from .core.tag import Tag
 from .core.v1_coalesce_process import V1CoalesceProcess
 from .core.view import View
+from .core.workflow_notification import WorkflowNotification
 from .cube import Cube
 from .cube_dimension import CubeDimension
 from .cube_field import CubeField

pyatlan/model/assets/core/__init__.py

@@ -140,6 +140,7 @@
 from .monte_carlo import MonteCarlo
 from .namespace import Namespace
 from .no_s_q_l import NoSQL
+from .notification import Notification
 from .partial import Partial
 from .partial_field import PartialField
 from .partial_object import PartialObject
@@ -211,6 +212,7 @@
 from .tag import Tag
 from .v1_coalesce_process import V1CoalesceProcess
 from .view import View
+from .workflow_notification import WorkflowNotification
 
 # Update asset forward references:
 localns = locals()
@@ -222,6 +224,7 @@
 Process.Attributes.update_forward_refs(**localns)
 AtlasGlossaryCategory.Attributes.update_forward_refs(**localns)
 AccessControl.Attributes.update_forward_refs(**localns)
+Notification.Attributes.update_forward_refs(**localns)
 AuthPolicy.Attributes.update_forward_refs(**localns)
 StakeholderTitle.Attributes.update_forward_refs(**localns)
 Catalog.Attributes.update_forward_refs(**localns)
@@ -236,6 +239,7 @@
 V1CoalesceProcess.Attributes.update_forward_refs(**localns)
 ColumnProcess.Attributes.update_forward_refs(**localns)
 Persona.Attributes.update_forward_refs(**localns)
+WorkflowNotification.Attributes.update_forward_refs(**localns)
 App.Attributes.update_forward_refs(**localns)
 Airflow.Attributes.update_forward_refs(**localns)
 ADF.Attributes.update_forward_refs(**localns)
@@ -329,8 +333,8 @@
 Snowflake.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)
 Table.Attributes.update_forward_refs(**localns)
 SnowflakePipe.Attributes.update_forward_refs(**localns)

pyatlan/model/assets/core/asset.py

@@ -1063,13 +1063,13 @@ def __setattr__(self, name, value):
         "assetSodaLastSyncRunAt", "assetSodaLastSyncRunAt"
     )
     """
-    
+
     """
     ASSET_SODA_LAST_SCAN_AT: ClassVar[NumericField] = NumericField(
         "assetSodaLastScanAt", "assetSodaLastScanAt"
     )
     """
-    
+
     """
     ASSET_SODA_CHECK_STATUSES: ClassVar[TextField] = TextField(
         "assetSodaCheckStatuses", "assetSodaCheckStatuses"
@@ -1081,7 +1081,7 @@ def __setattr__(self, name, value):
         "assetSodaSourceURL", "assetSodaSourceURL"
     )
     """
-    
+
     """
     ASSET_ICON: ClassVar[TextField] = TextField("assetIcon", "assetIcon")
     """
@@ -1147,7 +1147,7 @@ def __setattr__(self, name, value):
         "isAIGenerated", "isAIGenerated"
     )
     """
-    
+
     """
     ASSET_COVER_IMAGE: ClassVar[TextField] = TextField(
         "assetCoverImage", "assetCoverImage"

pyatlan/model/assets/core/atlan_app_workflow.py

@@ -107,6 +107,12 @@ def __setattr__(self, name, value):
     """
     TBC
     """
+    WORKFLOW_NOTIFICATION_NOTIFICATIONS: ClassVar[RelationField] = RelationField(
+        "workflowNotificationNotifications"
+    )
+    """
+    TBC
+    """
     ATLAN_APP_WORKFLOW_RUNS: ClassVar[RelationField] = RelationField(
         "atlanAppWorkflowRuns"
     )
@@ -127,6 +133,7 @@ def __setattr__(self, name, value):
         "atlan_app_workflow_deployment_name",
         "atlan_app_workflow_runtime_mode",
         "atlan_app",
+        "workflow_notification_notifications",
         "atlan_app_workflow_runs",
     ]
 
@@ -310,6 +317,26 @@ def atlan_app(self, atlan_app: Optional[AtlanApp]):
             self.attributes = self.Attributes()
         self.attributes.atlan_app = atlan_app
 
+    @property
+    def workflow_notification_notifications(
+        self,
+    ) -> Optional[List[WorkflowNotification]]:
+        return (
+            None
+            if self.attributes is None
+            else self.attributes.workflow_notification_notifications
+        )
+
+    @workflow_notification_notifications.setter
+    def workflow_notification_notifications(
+        self, workflow_notification_notifications: Optional[List[WorkflowNotification]]
+    ):
+        if self.attributes is None:
+            self.attributes = self.Attributes()
+        self.attributes.workflow_notification_notifications = (
+            workflow_notification_notifications
+        )
+
     @property
     def atlan_app_workflow_runs(self) -> Optional[List[AppWorkflowRun]]:
         return (
@@ -353,6 +380,9 @@ class Attributes(AtlanApp.Attributes):
         atlan_app: Optional[AtlanApp] = Field(
             default=None, description=""
         )  # relationship
+        workflow_notification_notifications: Optional[List[WorkflowNotification]] = (
+            Field(default=None, description="")
+        )  # relationship
         atlan_app_workflow_runs: Optional[List[AppWorkflowRun]] = Field(
             default=None, description=""
         )  # relationship
@@ -369,3 +399,4 @@ class Attributes(AtlanApp.Attributes):
 
 from .app_workflow_run import AppWorkflowRun  # noqa: E402, F401
 from .atlan_app import AtlanApp  # noqa: E402, F401
+from .workflow_notification import WorkflowNotification  # noqa: E402, F401