mirror of https://github.com/apache/superset.git
fix: datasourcedao datasource not found error (#20725)
* first fix * remove prints * run pc * fix linter * commit
This commit is contained in:
parent
84d4302628
commit
1d9d505a6e
|
@ -15,6 +15,7 @@
|
||||||
# specific language governing permissions and limitations
|
# specific language governing permissions and limitations
|
||||||
# under the License.
|
# under the License.
|
||||||
|
|
||||||
|
import logging
|
||||||
from typing import Dict, Type, Union
|
from typing import Dict, Type, Union
|
||||||
|
|
||||||
from sqlalchemy.orm import Session
|
from sqlalchemy.orm import Session
|
||||||
|
@ -27,6 +28,8 @@ from superset.models.sql_lab import Query, SavedQuery
|
||||||
from superset.tables.models import Table
|
from superset.tables.models import Table
|
||||||
from superset.utils.core import DatasourceType
|
from superset.utils.core import DatasourceType
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
Datasource = Union[Dataset, SqlaTable, Table, Query, SavedQuery]
|
Datasource = Union[Dataset, SqlaTable, Table, Query, SavedQuery]
|
||||||
|
|
||||||
|
|
||||||
|
@ -57,6 +60,11 @@ class DatasourceDAO(BaseDAO):
|
||||||
)
|
)
|
||||||
|
|
||||||
if not datasource:
|
if not datasource:
|
||||||
|
logger.warning(
|
||||||
|
"Datasource not found datasource_type: %s, datasource_id: %s",
|
||||||
|
datasource_type,
|
||||||
|
datasource_id,
|
||||||
|
)
|
||||||
raise DatasourceNotFound()
|
raise DatasourceNotFound()
|
||||||
|
|
||||||
return datasource
|
return datasource
|
||||||
|
|
|
@ -815,7 +815,7 @@ class Superset(BaseSupersetView): # pylint: disable=too-many-public-methods
|
||||||
try:
|
try:
|
||||||
datasource = DatasourceDAO.get_datasource(
|
datasource = DatasourceDAO.get_datasource(
|
||||||
db.session,
|
db.session,
|
||||||
DatasourceType(cast(str, datasource_type)),
|
DatasourceType("table"),
|
||||||
datasource_id,
|
datasource_id,
|
||||||
)
|
)
|
||||||
except DatasetNotFoundError:
|
except DatasetNotFoundError:
|
||||||
|
|
Loading…
Reference in New Issue