From fdef9cbc96c3aeb1445e55632a6e6baccade16b4 Mon Sep 17 00:00:00 2001 From: "Michael S. Molina" <70410625+michael-s-molina@users.noreply.github.com> Date: Tue, 20 Jun 2023 13:53:13 -0300 Subject: [PATCH] fix: Viz migration adjustments - 2 (#24429) --- superset/migrations/shared/migrate_viz/processors.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/superset/migrations/shared/migrate_viz/processors.py b/superset/migrations/shared/migrate_viz/processors.py index 4044d9f4bcdff..fa7043fc38dd1 100644 --- a/superset/migrations/shared/migrate_viz/processors.py +++ b/superset/migrations/shared/migrate_viz/processors.py @@ -14,6 +14,8 @@ # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. +from typing import Any + from .base import MigrateViz @@ -21,6 +23,7 @@ class MigrateTreeMap(MigrateViz): source_viz_type = "treemap" target_viz_type = "treemap_v2" remove_keys = {"metrics"} + rename_keys = {"order_desc": "sort_by_metric"} def _pre_action(self) -> None: if ( @@ -105,3 +108,7 @@ def _pre_action(self) -> None: self.data["truncateYAxis"] = True self.data["metrics"] = [self.data.get("metric")] self.data["metrics_b"] = [self.data.get("metric_2")] + + def _migrate_temporal_filter(self, rv_data: dict[str, Any]) -> None: + super()._migrate_temporal_filter(rv_data) + rv_data["adhoc_filters_b"] = rv_data.get("adhoc_filters") or []