Add enums to support QuickSight and Kafka assets

Author: ErnestoLoma

.github/workflows/pytest.yaml

@@ -30,6 +30,6 @@ jobs:
       - name: Test with pytest
         env: # Or as an environment variable
           ATLAN_API_KEY: ${{ secrets.MARK_ATLAN_API_KEY }}
-          ATLAN_HOST: https://mark.atlan.com
+          ATLAN_BASE_URL: https://mark.atlan.com
         run: |
           pytest

pyatlan/generator/generate_from_typdefs.py

@@ -40,6 +40,12 @@
     ("query_username_strategy", "QueryUsernameStrategy"),
     ("google_datastudio_asset_type", "GoogleDatastudioAssetType"),
     ("powerbi_endorsement", "PowerbiEndorsement"),
+    ("kafka_topic_compression_type", "KafkaTopicCompressionType"),
+    ("kafka_topic_cleanup_policy", "PowerbiEndorsement"),
+    ("quick_sight_folder_type", "QuickSightFolderType"),
+    ("quick_sight_dataset_field_type", "QuickSightDatasetFieldType"),
+    ("quick_sight_analysis_status", "QuickSightAnalysisStatus"),
+    ("quick_sight_dataset_import_mode", "QuickSightDatasetImportMode"),
 ]
 ARRAY_REPLACEMENTS = [("array<string>", "set{string}")]
 

pyatlan/generator/templates/entity.jinja2

@@ -27,6 +27,11 @@ from pyatlan.model.enums import (
     CertificateStatus,
     EntityStatus,
     IconType,
+    KafkaTopicCompressionType,
+    QuickSightFolderType,
+    QuickSightDatasetFieldType,
+    QuickSightAnalysisStatus,
+    QuickSightDatasetImportMode,
     QueryUsernameStrategy,
     SourceCostUnitType,
     GoogleDatastudioAssetType,
@@ -42,6 +47,7 @@ from pyatlan.model.structs import (
     GoogleLabel,
     GoogleTag,
     Histogram,
+    KafkaTopicConsumption,
     PopularityInsights,
 )
 from pyatlan.utils import next_id