Skip to content

Commit cbafaee

Browse files
committed
Verify audit migration before and after upgrade
1 parent b68f7ce commit cbafaee

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

‎src/openstatspec/sql/inplace_transform.py‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -571,7 +571,14 @@ def install_in_place_transformation_schema(
571571
engine = create_engine(database_url)
572572
try:
573573
with engine.begin() as connection:
574-
require_verified_catalog(connection)
574+
require_verified_catalog(
575+
connection,
576+
allowed_migrations={
577+
"transformation_apply": {
578+
"source_kind", "frontend_contract",
579+
},
580+
},
581+
)
575582
apply_audit_catalog(MetaData()).create(connection, checkfirst=True)
576583
columns = {
577584
str(column["name"])
@@ -588,6 +595,7 @@ def install_in_place_transformation_schema(
588595
f"ALTER TABLE {quote('transformation_apply')} "
589596
f"ADD COLUMN {quote(name)} {sql_type}"
590597
)
598+
require_verified_catalog(connection)
591599
finally:
592600
engine.dispose()
593601

0 commit comments

Comments
 (0)