Merge pull request #505 from atlanhq/APP-4750

APP-4750: Add creators for Amazon `QuickSight` assets

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

@@ -0,0 +1,43 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+    ) -> QuickSightAnalysis: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_analysis_folders: List[str],
+    ) -> QuickSightAnalysis: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_analysis_folders: Optional[List[str]] = None,
+    ) -> QuickSightAnalysis:
+        validate_required_fields(
+            ["name", "connection_qualified_name", "quick_sight_id"],
+            [name, connection_qualified_name, quick_sight_id],
+        )
+        attributes = QuickSightAnalysis.Attributes.creator(
+            name=name,
+            connection_qualified_name=connection_qualified_name,
+            quick_sight_id=quick_sight_id,
+            quick_sight_analysis_folders=quick_sight_analysis_folders,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,63 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_sheet_id: str,
+        quick_sight_sheet_name: str,
+        quick_sight_analysis_qualified_name: str,
+    ) -> QuickSightAnalysisVisual: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_sheet_id: str,
+        quick_sight_sheet_name: str,
+        quick_sight_analysis_qualified_name: str,
+        connection_qualified_name: str,
+    ) -> QuickSightAnalysisVisual: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_sheet_id: str,
+        quick_sight_sheet_name: str,
+        quick_sight_analysis_qualified_name: str,
+        connection_qualified_name: Optional[str] = None,
+    ) -> QuickSightAnalysisVisual:
+        validate_required_fields(
+            [
+                "name",
+                "quick_sight_id",
+                "quick_sight_sheet_name",
+                "quick_sight_sheet_id",
+                "quick_sight_analysis_qualified_name",
+            ],
+            [
+                name,
+                quick_sight_id,
+                quick_sight_sheet_name,
+                quick_sight_sheet_id,
+                quick_sight_analysis_qualified_name,
+            ],
+        )
+        attributes = QuickSightAnalysisVisual.Attributes.creator(
+            name=name,
+            quick_sight_id=quick_sight_id,
+            quick_sight_sheet_id=quick_sight_sheet_id,
+            quick_sight_sheet_name=quick_sight_sheet_name,
+            quick_sight_analysis_qualified_name=quick_sight_analysis_qualified_name,
+            connection_qualified_name=connection_qualified_name,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,43 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+    ) -> QuickSightDashboard: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_dashboard_folders: List[str],
+    ) -> QuickSightDashboard: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_dashboard_folders: Optional[List[str]] = None,
+    ) -> QuickSightDashboard:
+        validate_required_fields(
+            ["name", "connection_qualified_name", "quick_sight_id"],
+            [name, connection_qualified_name, quick_sight_id],
+        )
+        attributes = QuickSightDashboard.Attributes.creator(
+            name=name,
+            connection_qualified_name=connection_qualified_name,
+            quick_sight_id=quick_sight_id,
+            quick_sight_dashboard_folders=quick_sight_dashboard_folders,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,63 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_sheet_id: str,
+        quick_sight_sheet_name: str,
+        quick_sight_dashboard_qualified_name: str,
+    ) -> QuickSightDashboardVisual: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_sheet_id: str,
+        quick_sight_sheet_name: str,
+        quick_sight_dashboard_qualified_name: str,
+        connection_qualified_name: str,
+    ) -> QuickSightDashboardVisual: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_sheet_id: str,
+        quick_sight_sheet_name: str,
+        quick_sight_dashboard_qualified_name: str,
+        connection_qualified_name: Optional[str] = None,
+    ) -> QuickSightDashboardVisual:
+        validate_required_fields(
+            [
+                "name",
+                "quick_sight_id",
+                "quick_sight_sheet_id",
+                "quick_sight_sheet_name",
+                "quick_sight_dashboard_qualified_name",
+            ],
+            [
+                name,
+                quick_sight_id,
+                quick_sight_sheet_id,
+                quick_sight_sheet_name,
+                quick_sight_dashboard_qualified_name,
+            ],
+        )
+        attributes = QuickSightDashboardVisual.Attributes.creator(
+            name=name,
+            quick_sight_id=quick_sight_id,
+            quick_sight_sheet_id=quick_sight_sheet_id,
+            quick_sight_sheet_name=quick_sight_sheet_name,
+            quick_sight_dashboard_qualified_name=quick_sight_dashboard_qualified_name,
+            connection_qualified_name=connection_qualified_name,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,46 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+    ) -> QuickSightDataset: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_dataset_import_mode: QuickSightDatasetImportMode,
+        quick_sight_dataset_folders: List[str],
+    ) -> QuickSightDataset: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_dataset_import_mode: Optional[QuickSightDatasetImportMode] = None,
+        quick_sight_dataset_folders: Optional[List[str]] = None,
+    ) -> QuickSightDataset:
+        validate_required_fields(
+            ["name", "connection_qualified_name", "quick_sight_id"],
+            [name, connection_qualified_name, quick_sight_id],
+        )
+        attributes = QuickSightDataset.Attributes.creator(
+            name=name,
+            connection_qualified_name=connection_qualified_name,
+            quick_sight_id=quick_sight_id,
+            quick_sight_dataset_import_mode=quick_sight_dataset_import_mode,
+            quick_sight_dataset_folders=quick_sight_dataset_folders,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,46 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_dataset_qualified_name: str,
+        quick_sight_id: str,
+    ) -> QuickSightDatasetField: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_dataset_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_dataset_field_type: QuickSightDatasetFieldType,
+        connection_qualified_name: str,
+    ) -> QuickSightDatasetField: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        quick_sight_id: str,
+        quick_sight_dataset_qualified_name: str,
+        quick_sight_dataset_field_type: Optional[QuickSightDatasetFieldType] = None,
+        connection_qualified_name: Optional[str] = None,
+    ) -> QuickSightDatasetField:
+        validate_required_fields(
+            ["name", "quick_sight_id", "quick_sight_dataset_qualified_name"],
+            [name, quick_sight_id, quick_sight_dataset_qualified_name],
+        )
+        attributes = QuickSightDatasetField.Attributes.creator(
+            name=name,
+            quick_sight_id=quick_sight_id,
+            connection_qualified_name=connection_qualified_name,
+            quick_sight_dataset_qualified_name=quick_sight_dataset_qualified_name,
+            quick_sight_dataset_field_type=quick_sight_dataset_field_type,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,43 @@
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+    ) -> QuickSightFolder: ...
+
+    @overload
+    @classmethod
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_folder_type: QuickSightFolderType,
+    ) -> QuickSightFolder: ...
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        connection_qualified_name: str,
+        quick_sight_id: str,
+        quick_sight_folder_type: Optional[QuickSightFolderType] = None,
+    ) -> QuickSightFolder:
+        validate_required_fields(
+            ["name", "connection_qualified_name", "quick_sight_id"],
+            [name, connection_qualified_name, quick_sight_id],
+        )
+        attributes = QuickSightFolder.Attributes.creator(
+            name=name,
+            connection_qualified_name=connection_qualified_name,
+            quick_sight_id=quick_sight_id,
+            quick_sight_folder_type=quick_sight_folder_type,
+        )
+        return cls(attributes=attributes)

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

@@ -0,0 +1,32 @@
+
+        @classmethod
+        @init_guid
+        def creator(
+            cls,
+            *,
+            name: str,
+            connection_qualified_name: str,
+            quick_sight_id: str,
+            quick_sight_analysis_folders: Optional[List[str]] = None,
+        ) -> QuickSightAnalysis.Attributes:
+            validate_required_fields(
+                ["name", "connection_qualified_name", "quick_sight_id"],
+                [name, connection_qualified_name, quick_sight_id],
+            )
+            folders = None
+            if quick_sight_analysis_folders:
+                folders = [
+                    QuickSightFolder.ref_by_qualified_name(quick_sight_folder_qn)
+                    for quick_sight_folder_qn in quick_sight_analysis_folders
+                ]
+
+            return QuickSightAnalysis.Attributes(
+                name=name,
+                quick_sight_id=quick_sight_id,
+                qualified_name=f"{connection_qualified_name}/{quick_sight_id}",
+                connection_qualified_name=connection_qualified_name,
+                connector_name=AtlanConnectorType.get_connector_name(
+                    connection_qualified_name
+                ),
+                quick_sight_analysis_folders=folders,
+            )