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 import ImportDatasetsCommand
from superset.datasets.commands.importers.v1.utils import import_dataset from superset.datasets.commands.importers.v1.utils import import_dataset
from superset.datasets.schemas import ImportV1DatasetSchema from superset.datasets.schemas import ImportV1DatasetSchema
from superset.models.core import Database
from superset.models.dashboard import dashboard_slices from superset.models.dashboard import dashboard_slices
from superset.utils.core import get_example_database
class ImportExamplesCommand(ImportModelsCommand): 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 # 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. # 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. # We need to determine its ID so we can point the dataset to it.
examples_db = ( examples_db = get_example_database()
db.session.query(Database).filter_by(database_name="examples").first()
)
dataset_info: Dict[str, Dict[str, Any]] = {} dataset_info: Dict[str, Dict[str, Any]] = {}
for file_name, config in configs.items(): for file_name, config in configs.items():
if file_name.startswith("datasets/"): if file_name.startswith("datasets/"):

View File

@ -53,7 +53,7 @@ slice_user = Table(
logger = logging.getLogger(__name__) 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 Model, AuditMixinNullable, ImportExportMixin
): ):
"""A slice is essentially a report or a view on data""" """A slice is essentially a report or a view on data"""