aligned related objects to superset import

mitm_tooling/transformation/superset/definitions/mitm_dataset.py

@@ -10,19 +10,21 @@ class MitMDatasetIdentifier(BaseSupersetDefinition):
     uuid: StrUUID | None = None
 
 
-class RelatedTable(BaseSupersetDefinition):
-    table_id: SupersetId | None = None
-    table_uuid: StrUUID
+class RelatedObjectIdentifier(BaseSupersetDefinition):
+    id: SupersetId | None = None
+    uuid: StrUUID
+
+
+class RelatedTable(RelatedObjectIdentifier):
+    pass
 
 
-class RelatedSlice(BaseSupersetDefinition):
-    slice_id: SupersetId | None = None
-    slice_uuid: StrUUID
+class RelatedSlice(RelatedObjectIdentifier):
+    pass
 
 
-class RelatedDashboard(BaseSupersetDefinition):
-    dashboard_id: SupersetId | None = None
-    dashboard_uuid: StrUUID
+class RelatedDashboard(RelatedObjectIdentifier):
+    pass
 
 
 class SupersetMitMDatasetDef(SupersetDefFile):

mitm_tooling/transformation/superset/factories/mitm_dataset.py

@@ -12,11 +12,11 @@ def mk_related_obj(kind: Literal['table', 'slice', 'dashboard'],
                    uuid: UUID) -> RelatedTable | RelatedSlice | RelatedDashboard | None:
     match kind:
         case 'table':
-            return RelatedTable(table_uuid=uuid)
+            return RelatedTable(uuid=uuid)
         case 'slice':
-            return RelatedSlice(slice_uuid=uuid)
+            return RelatedSlice(uuid=uuid)
         case 'dashboard':
-            return RelatedDashboard(dashboard_uuid=uuid)
+            return RelatedDashboard(uuid=uuid)
 
 
 def mk_related_objs(kind: Literal['table', 'slice', 'dashboard'], uuids: Iterable[UUID]) -> Iterable[RelatedTable] | \