add log for values unseen

apache · May 3, 2023 · 3bbb7c4 · 3bbb7c4
1 parent ad19cd9
commit 3bbb7c4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/superset/migrations/versions/2023-03-27_12-30_7e67aecbf3f1_chart_ds_constraint.py b/superset/migrations/versions/2023-03-27_12-30_7e67aecbf3f1_chart_ds_constraint.py
@@ -63,9 +63,15 @@ def upgrade():
     session = db.Session(bind=bind)
 
     with op.batch_alter_table("slices") as batch_op:
-        for slc in session.query(Slice).filter(Slice.datasource_type == "query").all():
-            upgrade_slc(slc)
-            session.add(slc)
+        for slc in session.query(Slice).filter(Slice.datasource_type != "table").all():
+            if slc.datasource_type == "query":
+                upgrade_slc(slc)
+                session.add(slc)
+
+            else:
+                print(
+                    f"unknown value detected for slc.datasource_type: {slc.datasource_type}"
+                )
 
         batch_op.create_check_constraint(
             "ck_chart_datasource", "datasource_type in ('table')"