From ed6e18d7ff7413f66614ed535e9cb9df336fddf6 Mon Sep 17 00:00:00 2001 From: TeachMeTW Date: Mon, 6 Jan 2025 09:17:41 -0800 Subject: [PATCH] Modified userstat where it passes in the trip_key as an input instead of hardcoding analysis/confirmed_trip --- emission/analysis/result/user_stat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/emission/analysis/result/user_stat.py b/emission/analysis/result/user_stat.py index fa1d7ac95..e671f6847 100644 --- a/emission/analysis/result/user_stat.py +++ b/emission/analysis/result/user_stat.py @@ -61,9 +61,9 @@ def get_and_store_user_stats(user_id: str, trip_key: str) -> None: end_ts_result = ts.get_first_value_for_field(trip_key, "data.end_ts", pymongo.DESCENDING) end_ts = None if end_ts_result == -1 else end_ts_result - total_trips = ts.find_entries_count(key_list=["analysis/confirmed_trip"]) + total_trips = ts.find_entries_count(key_list=[trip_key], extra_query_list=[{'metadata.origin_key': 'analysis/confirmed_trip'}]) labeled_trips = ts.find_entries_count( - key_list=["analysis/confirmed_trip"], + key_list=[trip_key], extra_query_list=[{'data.user_input': {'$ne': {}}}] )