sweatstack 0.46.0__py3-none-any.whl → 0.47.0__py3-none-any.whl

sweatstack/client.py

@@ -16,7 +16,7 @@ from http.server import BaseHTTPRequestHandler, HTTPServer
 from importlib.metadata import version
 from io import BytesIO
 from pathlib import Path
-from typing import Any, Generator, get_type_hints, List, Literal
+from typing import Any, Dict, Generator, get_type_hints, List, Literal
 from urllib.parse import parse_qs, urlparse
 
 import httpx
@@ -25,7 +25,8 @@ from platformdirs import user_data_dir
 
 from .constants import DEFAULT_URL
 from .schemas import (
-    ActivityDetails, ActivitySummary, Metric, Sport, TraceDetails, UserInfoResponse, UserSummary
+    ActivityDetails, ActivitySummary, BackfillStatus, Metric, Sport,
+    TraceDetails, UserInfoResponse, UserSummary
 )
 from .utils import decode_jwt_body, make_dataframe_streamlit_compatible
 
@@ -1230,6 +1231,67 @@ class Client(OAuth2Mixin, DelegationMixin, TokenStorageMixin):
 
         return self._get_user_by_id(user_id)
 
+    def _parse_backfill_line(self, line: str) -> BackfillStatus | None:
+        """Parse a single NDJSON line from backfill status stream."""
+        try:
+            return BackfillStatus.model_validate_json(line)
+        except Exception:
+            pass
+        return None
+
+    def watch_backfill_status(self, *, auto_reconnect: bool = False) -> Generator[BackfillStatus, None, None]:
+        """Watches backfill status from the activities backfill-status endpoint.
+
+        This method connects to the backfill status event stream and yields
+        backfill_loaded_until timestamps as they are received. The connection
+        automatically closes after 60 seconds, but can be configured to auto-reconnect.
+
+        Args:
+            auto_reconnect: Whether to automatically reconnect when the connection
+                closes and continue receiving updates. Defaults to False.
+
+        Yields:
+            BackfillStatus: A BackfillStatus object for each received message.
+
+        Raises:
+            HTTPStatusError: If the API request fails.
+        """
+        while True:
+            try:
+                with self._http_client() as client:
+                    with client.stream("GET", "/api/v1/activities/backfill-status") as response:
+                        self._raise_for_status(response)
+
+                        for line in response.iter_lines():
+                            if line.strip():
+                                parsed = self._parse_backfill_line(line)
+                                if parsed:
+                                    yield parsed
+
+            except httpx.RequestError:
+                if not auto_reconnect:
+                    raise
+                time.sleep(1)
+            if not auto_reconnect:
+                break
+
+    def get_backfill_status(self) -> BackfillStatus:
+        """Gets the current backfill status from the activities backfill-status endpoint.
+
+        This method connects to the backfill status event stream and returns
+        the first backfill_loaded_until timestamp received.
+
+        Returns:
+            BackfillStatus: A BackfillStatus object containing the current backfill status.
+
+        Raises:
+            HTTPStatusError: If the API request fails.
+            ValueError: If no status message is received.
+        """
+        for status in self.watch_backfill_status(auto_reconnect=False):
+            return status
+        raise ValueError("No backfill status received")
+
 
 _default_client = Client()
 
@@ -1276,6 +1338,9 @@ _generate_singleton_methods(
         "get_userinfo",
         "whoami",
 
+        "get_backfill_status",
+        "watch_backfill_status",
+
         "get_activities",
 
         "get_activity",

sweatstack/openapi_schemas.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2025-08-01T16:59:37+00:00
+#   timestamp: 2025-08-07T11:42:52+00:00
 
 from __future__ import annotations
 
@@ -36,6 +36,12 @@ class Prompt(Enum):
     select_account = 'select_account'
 
 
+class BackfillStatus(BaseModel):
+    backfill_loaded_until: Optional[datetime] = Field(
+        ..., title='Backfill Loaded Until'
+    )
+
+
 class BodyAddEmailPartialsAddEmailPost(BaseModel):
     email: EmailStr = Field(..., title='Email')
 
@@ -664,6 +670,7 @@ class BodyLoginPostLoginPost(BaseModel):
     email: EmailStr = Field(..., title='Email')
     password: SecretStr = Field(..., title='Password')
     tz: Tz = Field(..., title='Tz')
+    state: Optional[str] = Field(None, title='State')
 
 
 class BodyRegisterPostRegisterPost(BaseModel):

sweatstack/schemas.py

@@ -2,7 +2,7 @@ from enum import Enum
 from typing import List, Union
 
 from .openapi_schemas import (
-    ActivityDetails, ActivitySummary, Metric, Scope, Sport,
+    ActivityDetails, ActivitySummary, BackfillStatus, Metric, Scope, Sport,
     TraceDetails, UserInfoResponse, UserSummary
 )
 

{sweatstack-0.46.0.dist-info → sweatstack-0.47.0.dist-info}/METADATA

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

{sweatstack-0.46.0.dist-info → sweatstack-0.47.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=bea8IY72fH93PES-dKMbNaB02k_GepRDzVMxEQuBI4g,46966
+sweatstack/client.py,sha256=31iiErq1nWk-zFgApTN_iCWn7an7bB3VOBygGZgAOJ8,49517
 sweatstack/constants.py,sha256=fGO6ksOv5HeISv9lHRoYm4besW1GTveXS8YD3K0ljg0,41
 sweatstack/ipython_init.py,sha256=OtBB9dQvyLXklD4kA2x1swaVtU9u73fG4V4-zz4YRAg,139
 sweatstack/jupyterlab_oauth2_startup.py,sha256=YcjXvzeZ459vL_dCkFi1IxX_RNAu80ZX9rwa0OXJfTM,1023
-sweatstack/openapi_schemas.py,sha256=GaA16d2jAryI1xsxAmRPzyyqhE4jvVKiF6omyIAK4bA,46028
+sweatstack/openapi_schemas.py,sha256=VvquBdbssdB9D1KeJYQCx51hy1Df4SS0PjzGWXcUaew,46221
 sweatstack/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sweatstack/schemas.py,sha256=cYefSktNv9VJoq4PQ5SrUvZRRwB637xNOaj7kjzjocM,3667
+sweatstack/schemas.py,sha256=44sXrNCYR0qtDsH8sUKR8df9zpcVA-TimkinXJBesOg,3683
 sweatstack/streamlit.py,sha256=_PER03s0dYu5eF1MZdewPDqSvYHqMr0lZLu_EnGit3Y,13257
 sweatstack/sweatshell.py,sha256=MYLNcWbOdceqKJ3S0Pe8dwHXEeYsGJNjQoYUXpMTftA,333
 sweatstack/utils.py,sha256=AwHRdC1ziOZ5o9RBIB21Uxm-DoClVRAJSVvgsmSmvps,1801
 sweatstack/Sweat Stack examples/Getting started.ipynb,sha256=k2hiSffWecoQ0VxjdpDcgFzBXDQiYEebhnAYlu8cgX8,6335204
-sweatstack-0.46.0.dist-info/METADATA,sha256=RLthZaCIu9YSePVaHuDfPX9TI5KpJX2Td0i0rm4TrYg,852
-sweatstack-0.46.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sweatstack-0.46.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
-sweatstack-0.46.0.dist-info/RECORD,,
+sweatstack-0.47.0.dist-info/METADATA,sha256=UuRlROP5y90JVM6eopj0PklP0WvdkB1TWxy6RwALrS0,852
+sweatstack-0.47.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sweatstack-0.47.0.dist-info/entry_points.txt,sha256=kCzOUQI3dqbTpEYqtgYDeiKFaqaA7BMlV6D24BMzCFU,208
+sweatstack-0.47.0.dist-info/RECORD,,