From c3955b2472e599d686b51ec314c6a3fd43b85c63 Mon Sep 17 00:00:00 2001 From: "Hugh A. Miles II" Date: Tue, 29 Mar 2022 22:24:46 -0700 Subject: [PATCH] Update superset/migrations/versions/2ed890b36b94_rm_time_range_endpoints_from_qc.py Co-authored-by: Yongjie Zhao --- .../2ed890b36b94_rm_time_range_endpoints_from_qc.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/superset/migrations/versions/2ed890b36b94_rm_time_range_endpoints_from_qc.py b/superset/migrations/versions/2ed890b36b94_rm_time_range_endpoints_from_qc.py index 67d74e93d00ec..bc89e62b98e28 100644 --- a/superset/migrations/versions/2ed890b36b94_rm_time_range_endpoints_from_qc.py +++ b/superset/migrations/versions/2ed890b36b94_rm_time_range_endpoints_from_qc.py @@ -47,14 +47,11 @@ class Slice(Base): def upgrade(): bind = op.get_bind() session = db.Session(bind=bind) - - for m in session.query(MyModel): - queries = json.loads(m.queries) - for q in queries: - extras = q.get("extras") - if extras: - extras.pop("time_range_endpoints", None) - m.queries = json.dumps(queries) + for slc in session.query(Slice): + queries = json.loads(slc.queries) + for query in queries: + query.get("extras", {}).pop("time_range_endpoints", None) + slc.queries = json.dumps(queries) session.commit() session.close()