chore(dao): Remove redundant convenience methods (#24967)

This commit is contained in:
John Bodley 2023-08-14 12:07:04 -07:00 committed by GitHub
parent 12f2b0f8e7
commit 0879b9215c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 56 deletions

View File

@ -301,31 +301,6 @@ class DatasetDAO(BaseDAO[SqlaTable]): # pylint: disable=too-many-public-methods
.one_or_none()
)
@classmethod
def update_column(
cls,
model: TableColumn,
properties: dict[str, Any],
commit: bool = True,
) -> TableColumn:
return DatasetColumnDAO.update(model, properties, commit=commit)
@classmethod
def create_column(
cls, properties: dict[str, Any], commit: bool = True
) -> TableColumn:
"""
Creates a Dataset model on the metadata DB
"""
return DatasetColumnDAO.create(attributes=properties, commit=commit)
@classmethod
def delete_column(cls, model: TableColumn, commit: bool = True) -> None:
"""
Deletes a Dataset column
"""
DatasetColumnDAO.delete(model, commit=commit)
@classmethod
def find_dataset_metric(cls, dataset_id: int, metric_id: int) -> SqlMetric | None:
# We want to apply base dataset filters
@ -334,33 +309,6 @@ class DatasetDAO(BaseDAO[SqlaTable]): # pylint: disable=too-many-public-methods
return None
return db.session.query(SqlMetric).get(metric_id)
@classmethod
def delete_metric(cls, model: SqlMetric, commit: bool = True) -> None:
"""
Deletes a Dataset metric
"""
DatasetMetricDAO.delete(model, commit=commit)
@classmethod
def update_metric(
cls,
model: SqlMetric,
properties: dict[str, Any],
commit: bool = True,
) -> SqlMetric:
return DatasetMetricDAO.update(model, properties, commit=commit)
@classmethod
def create_metric(
cls,
properties: dict[str, Any],
commit: bool = True,
) -> SqlMetric:
"""
Creates a Dataset model on the metadata DB
"""
return DatasetMetricDAO.create(attributes=properties, commit=commit)
@classmethod
def delete(
cls,

View File

@ -20,7 +20,7 @@ from typing import Optional
from superset import security_manager
from superset.commands.base import BaseCommand
from superset.connectors.sqla.models import TableColumn
from superset.daos.dataset import DatasetDAO
from superset.daos.dataset import DatasetColumnDAO, DatasetDAO
from superset.daos.exceptions import DAODeleteFailedError
from superset.datasets.columns.commands.exceptions import (
DatasetColumnDeleteFailedError,
@ -43,7 +43,7 @@ class DeleteDatasetColumnCommand(BaseCommand):
assert self._model
try:
DatasetDAO.delete_column(self._model)
DatasetColumnDAO.delete(self._model)
except DAODeleteFailedError as ex:
logger.exception(ex.exception)
raise DatasetColumnDeleteFailedError() from ex

View File

@ -20,7 +20,7 @@ from typing import Optional
from superset import security_manager
from superset.commands.base import BaseCommand
from superset.connectors.sqla.models import SqlMetric
from superset.daos.dataset import DatasetDAO
from superset.daos.dataset import DatasetDAO, DatasetMetricDAO
from superset.daos.exceptions import DAODeleteFailedError
from superset.datasets.metrics.commands.exceptions import (
DatasetMetricDeleteFailedError,
@ -43,7 +43,7 @@ class DeleteDatasetMetricCommand(BaseCommand):
assert self._model
try:
DatasetDAO.delete_metric(self._model)
DatasetMetricDAO.delete(self._model)
except DAODeleteFailedError as ex:
logger.exception(ex.exception)
raise DatasetMetricDeleteFailedError() from ex