in superset/migrations/versions/2018-11-12_13-31_4ce8df208545_migrate_time_range_for_default_filters.py [0:0]
def upgrade(): # noqa: C901
bind = op.get_bind()
session = db.Session(bind=bind)
dashboards = session.query(Dashboard).all()
for i, dashboard in enumerate(dashboards):
print(f"scanning dashboard ({i + 1}/{len(dashboards)}) >>>>")
if dashboard.json_metadata:
json_metadata = json.loads(dashboard.json_metadata)
has_update = False
# default_filters: it may define time_range filter
default_filters = json_metadata.get("default_filters")
if default_filters and default_filters != "{}":
try:
filters = json.loads(default_filters)
keys = [
key
for key, val in filters.items()
if val.get("__from") or val.get("__to")
]
if len(keys):
for key in keys:
val = filters[key]
__from = val.pop("__from", "")
__to = val.pop("__to", "")
# if user already defined __time_range,
# just abandon __from and __to
if "__time_range" not in val:
val["__time_range"] = f"{__from} : {__to}"
json_metadata["default_filters"] = json.dumps(filters)
has_update = True
except Exception: # noqa: S110
pass
# filter_immune_slice_fields:
# key: chart id, value: field names that escape from filters
filter_immune_slice_fields = json_metadata.get("filter_immune_slice_fields")
if filter_immune_slice_fields:
keys = [
key
for key, val in filter_immune_slice_fields.items()
if "__from" in val or "__to" in val
]
if len(keys):
for key in keys:
val = filter_immune_slice_fields[key]
if "__from" in val:
val.remove("__from")
if "__to" in val:
val.remove("__to")
# if user already defined __time_range,
# just abandon __from and __to
if "__time_range" not in val:
val.append("__time_range")
json_metadata["filter_immune_slice_fields"] = (
filter_immune_slice_fields
)
has_update = True
if has_update:
dashboard.json_metadata = json.dumps(json_metadata)
session.commit()
session.close()