sweatstack 0.17.0__py3-none-any.whl → 0.19.0__py3-none-any.whl

sweatstack/client.py

@@ -15,10 +15,14 @@ from typing import Any, Generator, get_type_hints, List, Literal
 from urllib.parse import parse_qs, urlparse
 
 import httpx
+import ipywidgets as widgets
 import pandas as pd
+from IPython.display import display
 
 from .constants import DEFAULT_URL
-from .schemas import ActivityDetails, ActivitySummary, Sport, TraceDetails
+from .schemas import (
+    ActivityDetails, ActivitySummary, Sport, TraceDetails, UserSummary
+)
 from .utils import decode_jwt_body, make_dataframe_streamlit_compatible
 
 
@@ -62,6 +66,7 @@ class OAuth2Mixin:
             "redirect_uri": redirect_uri,
             "code_challenge": code_challenge,
             "scope": "data:read",
+            "prompt": "none",
         }
         base_url = self.url
         path = "/oauth/authorize"
@@ -83,7 +88,7 @@ class OAuth2Mixin:
                 "grant_type": "authorization_code",
                 "client_id": OAUTH2_CLIENT_ID,
                 "code": server.code,
-                "code_verifier": code_verifier
+                "code_verifier": code_verifier,
             }
             response = httpx.post(
                 f"{self.url}/api/v1/oauth/token",
@@ -103,7 +108,55 @@ class OAuth2Mixin:
             raise Exception("SweatStack Python login failed. Please try again.")
 
 
-class Client(OAuth2Mixin):
+class DelegationMixin:
+    def _get_delegated_token(self, user_id: str):
+        with self._http_client() as client:
+            response = client.post(
+                "/api/v1/oauth/delegated-token",
+                json={"sub": user_id},
+            )
+            response.raise_for_status()
+
+        return response.json()
+
+    def switch_user(self, user_id: str):
+        token_response = self._get_delegated_token(user_id)
+        self.api_key = token_response["access_token"]
+        self.refresh_token = token_response["refresh_token"]
+
+    def _get_principal_token(self):
+        with self._http_client() as client:
+            response = client.get(
+                "/api/v1/oauth/principal-token",
+            )
+            response.raise_for_status()
+        return response.json()
+
+    def switch_back(self):
+        token_response = self._get_principal_token()
+        self.api_key = token_response["access_token"]
+        self.refresh_token = token_response["refresh_token"]
+
+    def delegated_client(self, user_id: str):
+        token_response = self._get_delegated_token(user_id)
+        return self.__class__(
+            api_key=token_response["access_token"],
+            refresh_token=token_response["refresh_token"],
+            url=self.url,
+            streamlit_compatible=self.streamlit_compatible,
+        )
+
+    def principal_client(self):
+        token_response = self._get_principal_token()
+        return self.__class__(
+            api_key=token_response["access_token"],
+            refresh_token=token_response["refresh_token"],
+            url=self.url,
+            streamlit_compatible=self.streamlit_compatible,
+        )
+
+
+class Client(OAuth2Mixin, DelegationMixin, JupyterInteractivityMixin):
     def __init__(
         self,
         api_key: str | None = None,
@@ -553,6 +606,13 @@ class Client(OAuth2Mixin):
             response.raise_for_status()
             return response.json()
 
+    def list_users(self) -> list[UserSummary]:
+        with self._http_client() as client:
+            response = client.get(
+                url="/api/v1/users/",
+            )
+            response.raise_for_status()
+            return [UserSummary.model_validate(user) for user in response.json()]
 
 _default_client = Client()
 
@@ -593,6 +653,8 @@ _generate_singleton_methods(
     [
         "login",
 
+        "list_users",
+
         "get_activities",
 
         "get_activity",
@@ -611,5 +673,10 @@ _generate_singleton_methods(
 
         "get_sports",
         "get_tags",
+
+        "switch_user",
+        "switch_back",
+        "delegated_client",
+        "principal_client",
     ]
 )

sweatstack/openapi_schemas.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2025-02-13T12:03:28+00:00
+#   timestamp: 2025-03-03T13:21:38+00:00
 
 from __future__ import annotations
 
@@ -59,16 +59,6 @@ class BodySaveOrUpdateIntegrationProviderTenantsTenantIdIntegrationProvidersInte
     redirect_url: str = Field(..., title='Redirect Url')
 
 
-class BodyTokenOauthTokenPost(BaseModel):
-    grant_type: Literal['authorization_code'] = Field(..., title='Grant Type')
-    code: str = Field(..., title='Code')
-    client_id: str = Field(..., title='Client Id')
-    client_secret: Optional[str] = Field(None, title='Client Secret')
-    code_verifier: Optional[str] = Field(None, title='Code Verifier')
-    refresh_token: Optional[str] = Field(None, title='Refresh Token')
-    tz_offset: Optional[int] = Field(0, title='Tz Offset')
-
-
 class BodyUpdateApplicationApplicationsApplicationIdPut(BaseModel):
     name: str = Field(..., title='Name')
     description: str = Field(..., title='Description')
@@ -122,7 +112,6 @@ class GarminFileTypes(Enum):
 class GrantType(Enum):
     authorization_code = 'authorization_code'
     refresh_token = 'refresh_token'
-    implicit = 'implicit'
 
 
 class HeartRateSummary(BaseModel):
@@ -166,6 +155,8 @@ class Scope(Enum):
     data_read = 'data:read'
     data_write = 'data:write'
     admin = 'admin'
+    openid = 'openid'
+    profile = 'profile'
 
 
 class Smo2Summary(BaseModel):
@@ -233,13 +224,12 @@ class TemperatureSummary(BaseModel):
 
 class TokenRequest(BaseModel):
     grant_type: GrantType
+    client_id: Optional[str] = Field(None, title='Client Id')
     code: Optional[str] = Field(None, title='Code')
+    client_secret: Optional[str] = Field(None, title='Client Secret')
+    code_verifier: Optional[str] = Field(None, title='Code Verifier')
     refresh_token: Optional[str] = Field(None, title='Refresh Token')
-    client_id: Optional[str] = Field(None, title='Client Id')
-    client_secret: Optional[SecretStr] = Field(None, title='Client Secret')
-    tz_offset: Optional[int] = Field(
-        3600, description='Timezone DST offset in seconds', title='Tz Offset'
-    )
+    tz_offset: Optional[int] = Field(0, title='Tz Offset')
 
 
 class TokenResponse(BaseModel):
@@ -248,6 +238,7 @@ class TokenResponse(BaseModel):
     expires_in: int = Field(..., title='Expires In')
     refresh_token: str = Field(..., title='Refresh Token')
     scope: Optional[str] = Field(None, title='Scope')
+    id_token: Optional[str] = Field(None, title='Id Token')
 
 
 class TraceCreateOrUpdate(BaseModel):
@@ -259,6 +250,22 @@ class TraceCreateOrUpdate(BaseModel):
     speed: Optional[confloat(ge=0.0)] = Field(None, title='Speed')
     heart_rate: Optional[conint(ge=0)] = Field(None, title='Heart Rate')
     tags: Optional[List[str]] = Field(None, title='Tags')
+    sport: Optional[Sport] = None
+
+
+class UserInfoResponse(BaseModel):
+    sub: str = Field(..., title='Sub')
+    given_name: Optional[str] = Field(None, title='Given Name')
+    family_name: Optional[str] = Field(None, title='Family Name')
+    name: str = Field(..., title='Name')
+
+
+class UserSummary(BaseModel):
+    id: str = Field(..., title='Id')
+    first_name: Optional[str] = Field(..., title='First Name')
+    last_name: Optional[str] = Field(..., title='Last Name')
+    scopes: List[Scope] = Field(..., title='Scopes')
+    display_name: str = Field(..., title='Display Name')
 
 
 class ValidationError(BaseModel):
@@ -283,10 +290,16 @@ class BodyAuthorizeOauthAuthorizePost(BaseModel):
     redirect_uri: Optional[str] = Field(None, title='Redirect Uri')
     scopes: List[Scope] = Field(..., min_length=1, title='Scopes')
     state: Optional[str] = Field(None, title='State')
+    nonce: Optional[str] = Field(None, title='Nonce')
     code_challenge: Optional[str] = Field(None, title='Code Challenge')
     code_challenge_method: Optional[str] = Field(None, title='Code Challenge Method')
 
 
+class DelegatedTokenRequest(BaseModel):
+    sub: str = Field(..., title='Sub')
+    scopes: List[Scope] = Field(..., title='Scopes')
+
+
 class GarminActivityFileData(BaseModel):
     userId: str = Field(..., title='Userid')
     userAccessToken: str = Field(..., title='Useraccesstoken')
@@ -362,6 +375,7 @@ class TraceDetails(BaseModel):
     heart_rate: Optional[conint(ge=0)] = Field(None, title='Heart Rate')
     lap: Optional[Lap] = None
     activity: Optional[ActivitySummary] = None
+    sport: Optional[Sport] = None
 
 
 ActivityDetails.model_rebuild()

sweatstack/schemas.py

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

{sweatstack-0.17.0.dist-info → sweatstack-0.19.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.17.0
+Version: 0.19.0
 Summary: The official Python client for SweatStack
 Author-email: Aart Goossens <aart@gssns.io>
 Requires-Python: >=3.12
@@ -8,10 +8,10 @@ Requires-Dist: httpx>=0.28.1
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: pyarrow>=19.0.0
 Requires-Dist: pydantic>=2.10.5
-Provides-Extra: jupyterlab
-Requires-Dist: ipython>=8.31.0; extra == 'jupyterlab'
-Requires-Dist: jupyterlab>=4.3.4; extra == 'jupyterlab'
-Requires-Dist: matplotlib; extra == 'jupyterlab'
+Provides-Extra: jupyter
+Requires-Dist: ipython>=8.31.0; extra == 'jupyter'
+Requires-Dist: jupyterlab>=4.3.4; extra == 'jupyter'
+Requires-Dist: matplotlib>=3.10.0; extra == 'jupyter'
 Provides-Extra: streamlit
 Requires-Dist: streamlit>=1.42.0; extra == 'streamlit'
 Description-Content-Type: text/markdown

{sweatstack-0.17.0.dist-info → sweatstack-0.19.0.dist-info}/RECORD

@@ -1,16 +1,17 @@
 sweatstack/__init__.py,sha256=tiVfgKlswRPaDMEy0gA7u8rveqEYZTA_kyB9lJ3J6Sc,21
 sweatstack/cli.py,sha256=N1NWOgEZR2yaJvIXxo9qvp_jFlypZYb0nujpbVNYQ6A,720
-sweatstack/client.py,sha256=A6PGQFd2hCupJ1UMjjvRyUx18mRQe-ZaHgE8snGKY_Q,20008
+sweatstack/client.py,sha256=KmKy84XULwTXpAr3IpNcke8tSbjEySUtQR3Qlio2r0Y,22256
 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=vtYHa6A0kADFbd_jK1O7Kbn8YkbPfzaIwsRDDYjYeSA,13038
+sweatstack/openapi_schemas.py,sha256=XlgiL7qkfcfoDHcQrIm9e5hvhY98onC0QmZWG69bl-s,13441
 sweatstack/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sweatstack/schemas.py,sha256=CdkeV6IRmIuvxae7C5dz-hVlb6hkzEYfqKHHgVJprmY,90
+sweatstack/schemas.py,sha256=CArM9jgrJDBUT-ZIIiGN_insqaEGV43MX134Ezf4jyA,103
 sweatstack/streamlit.py,sha256=F5oQdWkcJ76Um5fRqFT5QrjpEz8v3OaiH9kMQOYktgo,4466
 sweatstack/sweatshell.py,sha256=MYLNcWbOdceqKJ3S0Pe8dwHXEeYsGJNjQoYUXpMTftA,333
 sweatstack/utils.py,sha256=HtR1NNCKus59vfgfaCSFS-tHA11mtdcuUx5lS6ZX58g,1773
-sweatstack-0.17.0.dist-info/METADATA,sha256=fCrgu9umxH57CPonX4eKReH9OXaMpNbJ3pz2-3dy8oM,2966
-sweatstack-0.17.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.17.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.17.0.dist-info/RECORD,,
+sweatstack/Sweat Stack examples/Getting started.ipynb,sha256=k2hiSffWecoQ0VxjdpDcgFzBXDQiYEebhnAYlu8cgX8,6335204
+sweatstack-0.19.0.dist-info/METADATA,sha256=vaOKnv63psIaEa6E-ZygllTXhqfILyIhw2L4w4BqhnE,2962
+sweatstack-0.19.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sweatstack-0.19.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
+sweatstack-0.19.0.dist-info/RECORD,,