From 1e14a994197bbc9e3eebeb78b2059b07e0a5f66a Mon Sep 17 00:00:00 2001
From: Edward Betts <edward@4angle.com>
Date: Sun, 16 Jun 2024 11:31:23 +0100
Subject: [PATCH] Travel stats: distance and price by year and travel type

Closes: #155
---
 templates/trip/list.html  |  3 ++-
 templates/trip/stats.html | 34 ++++++++++++++++++++++++++++++++
 web_view.py               | 41 +++++++++++++++++++++++++++++++++++++++
 3 files changed, 77 insertions(+), 1 deletion(-)
 create mode 100644 templates/trip/stats.html

diff --git a/templates/trip/list.html b/templates/trip/list.html
index f7a0580..547d1c0 100644
--- a/templates/trip/list.html
+++ b/templates/trip/list.html
@@ -48,11 +48,12 @@
 {% endblock %}
 
 {% macro section(heading, item_list) %}
-  {% if item_list %}
     {% set items = item_list | list %}
     <div class="heading"><h2>{{ heading }}</h2></div>
+    <p><a href="{{ url_for("trip_stats") }}">Trip statistics</a></p>
     <p>{{ items | count }} trips</p>
 
+  {% if item_list %}
     <div>Total distance: {{ format_distance(total_distance) }}</div>
 
     {% for transport_type, distance in distances_by_transport_type %}
diff --git a/templates/trip/stats.html b/templates/trip/stats.html
new file mode 100644
index 0000000..4b535d5
--- /dev/null
+++ b/templates/trip/stats.html
@@ -0,0 +1,34 @@
+{% extends "base.html" %}
+
+{% from "macros.html" import format_distance with context %}
+
+{% set heading = "Trip statistics" %}
+
+{% block title %}{{ heading }} - Edward Betts{% endblock %}
+
+{% block content %}
+<div class="container">
+  <h1>Trip statistics</h1>
+  <div>Trips: {{ count }}</div>
+  <div>Total distance: {{ format_distance(total_distance) }}</div>
+
+  {% for transport_type, distance in distances_by_transport_type %}
+    <div>
+      {{ transport_type | title }}
+      distance: {{format_distance(distance) }}
+    </div>
+  {% endfor %}
+
+  {% for year, year_stats in yearly_stats | dictsort %}
+    <h4>{{ year }}</h4>
+    <div>Trips in {{ year }}: {{ year_stats.count }}</div>
+    <div>Total distance in {{ year}}: {{ format_distance(year_stats.total_distance) }}</div>
+    {% for transport_type, distance in year_stats.distances_by_transport_type.items() %}
+      <div>
+        {{ transport_type | title }}
+        distance: {{format_distance(distance) }}
+      </div>
+    {% endfor %}
+  {% endfor %}
+</div>
+{% endblock %}
diff --git a/web_view.py b/web_view.py
index 7bfde9f..5c39a50 100755
--- a/web_view.py
+++ b/web_view.py
@@ -526,6 +526,47 @@ def birthday_list() -> str:
     return flask.render_template("birthday_list.html", items=items, today=today)
 
 
+def calculate_yearly_stats(trips: list[Trip]) -> dict[int, StrDict]:
+    """Calculate total distance and distance by transport type grouped by year."""
+    yearly_stats: defaultdict[int, StrDict] = defaultdict(dict)
+    for trip in trips:
+        year = trip.start.year
+        dist = trip.total_distance()
+        yearly_stats[year].setdefault("count", 0)
+        yearly_stats[year]["count"] += 1
+        if dist:
+            yearly_stats[year]["total_distance"] = (
+                yearly_stats[year].get("total_distance", 0) + trip.total_distance()
+            )
+        for transport_type, distance in trip.distances_by_transport_type():
+            yearly_stats[year].setdefault("distances_by_transport_type", {})
+            yearly_stats[year]["distances_by_transport_type"][transport_type] = (
+                yearly_stats[year]["distances_by_transport_type"].get(transport_type, 0)
+                + distance
+            )
+    return dict(yearly_stats)
+
+
+@app.route("/trip/stats")
+def trip_stats() -> str:
+    """Travel stats: distance and price by year and travel type."""
+    route_distances = agenda.travel.load_route_distances(app.config["DATA_DIR"])
+    trip_list = get_trip_list(route_distances)
+    today = date.today()
+
+    past = [item for item in trip_list if (item.end or item.start) < today]
+
+    yearly_stats = calculate_yearly_stats(past)
+
+    return flask.render_template(
+        "trip/stats.html",
+        count=len(past),
+        total_distance=calc_total_distance(past),
+        distances_by_transport_type=sum_distances_by_transport_type(past),
+        yearly_stats=yearly_stats,
+    )
+
+
 @app.route("/callback")
 def auth_callback() -> tuple[str, int] | werkzeug.Response:
     """Process the authentication callback."""