Changeset 778d53d in OpenWorkouts-current for ow/models/user.py

Timestamp:

Mar 5, 2019, 11:45:32 PM (5 years ago)

Author:

Borja Lopez <borja@…>

Branches:

current

Children:

aa6dcaf

Parents:

35953eb

Message:

(#7) Show per-sport stats in the profile page:

• Show a dropdown list of sports for which the user has activities. By default we choose the sport with most activities.

• Show a dropdown list of years for which the user has activities. By default we show stats for the current year. If the user picks up a different year, we show the totals (distance, time, elevation, number of workouts) for that year.

• Show the totals of all time for the chosen sport

File:

• ow/models/user.py (modified) (2 diffs)

ow/models/user.py

@@ -113 +113 @@
     def num_workouts(self):
         return len(self.workout_ids())
+
+    @property
+    def favorite_sport(self):
+        """
+        Return which sport is the one with most workouts for this user.
+        In case of more than one sport with the maximum number of workouts,
+        return the first in reversed alphabetical ordering
+        """
+        sports = {}
+        for w in self.workouts():
+            if w.sport not in sports.keys():
+                sports[w.sport] = 0
+            sports[w.sport] += 1
+        _sports = sorted(sports.items(), reverse=True,
+                         key=lambda x: (x[1], x[0]))
+        if _sports:
+            return _sports[0][0]
+        return None
+
+    @property
+    def activity_sports(self):
+        return sorted(list(set(w.sport for w in self.workouts())))
 
     @property
@@ -368 +390 @@
 
         return stats
+
+    def sport_totals(self, sport=None, year=None):
+        """
+        Return totals for this user, filtered by sport.
+
+        If no sport is passed, the favorite sport is picked up
+
+        If the additional parameter year is passed, show stats only
+        for that year
+        """
+        totals = {
+            'workouts': 0,
+            'time': timedelta(0),
+            'distance': Decimal(0),
+            'elevation': Decimal(0),
+        }
+        if self.activity_sports:
+            sport = sport or self.favorite_sport
+            for workout in self.workouts():
+                if workout.sport == sport:
+                    if year is None or workout.start.year == year:
+                        totals['workouts'] += 1
+                        totals['time'] += workout.duration or timedelta(0)
+                        totals['distance'] += workout.distance or Decimal(0)
+                        totals['elevation'] += workout.uphill or Decimal(0)
+        return totals