Merge pull request #388 from atlanhq/DVX-647

DVX-647: (temp) Fixed class generator by adding core assets exclusion

docs/asset/anomalo.rst

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

docs/asset/anomalocheck.rst

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

docs/asset/cosmosmongodbaccount.rst

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

docs/asset/model.rst

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

docs/asset/modelattribute.rst

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

docs/asset/modelattributeassociation.rst

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

docs/asset/modeldatamodel.rst

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

docs/asset/modelentity.rst

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

docs/asset/modelentityassociation.rst

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

docs/asset/modelversion.rst

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

docs/assets.rst

@@ -20,9 +20,8 @@ You can interact with all of the following different kinds of assets:
     asset/airflow
     asset/airflowdag
     asset/airflowtask
-    asset/app
-    asset/appapplication
-    asset/appcomponent
+    asset/anomalo
+    asset/anomalocheck
     asset/asset
     asset/atlasglossary
     asset/atlasglossarycategory
@@ -66,6 +65,7 @@ You can interact with all of the following different kinds of assets:
     asset/columnprocess
     asset/connection
     asset/cosmosmongodb
+    asset/cosmosmongodbaccount
     asset/cosmosmongodbcollection
     asset/cosmosmongodbdatabase
     asset/cube
@@ -163,6 +163,13 @@ You can interact with all of the following different kinds of assets:
     asset/modequery
     asset/modereport
     asset/modeworkspace
+    asset/model
+    asset/modelattribute
+    asset/modelattributeassociation
+    asset/modeldatamodel
+    asset/modelentity
+    asset/modelentityassociation
+    asset/modelversion
     asset/mongodb
     asset/mongodbcollection
     asset/mongodbdatabase

pyatlan/generator/class_generator.py

@@ -54,6 +54,7 @@
     ("atlas_operation", "AtlasOperation"),
     ("matillion_job_type", "MatillionJobType"),
     ("incident_severity", "IncidentSeverity"),
+    ("table_type", "TableType"),
 ]
 PRIMITIVE_MAPPINGS = {
     "string": "str",
@@ -215,8 +216,11 @@ class AssetInfo:
         "Referenceable",
         "Asset",
         "AuthPolicy",
-        "DataModel",
-        "DataModeling",
+        "PowerBIDatasource",
+        "PowerBIReport",
+        "PowerBITile",
+        "PowerBIColumn",
+        "PowerBIMeasure",
         "MatillionGroup",
         "Stakeholder",
         "StakeholderTitle",

pyatlan/generator/templates/imports.jinja2

@@ -68,6 +68,7 @@ from pyatlan.model.enums import (
     SchemaRegistrySchemaCompatibility,
     SchemaRegistrySchemaType,
     SourceCostUnitType,
+    table_type,
 )
 from pyatlan.model.fields.atlan_fields import (
     BooleanField,
@@ -102,6 +103,7 @@ from pyatlan.model.structs import (
     PopularityInsights,
     SourceTagAttribute,
     StarredDetails,
+    DbtJobRun,
 )
 from pyatlan.utils import (
     init_guid,

pyatlan/model/assets/__init__.py

@@ -21,8 +21,8 @@
         "Folder",
         "Airflow",
         "DataContract",
-        "App",
         "DataQuality",
+        "BI",
         "Resource",
         "DataMesh",
         "SQL",
@@ -34,11 +34,11 @@
         "Stakeholder",
         "AirflowDag",
         "AirflowTask",
-        "AppApplication",
-        "AppComponent",
+        "Anomalo",
         "MonteCarlo",
         "Metric",
         "Soda",
+        "PowerBI",
         "Readme",
         "File",
         "Link",
@@ -70,9 +70,21 @@
         "DbtSource",
         "SparkJob",
         "SchemaRegistrySubject",
+        "AnomaloCheck",
         "MCIncident",
         "MCMonitor",
         "SodaCheck",
+        "PowerBIReport",
+        "PowerBIMeasure",
+        "PowerBIColumn",
+        "PowerBITable",
+        "PowerBITile",
+        "PowerBIDatasource",
+        "PowerBIWorkspace",
+        "PowerBIDataset",
+        "PowerBIDashboard",
+        "PowerBIDataflow",
+        "PowerBIPage",
         "SnowflakeDynamicTable",
         "DynamoDBSecondaryIndex",
     ],
@@ -95,12 +107,12 @@
     "purpose": ["Purpose"],
     "collection": ["Collection"],
     "object_store": ["ObjectStore"],
-    "b_i": ["BI"],
     "saa_s": ["SaaS"],
     "d_m": ["DM"],
     "multi_dimensional_dataset": ["MultiDimensionalDataset"],
     "event_store": ["EventStore"],
     "no_s_q_l": ["NoSQL"],
+    "model": ["Model"],
     "insight": ["Insight"],
     "a_p_i": ["API"],
     "google": ["Google"],
@@ -123,7 +135,6 @@
     "metabase": ["Metabase"],
     "quick_sight": ["QuickSight"],
     "thoughtspot": ["Thoughtspot"],
-    "power_b_i": ["PowerBI"],
     "micro_strategy": ["MicroStrategy"],
     "cognos": ["Cognos"],
     "superset": ["Superset"],
@@ -147,6 +158,12 @@
     "dynamo_d_b": ["DynamoDB"],
     "mongo_d_b": ["MongoDB"],
     "dbt_tag": ["DbtTag"],
+    "model_attribute": ["ModelAttribute"],
+    "model_entity": ["ModelEntity"],
+    "model_version": ["ModelVersion"],
+    "model_entity_association": ["ModelEntityAssociation"],
+    "model_attribute_association": ["ModelAttributeAssociation"],
+    "model_data_model": ["ModelDataModel"],
     "a_p_i_spec": ["APISpec"],
     "a_p_i_path": ["APIPath"],
     "data_studio_asset": ["DataStudioAsset"],
@@ -221,17 +238,6 @@
     "thoughtspot_view": ["ThoughtspotView"],
     "thoughtspot_dashlet": ["ThoughtspotDashlet"],
     "thoughtspot_answer": ["ThoughtspotAnswer"],
-    "power_b_i_report": ["PowerBIReport"],
-    "power_b_i_measure": ["PowerBIMeasure"],
-    "power_b_i_column": ["PowerBIColumn"],
-    "power_b_i_table": ["PowerBITable"],
-    "power_b_i_tile": ["PowerBITile"],
-    "power_b_i_datasource": ["PowerBIDatasource"],
-    "power_b_i_workspace": ["PowerBIWorkspace"],
-    "power_b_i_dataset": ["PowerBIDataset"],
-    "power_b_i_dashboard": ["PowerBIDashboard"],
-    "power_b_i_dataflow": ["PowerBIDataflow"],
-    "power_b_i_page": ["PowerBIPage"],
     "micro_strategy_report": ["MicroStrategyReport"],
     "micro_strategy_project": ["MicroStrategyProject"],
     "micro_strategy_metric": ["MicroStrategyMetric"],
@@ -275,6 +281,7 @@
     "kafka_consumer_group": ["KafkaConsumerGroup"],
     "azure_service_bus_namespace": ["AzureServiceBusNamespace"],
     "azure_service_bus_topic": ["AzureServiceBusTopic"],
+    "cosmos_mongo_d_b_account": ["CosmosMongoDBAccount"],
     "cosmos_mongo_d_b_collection": ["CosmosMongoDBCollection"],
     "cosmos_mongo_d_b_database": ["CosmosMongoDBDatabase"],
     "qlik_stream": ["QlikStream"],