From 4b6f4481707663eb9272b0316fd62b2e8f742c54 Mon Sep 17 00:00:00 2001 From: Taha Yassine Kraiem Date: Mon, 8 Jul 2024 16:16:50 +0200 Subject: [PATCH] fix(chalice): fixed get heatmap with incomplete payload --- api/chalicelib/core/custom_metrics.py | 7 +++++-- ee/api/chalicelib/core/custom_metrics.py | 7 +++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/api/chalicelib/core/custom_metrics.py b/api/chalicelib/core/custom_metrics.py index ba8ca8ff10..d681eb48f7 100644 --- a/api/chalicelib/core/custom_metrics.py +++ b/api/chalicelib/core/custom_metrics.py @@ -692,9 +692,12 @@ def make_chart_from_card(project_id, user_id, metric_id, data: schemas.CardSessi data=data.model_dump()) elif metric.metric_type == schemas.MetricType.heat_map: if raw_metric["data"] and raw_metric["data"].get("sessionId"): - return heatmaps.get_selected_session(project_id=project_id, session_id=raw_metric["data"]["sessionId"]) + return heatmaps.get_selected_session(project_id=project_id, + session_id=raw_metric["data"]["sessionId"]) else: - return heatmaps.search_short_session(project_id=project_id, data=metric, user_id=user_id) + return heatmaps.search_short_session(project_id=project_id, + data=schemas.HeatMapSessionsSearch(**metric.model_dump()), + user_id=user_id) return get_chart(project_id=project_id, data=metric, user_id=user_id) diff --git a/ee/api/chalicelib/core/custom_metrics.py b/ee/api/chalicelib/core/custom_metrics.py index 10544382be..df7a8ffca2 100644 --- a/ee/api/chalicelib/core/custom_metrics.py +++ b/ee/api/chalicelib/core/custom_metrics.py @@ -739,9 +739,12 @@ def make_chart_from_card(project_id, user_id, metric_id, data: schemas.CardSessi data=data.model_dump()) elif metric.metric_type == schemas.MetricType.heat_map: if raw_metric["data"] and raw_metric["data"].get("sessionId"): - return heatmaps.get_selected_session(project_id=project_id, session_id=raw_metric["data"]["sessionId"]) + return heatmaps.get_selected_session(project_id=project_id, + session_id=raw_metric["data"]["sessionId"]) else: - return heatmaps.search_short_session(project_id=project_id, data=metric, user_id=user_id) + return heatmaps.search_short_session(project_id=project_id, + data=schemas.HeatMapSessionsSearch(**metric.model_dump()), + user_id=user_id) return get_chart(project_id=project_id, data=metric, user_id=user_id)