chore: use shillelagh instead of gsheetsdb (#13185)

* chore: use shillelagh instead of gsheetsdb

* Fix tests

* Clean up code and remove duplication

* Fix test

* Tighten dep
This commit is contained in:
Beto Dealmeida
2021-02-18 09:48:18 -08:00
committed by GitHub
parent 3c58fc5ef5
commit 3d23adec5e
12 changed files with 70 additions and 43 deletions

View File

@@ -31,8 +31,8 @@ from superset.databases.commands.exceptions import (
DatabaseTestConnectionUnexpectedError,
)
from superset.databases.dao import DatabaseDAO
from superset.exceptions import SupersetSecurityException
from superset.models.core import Database
from superset.security.analytics_db_safety import DBSecurityException
logger = logging.getLogger(__name__)
@@ -70,7 +70,7 @@ class TestConnectionDatabaseCommand(BaseCommand):
)
except DBAPIError:
raise DatabaseTestConnectionFailedError()
except DBSecurityException as ex:
except SupersetSecurityException as ex:
raise DatabaseSecurityUnsafeError(message=str(ex))
except Exception:
raise DatabaseTestConnectionUnexpectedError()