fix: create example DB if needed (#16451)

* fix: create example DB if needed

* fix lint
This commit is contained in:
Beto Dealmeida 2021-08-27 07:19:37 -07:00 committed by GitHub
parent 147637a02d
commit 62d8ab7f9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 5 deletions

View File

@ -43,8 +43,8 @@ from superset.databases.schemas import ImportV1DatabaseSchema
from superset.datasets.commands.importers.v1 import ImportDatasetsCommand
from superset.datasets.commands.importers.v1.utils import import_dataset
from superset.datasets.schemas import ImportV1DatasetSchema
from superset.models.core import Database
from superset.models.dashboard import dashboard_slices
from superset.utils.core import get_example_database
class ImportExamplesCommand(ImportModelsCommand):
@ -104,9 +104,7 @@ class ImportExamplesCommand(ImportModelsCommand):
# If database_uuid is not in the list of UUIDs it means that the examples
# database was created before its UUID was frozen, so it has a random UUID.
# We need to determine its ID so we can point the dataset to it.
examples_db = (
db.session.query(Database).filter_by(database_name="examples").first()
)
examples_db = get_example_database()
dataset_info: Dict[str, Dict[str, Any]] = {}
for file_name, config in configs.items():
if file_name.startswith("datasets/"):

View File

@ -53,7 +53,7 @@ slice_user = Table(
logger = logging.getLogger(__name__)
class Slice( # pylint: disable=too-many-public-methods
class Slice( # pylint: disable=too-many-public-methods, too-many-instance-attributes
Model, AuditMixinNullable, ImportExportMixin
):
"""A slice is essentially a report or a view on data"""