Modified userstat where it passes in the trip_key as an input instead…

… of hardcoding analysis/confirmed_trip
e-mission · Jan 8, 2025 · ed6e18d · ed6e18d
1 parent 1df808d
commit ed6e18d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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': {}}}]
         )