fix(dataset): handle missing database in migration (#18948)

This commit is contained in:
Ville Brofeldt
2022-02-25 20:50:55 +02:00
committed by GitHub
parent 147dc5af71
commit 2baceddbf1

View File

@@ -321,8 +321,10 @@ def after_insert(target: SqlaTable) -> None: # pylint: disable=too-many-locals
# table names
database = (
target.database
or session.query(Database).filter_by(id=target.database_id).one()
or session.query(Database).filter_by(id=target.database_id).first()
)
if not database:
return
engine = database.get_sqla_engine(schema=target.schema)
conditional_quote = engine.dialect.identifier_preparer.quote