pt
/
superset
mirror of https://github.com/apache/superset.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: update migration chart_ds_constraint to have seperate batch op (#23942)

This commit is contained in:
Hugh A. Miles II 2023-05-04 16:53:43 -04:00 committed by GitHub
parent b7689fbe53
commit f3f5d926c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
superset/migrations/versions/2023-03-27_12-30_7e67aecbf3f1_chart_ds_constraint.py
View File

@ -84,7 +84,6 @@ def upgrade_slc(slc: Slice) -> None:
def upgrade():
bind = op.get_bind()
session = db.Session(bind=bind)
with op.batch_alter_table("slices") as batch_op:
for slc in session.query(Slice).filter(Slice.datasource_type != "table").all():
if slc.datasource_type == "query":
@ -97,6 +96,10 @@ def upgrade():
slc.datasource_type,
)
# need commit the updated values for Slice.datasource_type before creating constraint
session.commit()
with op.batch_alter_table("slices") as batch_op:
batch_op.create_check_constraint(
"ck_chart_datasource", "datasource_type in ('table')"
)