Skip to content
Snippets Groups Projects
Commit f357353f authored by Leah Tacke genannt Unterberg's avatar Leah Tacke genannt Unterberg
Browse files

minor fixes

parent 42d9e355
Branches
No related tags found
No related merge requests found
......@@ -13,7 +13,7 @@ sync:
build:
uv build
publish
publish:
uv publish
requirements:
......
from typing import Self
from uuid import UUID
import pydantic
from mitm_tooling.transformation.superset.definitions import DatasetIdentifierMap, \
DatabaseIdentifier, MitMDatasetIdentifier, BaseSupersetDefinition, DashboardIdentifier
DatabaseIdentifier, MitMDatasetIdentifier, BaseSupersetDefinition, DashboardIdentifier, SupersetMitMDatasetDef
VizDashboardIdentifierMap = dict[str, DashboardIdentifier]
VizCollectionIdentifierMap = dict[str, VizDashboardIdentifierMap]
......@@ -18,6 +19,11 @@ class DatasourceIdentifierBundle(BaseSupersetDefinition):
if self.database is not None:
return self.database.uuid
@classmethod
def from_mitm_dataset(cls, mitm_dataset: SupersetMitMDatasetDef) -> Self:
return cls(database=DatabaseIdentifier(uuid=mitm_dataset.database_uuid),
ds_id_map={t.table_name: t for t in (mitm_dataset.tables or [])})
class MitMDatasetIdentifierBundle(DatasourceIdentifierBundle):
mitm_dataset: MitMDatasetIdentifier | None = None
......@@ -27,3 +33,12 @@ class MitMDatasetIdentifierBundle(DatasourceIdentifierBundle):
def mitm_dataset_uuid(self) -> UUID | None:
if self.mitm_dataset is not None:
return self.mitm_dataset.uuid
@classmethod
def from_mitm_dataset(cls, mitm_dataset: SupersetMitMDatasetDef) -> Self:
return cls(mitm_dataset=mitm_dataset.identifier,
database=DatabaseIdentifier(uuid=mitm_dataset.database_uuid),
ds_id_map={t.table_name: t for t in (mitm_dataset.tables or [])},
viz_id_map={
'default': {dash.dashboard_title: dash for dash in
(mitm_dataset.dashboards or [])}})
from mitm_tooling.definition import MITM
from mitm_tooling.representation import Header
from mitm_tooling.transformation.superset.definitions import SupersetDefFile, StrUUID, DatasetIdentifier, \
from ..definitions import SupersetDefFile, StrUUID, DatasetIdentifier, \
DashboardIdentifier, ChartIdentifier, \
MitMDatasetIdentifier, SupersetObjectMixin
......
[project]
name = "mitm-tooling"
version = "0.5.0"
version = "0.5.1"
description = ""
authors = [{ name = "Leah Tacke genannt Unterberg", email = "l.tgu@pads.rwth-aachen.de" }]
requires-python = ">=3.11,<3.14"
......@@ -30,16 +30,6 @@ dev = [
[tool.uv]
[[tool.uv.index]]
name = "testpypi"
url = "https://test.pypi.org/"
explicit = true
[[tool.uv.index]]
name = "PyPI"
url = "https://pypi.org/simple/"
default = true
[tool.hatch.build.targets.sdist]
include = ["mitm_tooling"]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment