iop
/
superset
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(load examples): load examples creating unnecessary main database (#12351) 

* fix(load examples): load examples creating unnecessary main database connection

* fix lint
This commit is contained in:
Daniel Vaz Gaspar 2021-01-08 21:57:35 +00:00 committed by GitHub
parent 1d2789944c
commit f354bb3d1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
superset/datasets/commands/importers/v1/utils.py
View File

@ -30,7 +30,7 @@ from sqlalchemy.sql.visitors import VisitableType
from superset.connectors.sqla.models import SqlaTable
from superset.models.core import Database
from superset.utils.core import get_example_database, get_main_database
from superset.utils.core import get_example_database
logger = logging.getLogger(__name__)
@ -136,7 +136,7 @@ def load_data(
df[column_name] = pd.to_datetime(df[column_name])
# reuse session when loading data if possible, to make import atomic
if example_database.sqlalchemy_uri == get_main_database().sqlalchemy_uri:
if example_database.sqlalchemy_uri == get_example_database().sqlalchemy_uri:
logger.info("Loading data inside the import transaction")
connection = session.connection()
else: