add code change

apache · Apr 19, 2022 · 5f43c5b · 5f43c5b
1 parent ce63dfe
commit 5f43c5b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/superset/migrations/versions/ad07e4fdbaba_rm_time_range_endpoints_from_qc_3.py b/superset/migrations/versions/ad07e4fdbaba_rm_time_range_endpoints_from_qc_3.py
@@ -50,10 +50,13 @@ def upgrade_slice(slc: Slice):
     except json.decoder.JSONDecodeError:
         return
 
-    queries = query_context.get("queries")
     query_context.get("form_data", {}).pop("time_range_endpoints", None)
-    for query in queries:
-        query.get("extras", {}).pop("time_range_endpoints", None)
+
+    if query_context.get("queries"):
+        queries = query_context["queries"]
+        for query in queries:
+            if query.get("extras", None).get("time_range_endpoints"):
+                del query["extras"]["time_range_endpoints"]
 
     slc.query_context = json.dumps(query_context)