sweatstack 0.27.0__py3-none-any.whl → 0.30.0__py3-none-any.whl

sweatstack/client.py

@@ -297,6 +297,7 @@ class Client(OAuth2Mixin, DelegationMixin):
         if end is not None:
             params["end"] = end.isoformat()
         if sports is not None:
+            sports = [sport.value if isinstance(sport, Sport) else sport for sport in sports]
             params["sports"] = sports
         if tags is not None:
             params["tags"] = tags

sweatstack/schemas.py

@@ -1,3 +1,5 @@
+from enum import Enum
+
 from .openapi_schemas import (
-    ActivityDetails, ActivitySummary, Sport, TraceDetails, UserSummary
+    ActivityDetails, ActivitySummary, Metric, Sport, TraceDetails, UserSummary
 )

sweatstack/streamlit.py

@@ -1,6 +1,6 @@
 import os
 import urllib.parse
-
+from datetime import date
 try:
     import streamlit as st
 except ImportError:
@@ -10,10 +10,11 @@ except ImportError:
         "pip install 'sweatstack[streamlit]'\n\n"
     )
 import httpx
-from sweatstack import Client
 
+from .client import Client
 from .constants import DEFAULT_URL
-
+from .schemas import Metric, Sport
+from .utils import format_sport
 
 class StreamlitAuth:
     def __init__(self, client_id=None, client_secret=None, scope=None, redirect_uri=None):
@@ -88,6 +89,11 @@ class StreamlitAuth:
 
         return authorization_url
 
+    def _set_api_key(self, api_key):
+        self.api_key = api_key
+        st.session_state["sweatstack_api_key"] = api_key
+        self.client = Client(self.api_key, streamlit_compatible=True)
+
     def _exchange_token(self, code):
         token_data = {
             "grant_type": "authorization_code",
@@ -107,10 +113,7 @@ class StreamlitAuth:
             raise Exception(f"SweatStack Python login failed. Please try again.") from e
         token_response = response.json()
 
-        self.api_key = token_response.get("access_token")
-        st.session_state["sweatstack_api_key"] = self.api_key
-
-        self.client = Client(self.api_key, streamlit_compatible=True)
+        self._set_api_key(token_response.get("access_token"))
 
         return
 
@@ -128,4 +131,101 @@ class StreamlitAuth:
             st.query_params.clear()
             st.rerun()
         else:
-            self._show_sweatstack_login()
+            self._show_sweatstack_login()
+
+    def select_user(self):
+        self.switch_to_principal_user()
+        other_users = self.client.get_users()
+        selected_user = st.selectbox(
+            "Select a user",
+            other_users,
+            format_func=lambda user: user.display_name,
+        )
+        self.client.switch_user(selected_user)
+        self._set_api_key(self.client.api_key)
+
+        return selected_user
+
+    def switch_to_principal_user(self):
+        self.client.switch_back()
+        self._set_api_key(self.client.api_key)
+
+    def select_activity(
+        self,
+        *,
+        start: date | None = None,
+        end: date | None = None,
+        sports: list[Sport] | None = None,
+        tags: list[str] | None = None,
+        limit: int | None = 100,
+    ):
+        """
+        Select an activity from the user's activities.
+        """
+
+        activities = self.client.get_activities(
+            start=start,
+            end=end,
+            sports=sports,
+            tags=tags,
+            limit=limit,
+        )
+        selected_activity = st.selectbox(
+            "Select an activity",
+            activities,
+            format_func=lambda activity: f"{activity.start.date().isoformat()} {format_sport(activity.sport)}",
+        )
+        return selected_activity
+
+    def select_sport(self, only_root: bool = False, allow_multiple: bool = False, only_available: bool = True):
+        if only_available:
+            sports = self.client.get_sports(only_root)
+        else:
+            if only_root:
+                sports = [sport for sport in Sport if "." not in sport.value]
+            else:
+                sports = Sport
+
+        if allow_multiple:
+            selected_sport = st.multiselect(
+                "Select sports",
+                sports,
+                format_func=format_sport,
+            )
+        else:
+            selected_sport = st.selectbox(
+                "Select a sport",
+                sports,
+                format_func=format_sport,
+            )
+        return selected_sport
+
+    def select_tag(self, allow_multiple: bool = False):
+        tags = self.client.get_tags()
+        if allow_multiple:
+            selected_tag = st.multiselect(
+                "Select tags",
+                tags,
+            )
+        else:
+            selected_tag = st.selectbox(
+                "Select a tag",
+                tags,
+                format_func=lambda tag: tag or "-",
+            )
+        return selected_tag
+
+    def select_metric(self, allow_multiple: bool = False):
+        if allow_multiple:
+            selected_metric = st.multiselect(
+                "Select metrics",
+                Metric,
+                format_func=lambda metric: metric.value,
+            )
+        else:
+            selected_metric = st.selectbox(
+                "Select a metric",
+                Metric,
+                format_func=lambda metric: metric.value,
+            )
+        return selected_metric

sweatstack/utils.py

@@ -4,6 +4,8 @@ from enum import Enum
 
 import pandas as pd
 
+from .schemas import Sport
+
 
 def decode_jwt_body(jwt: str) -> dict:
     payload = jwt.split(".")[1]
@@ -50,4 +52,17 @@ def make_dataframe_streamlit_compatible(df: pd.DataFrame) -> pd.DataFrame:
                         lambda x: x.value if isinstance(x, Enum) else x
                     )
 
-    return df_copy if df_copy is not None else df
+    return df_copy if df_copy is not None else df
+
+
+def format_sport(sport: Sport):
+    parts = sport.value.split(".")
+    base_sport = parts[0]
+    base_sport = base_sport.replace("_", " ")
+
+    if len(parts) == 1:
+        return base_sport
+
+    remainder = " ".join(parts[1:]).replace("_", " ")
+
+    return f"{base_sport} ({remainder})"

{sweatstack-0.27.0.dist-info → sweatstack-0.30.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.27.0
+Version: 0.30.0
 Summary: The official Python client for SweatStack
 Author-email: Aart Goossens <aart@gssns.io>
 Requires-Python: >=3.9

{sweatstack-0.27.0.dist-info → sweatstack-0.30.0.dist-info}/RECORD

@@ -1,17 +1,17 @@
 sweatstack/__init__.py,sha256=tiVfgKlswRPaDMEy0gA7u8rveqEYZTA_kyB9lJ3J6Sc,21
 sweatstack/cli.py,sha256=N1NWOgEZR2yaJvIXxo9qvp_jFlypZYb0nujpbVNYQ6A,720
-sweatstack/client.py,sha256=09c9f-ij0Pe4MIoCQustskz8YGxpzUxpSeVoCI2deLk,23713
+sweatstack/client.py,sha256=s4OwGibC31n-XPW2dc3CpwDor-2vkfq4mSAJE8svatQ,23807
 sweatstack/constants.py,sha256=fGO6ksOv5HeISv9lHRoYm4besW1GTveXS8YD3K0ljg0,41
 sweatstack/ipython_init.py,sha256=zBGUlMFkdpLvsNpOpwrNaKRUpUZhzaICvH8ODJgMPcI,229
 sweatstack/jupyterlab_oauth2_startup.py,sha256=eZ6xi0Sa4hO4vUanimq0SqjduHtiywCURSDNWk_I-7s,1200
 sweatstack/openapi_schemas.py,sha256=XlgiL7qkfcfoDHcQrIm9e5hvhY98onC0QmZWG69bl-s,13441
 sweatstack/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sweatstack/schemas.py,sha256=CArM9jgrJDBUT-ZIIiGN_insqaEGV43MX134Ezf4jyA,103
-sweatstack/streamlit.py,sha256=zBPIC29OvgqvuKNlezgLOTwgvS-Cbg885QFwuW4NMRk,4968
+sweatstack/schemas.py,sha256=Lkx0rH8LNIHfX8obHSa4EdJF5q4DzDursnhofzSTddQ,134
+sweatstack/streamlit.py,sha256=gsgiIDW-INGTvF24ANnX5LJ17ZxnvXx95sjSmtcTlnY,8062
 sweatstack/sweatshell.py,sha256=MYLNcWbOdceqKJ3S0Pe8dwHXEeYsGJNjQoYUXpMTftA,333
-sweatstack/utils.py,sha256=HtR1NNCKus59vfgfaCSFS-tHA11mtdcuUx5lS6ZX58g,1773
+sweatstack/utils.py,sha256=3K97OSWQy5KZ97QiBbW4Kx1wDGxwyA96ZWpwIbkcQZc,2090
 sweatstack/Sweat Stack examples/Getting started.ipynb,sha256=k2hiSffWecoQ0VxjdpDcgFzBXDQiYEebhnAYlu8cgX8,6335204
-sweatstack-0.27.0.dist-info/METADATA,sha256=YKWKjaX0UYaqA3ODVWZKlOqba4z-xKF0PI5TS8z3P7E,775
-sweatstack-0.27.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.27.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.27.0.dist-info/RECORD,,
+sweatstack-0.30.0.dist-info/METADATA,sha256=UcL9W5y7VlvMEUkW4N2OpwwOUPqwKjcV1LPbg2sKMCs,775
+sweatstack-0.30.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sweatstack-0.30.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
+sweatstack-0.30.0.dist-info/RECORD,,