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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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