Davide Scaini
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
"""Import metadata from Strava's activities.csv bulk export.
|
|
|
|
Strava export columns we care about:
|
|
Activity ID, Activity Date, Activity Name, Activity Type,
|
|
Activity Description, Filename
|
|
"""
|
|
|
|
import csv
|
|
import re
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
|
|
_STRAVA_DATE_FMTS = (
|
|
"%b %d, %Y, %I:%M:%S %p", # "Jun 1, 2024, 7:30:12 AM"
|
|
"%Y-%m-%d %H:%M:%S",
|
|
)
|
|
|
|
|
|
class StravaMetadata:
|
|
"""Maps original filename → Strava metadata."""
|
|
|
|
def __init__(self, csv_path: Path) -> None:
|
|
self._by_filename: dict[str, dict] = {}
|
|
self._load(csv_path)
|
|
|
|
def _load(self, path: Path) -> None:
|
|
with path.open(newline="", encoding="utf-8-sig") as f:
|
|
reader = csv.DictReader(f)
|
|
for row in reader:
|
|
filename = row.get("Filename", "").strip()
|
|
if not filename:
|
|
continue
|
|
# Strava stores paths like "activities/12345.fit.gz"
|
|
basename = Path(filename).name
|
|
self._by_filename[basename] = row
|
|
|
|
def lookup(self, source_file: str) -> Optional[dict]:
|
|
"""Return the Strava CSV row for a given source filename, or None."""
|
|
return self._by_filename.get(source_file)
|
|
|
|
def enrich(self, source_file: str, activity: object) -> None:
|
|
"""Mutate a ParsedActivity with Strava metadata if found."""
|
|
row = self.lookup(source_file)
|
|
if row is None:
|
|
return
|
|
|
|
if not activity.title and row.get("Activity Name"): # type: ignore[attr-defined]
|
|
activity.title = row["Activity Name"].strip() # type: ignore[attr-defined]
|
|
|
|
if not activity.description and row.get("Activity Description"): # type: ignore[attr-defined]
|
|
activity.description = row["Activity Description"].strip() # type: ignore[attr-defined]
|
|
|
|
if not activity.strava_id and row.get("Activity ID"): # type: ignore[attr-defined]
|
|
activity.strava_id = row["Activity ID"].strip() # type: ignore[attr-defined]
|