otf-api 0.13.2__tar.gz → 0.14.0__tar.gz

{otf_api-0.13.2/src/otf_api.egg-info → otf_api-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: otf-api
-Version: 0.13.2
+Version: 0.14.0
 Summary: Python OrangeTheory Fitness API Client
 Author-email: Jessica Smith <j.smith.git1@gmail.com>
 License-Expression: MIT

{otf_api-0.13.2 → otf_api-0.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "otf-api"
-version = "0.13.2"
+version = "0.14.0"
 description = "Python OrangeTheory Fitness API Client"
 authors = [{ name = "Jessica Smith", email = "j.smith.git1@gmail.com" }]
 requires-python = ">=3.11"
@@ -39,7 +39,6 @@ Documentation = "https://otf-api.readthedocs.io/en/stable/"
 dev = [
     "build==1.2.1",
     "bump-my-version>=0.23.0,<0.24",
-    "mypy==1.10.0",
     "mypy-boto3-cognito-identity>=1.38.0",
     "mypy-boto3-cognito-idp>=1.35.93,<2",
     "pre-commit==3.7.1",
@@ -51,7 +50,7 @@ dev = [
 docs = [
     "autodoc-pydantic>=2.2.0",
     "furo>=2024.8.6",
-    "sphinx==8.3.0",
+    "sphinx>=8.3.0",
     "sphinx-autobuild==2024.10.3",
     "sphinx-autodoc-typehints==3.2.0",
     "sphinx-autodoc2==0.5.0",
@@ -72,5 +71,5 @@ default-groups = ["dev"]
 pythonpath = ["src"]
 
 [build-system]
-requires = ["setuptools >= 61.0"]
+requires = ["setuptools >= 80.0"]
 build-backend = "setuptools.build_meta"

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/__init__.py

@@ -36,7 +36,7 @@ def _setup_logging() -> None:
 
 _setup_logging()
 
-__version__ = "0.13.2"
+__version__ = "0.14.0"
 
 
 __all__ = ["Otf", "OtfUser", "models"]

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/api/client.py

@@ -49,6 +49,21 @@ class OtfClient:
         )
         atexit.register(self.session.close)
 
+    def __getstate__(self):
+        """Get the state of the OtfClient instance for serialization."""
+        state = self.__dict__.copy()
+        # Remove circular references
+        state.pop("session", None)
+        return state
+
+    def __setstate__(self, state):  # noqa
+        """Set the state of the OtfClient instance from serialized data."""
+        self.__dict__.update(state)
+        self.session = httpx.Client(
+            headers=HEADERS, auth=self.user.httpx_auth, timeout=httpx.Timeout(20.0, connect=60.0)
+        )
+        atexit.register(self.session.close)
+
     def _build_request(
         self,
         method: str,

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/api/members/member_client.py

@@ -14,7 +14,7 @@ class MemberClient:
         self.client = client
         self.member_uuid = client.member_uuid
 
-    @CACHE.memoize(expire=600, tag="member_detail", ignore=(0,))
+    @CACHE.memoize(expire=600, tag="member_detail")
     def get_member_detail(self) -> dict:
         """Retrieve raw member details."""
         return self.client.default_request(

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/api/studios/studio_client.py

@@ -15,7 +15,7 @@ class StudioClient:
         self.client = client
         self.member_uuid = client.member_uuid
 
-    @CACHE.memoize(expire=600, tag="studio_detail", ignore=(0,))
+    @CACHE.memoize(expire=600, tag="studio_detail")
     def get_studio_detail(self, studio_uuid: str) -> dict:
         """Retrieve raw studio details."""
         return self.client.default_request("GET", f"/mobile/v1/studios/{studio_uuid}")["data"]

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/api/workouts/workout_api.py

@@ -242,13 +242,14 @@ class WorkoutApi:
         return workout
 
     def get_workouts(
-        self, start_date: date | str | None = None, end_date: date | str | None = None
+        self, start_date: date | str | None = None, end_date: date | str | None = None, max_data_points: int = 150
     ) -> list[models.Workout]:
         """Get the member's workouts, using the new bookings endpoint and the performance summary endpoint.
 
         Args:
             start_date (date | str | None): The start date for the workouts. If None, defaults to 30 days ago.
             end_date (date | str | None): The end date for the workouts. If None, defaults to today.
+            max_data_points (int): The maximum number of data points to return for the telemetry. Default is 150.
 
         Returns:
             list[Workout]: The member's workouts.
@@ -265,7 +266,7 @@ class WorkoutApi:
         bookings_dict = {b.workout.id: b for b in bookings if b.workout}
 
         perf_summaries_dict = self.client.get_perf_summaries_threaded(list(bookings_dict.keys()))
-        telemetry_dict = self.client.get_telemetry_threaded(list(perf_summaries_dict.keys()))
+        telemetry_dict = self.client.get_telemetry_threaded(list(perf_summaries_dict.keys()), max_data_points)
         perf_summary_to_class_uuid_map = self.client.get_perf_summary_to_class_uuid_mapping()
 
         workouts: list[models.Workout] = []

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/api/workouts/workout_client.py

@@ -43,7 +43,7 @@ class WorkoutClient:
         params = {"limit": limit} if limit else {}
         return self.performance_summary_request("GET", "/v1/performance-summaries", params=params)
 
-    @CACHE.memoize(expire=600, tag="performance_summary", ignore=(0,))
+    @CACHE.memoize(expire=600, tag="performance_summary")
     def get_performance_summary(self, performance_summary_id: str) -> dict:
         """Retrieve raw performance summary data."""
         return self.performance_summary_request("GET", f"/v1/performance-summaries/{performance_summary_id}")
@@ -54,7 +54,7 @@ class WorkoutClient:
             "history"
         ]
 
-    @CACHE.memoize(expire=600, tag="telemetry", ignore=(0,))
+    @CACHE.memoize(expire=600, tag="telemetry")
     def get_telemetry(self, performance_summary_id: str, max_data_points: int = 150) -> dict:
         """Retrieve raw telemetry data."""
         return self.telemetry_request(

{otf_api-0.13.2 → otf_api-0.14.0}/src/otf_api/auth/auth.py

@@ -340,6 +340,20 @@ class OtfCognito(Cognito):
         self.device_group_key = device_metadata.get("DeviceGroupKey", self.device_group_key)
         CACHE.write_device_data_to_cache(self.device_metadata)
 
+    def __getstate__(self):
+        """Get the state of the object for pickling."""
+        state = self.__dict__.copy()
+        del state["idp_client"]
+        del state["id_client"]
+        return state
+
+    def __setstate__(self, state):  # noqa
+        """Set the state of the object from a pickled state."""
+        self.__dict__.update(state)
+        self.idp_client = Session().client("cognito-idp", config=BOTO_CONFIG, region_name=REGION)  # type: ignore
+
+        self.id_client = Session().client("cognito-identity", config=BOTO_CONFIG, region_name=REGION)  # type: ignore
+
 
 class HttpxCognitoAuth(httpx.Auth):
     http_header: str = "Authorization"

{otf_api-0.13.2 → otf_api-0.14.0/src/otf_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: otf-api
-Version: 0.13.2
+Version: 0.14.0
 Summary: Python OrangeTheory Fitness API Client
 Author-email: Jessica Smith <j.smith.git1@gmail.com>
 License-Expression: MIT