From e3b7c15f2a7b060895369e72149f0391f273ed91 Mon Sep 17 00:00:00 2001 From: april Date: Thu, 11 Jan 2024 17:06:21 -0600 Subject: [PATCH] Reimplement human-readable class/category names in totals --- api/database/flights.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/api/database/flights.py b/api/database/flights.py index 4c9a3e2..d1a348e 100644 --- a/api/database/flights.py +++ b/api/database/flights.py @@ -123,10 +123,10 @@ async def retrieve_totals(user: str, start_date: datetime = None, end_date: date result = dict(result_list[0]) - # for entry in result["by_class"]: - # entry["aircraft_category"] = aircraft_category_dict[entry["aircraft_category"]] - # for cls in entry["classes"]: - # cls["aircraft_class"] = aircraft_class_dict[cls["aircraft_class"]] + for entry in result["by_class"]: + entry["aircraft_category"] = aircraft_category_dict[entry["aircraft_category"]] + for cls in entry["classes"]: + cls["aircraft_class"] = aircraft_class_dict[cls["aircraft_class"]] print(result)