fix: Make CSV output user friendly

Drop unnecessary column: id rename column totalcontributionTime -> totalSwipeTimeInSeconds rename column totalcontributions -> totalSwipes
hotosm · Sep 24, 2024 · 302d623 · 302d623
1 parent 475b66a
commit 302d623
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/backend/models/dtos/partner_stats_dto.py b/backend/models/dtos/partner_stats_dto.py
@@ -93,6 +93,20 @@ class GroupedPartnerStatsDTO(Model):
 
     def to_csv(self):
         df = pd.json_normalize(self.to_primitive()["members"])
+
+        df.drop(
+            columns=["id"],
+            inplace=True,
+            axis=1,
+        )
+        df.rename(
+            columns={
+                "totalcontributionTime": "totalSwipeTimeInSeconds",
+                "totalcontributions": "totalSwipes",
+            },
+            inplace=True,
+        )
+
         return df.to_csv(index=False)