Fix compatibility with Python < 3.11

liskin · Nov 21, 2023 · 75bb15c · 75bb15c
1 parent fc36381
commit 75bb15c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/strava_gear/cli.py b/src/strava_gear/cli.py
@@ -55,7 +55,8 @@
     '--show-vert/--hide-vert', default=False, show_default=True,
     help="Show vertical (elevation gain)")
 @click.option(
-    '--units', type=click.Choice(list(Units)), default=Units.METRIC, show_default=True,
+    '--units', type=click.Choice([u.name.lower() for u in Units]), default=Units.METRIC.name.lower(), show_default=True,
+    callback=lambda _ctx, _param, v: Units[v.upper()],  # TODO: drop when Python 3.11 is the oldest supported
     help="Show data in metric or imperial")
 def main(
     rules_input: TextIO,

diff --git a/src/strava_gear/report.py b/src/strava_gear/report.py
@@ -1,6 +1,6 @@
 from collections import defaultdict
 import csv
-from enum import StrEnum
+from enum import Enum
 from enum import auto
 from functools import partial
 from typing import Dict
@@ -18,7 +18,7 @@
 FOOT_IN_METERS: Final[float] = 0.3048
 
 
-class Units(StrEnum):
+class Units(Enum):
     METRIC = auto()
     IMPERIAL = auto()