From 82e9170900fa093158c451207a4cfbde3f66de3b Mon Sep 17 00:00:00 2001 From: Martin Ueding Date: Sat, 3 Feb 2024 17:55:29 +0100 Subject: [PATCH] GH-17: Add total for progress bar --- geo_activity_playground/core/activities.py | 3 +++ geo_activity_playground/importers/strava_api.py | 1 + 2 files changed, 4 insertions(+) diff --git a/geo_activity_playground/core/activities.py b/geo_activity_playground/core/activities.py index 570d2fd..cedd055 100644 --- a/geo_activity_playground/core/activities.py +++ b/geo_activity_playground/core/activities.py @@ -44,6 +44,9 @@ def __init__(self) -> None: self._loose_activities: list[ActivityMeta] = [] + def __len__(self) -> int: + return len(self.meta) + def add_activity(self, activity_meta: ActivityMeta) -> None: self._loose_activities.append(activity_meta) diff --git a/geo_activity_playground/importers/strava_api.py b/geo_activity_playground/importers/strava_api.py index 7ec01b6..45911bb 100644 --- a/geo_activity_playground/importers/strava_api.py +++ b/geo_activity_playground/importers/strava_api.py @@ -217,6 +217,7 @@ def download_missing_calories(repository: ActivityRepository) -> None: for activity in tqdm( client.get_activities(after="2000-01-01T00:00:00Z"), desc="Downloading calories from Strava", + total=len(repository), ): if repository.has_activity(activity.id): calories = get_detailed_activity(activity.id, client).calories