sweatstack 0.27.0__tar.gz → 0.30.0__tar.gz

{sweatstack-0.27.0 → sweatstack-0.30.0}/PKG-INFO

@@ -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 → sweatstack-0.30.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sweatstack"
-version = "0.27.0"
+version = "0.30.0"
 description = "The official Python client for SweatStack"
 readme = "README.md"
 authors = [

{sweatstack-0.27.0 → sweatstack-0.30.0}/src/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-0.30.0/src/sweatstack/schemas.py

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

{sweatstack-0.27.0 → sweatstack-0.30.0}/src/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-0.27.0 → sweatstack-0.30.0}/src/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/src/sweatstack/schemas.py

@@ -1,3 +0,0 @@
-from .openapi_schemas import (
-    ActivityDetails, ActivitySummary, Sport, TraceDetails, UserSummary
-)