feat: fix "All User Stats" download link

This commit brings in a new route, /data/all_user_stats.csv.

TODO: Simplify other CSV views.

Signed-off-by: Kevin Morris <kevr@0cost.org>

Author: kevr

home/templates/home/base.html

@@ -62,7 +62,7 @@
             Download
           </a>
           <div class="dropdown-menu" aria-labelledby="navbarDropdown">
-            <a class="dropdown-item" href="/data/users_agg.csv">All User Stats</a>
+            <a class="dropdown-item" href="/data/all_user_stats.csv">All User Stats</a>
             <a class="dropdown-item" href="/data/users.csv">All User Account Info</a>
             <a class="dropdown-item" href="/data/daily_walks.csv">All Walks</a>
             <a class="dropdown-item" href="/data/intentional_walks.csv">All Recorded Walks</a>

home/urls.py

@@ -16,6 +16,11 @@
         views.user_agg_csv_view,
         name="users_agg_csv_view",
     ),
+    path(
+        "data/all_user_stats.csv",
+        views.all_user_stats_csv_view,
+        name="all_user_stats_csv_view",
+    ),
     path("data/users.csv", views.users_csv_view, name="users_csv_view"),
     path(
         "data/daily_walks.csv",

home/views/__init__.py

@@ -8,6 +8,7 @@
 from .web.home import HomeView
 from .web.intentionalwalk import IntentionalWalkWebView
 from .web.user import UserListView
+from .web.data import all_user_stats_csv_view
 from .web.data import user_agg_csv_view
 from .web.data import users_csv_view
 from .web.data import daily_walks_csv_view

home/views/web/data.py

@@ -184,6 +184,70 @@ def _get_user_daily_step_counts(
     return daily_step_counts_by_user
 
 
+def all_user_stats_csv_view(request) -> HttpResponse:
+    columns = [
+        "Participant Name",
+        "Date Enrolled",
+        "Email",
+        "Zip Code",
+        "Sexual Orientation",
+        "Sexual Orientation Other",
+        "Gender Identity",
+        "Gender Identity Other",
+        "Race",
+        "Race Other",
+        "Is Latino",
+        "Age",
+        "Total Daily Walks",
+        "Total Daily Steps",
+        "Total Recorded Walks",
+        "Total Recorded Steps",
+    ]
+
+    users = Account.objects.all()
+    output = []
+    for user in users:
+        daily_walks = user.dailywalk_set.values("steps")
+        cnt_daily = len(daily_walks)
+        daily_steps = sum([w.get("steps") for w in daily_walks])
+
+        rec_walks = user.intentionalwalk_set.values("steps")
+        cnt_recorded = len(rec_walks)
+        rec_steps = sum([w.get("steps") for w in rec_walks])
+
+        output.append(
+            {
+                "Participant Name": user.name,
+                "Date Enrolled": user.created,
+                "Email": user.email,
+                "Zip Code": user.zip,
+                "Sexual Orientation": user.sexual_orien,
+                "Sexual Orientation Other": user.sexual_orien_other,
+                "Gender Identity": user.gender,
+                "Gender Identity Other": user.gender_other,
+                "Race": user.race,
+                "Race Other": user.race_other,
+                "Is Latino": user.is_latino,
+                "Age": user.age,
+                "Total Daily Walks": cnt_daily,
+                "Total Daily Steps": daily_steps,
+                "Total Recorded Walks": cnt_recorded,
+                "Total Recorded Steps": rec_steps,
+            }
+        )
+
+    response = HttpResponse(content_type="text/csv")
+    response[
+        "Content-Disposition"
+    ] = 'attachment; filename="all_users_stats.csv"'
+
+    writer = csv.DictWriter(response, fieldnames=columns)
+    writer.writeheader()
+    writer.writerows(output)
+
+    return response
+
+
 def user_agg_csv_view(request) -> HttpResponse:
     if not request.user.is_authenticated:
         return HttpResponse("You are not authorized to view this!")