otf-api 0.13.2__py3-none-any.whl → 0.14.0__py3-none-any.whl

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/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/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/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/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/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/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.dist-info → otf_api-0.14.0.dist-info}/METADATA

@@ -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.dist-info → otf_api-0.14.0.dist-info}/RECORD

@@ -1,26 +1,26 @@
-otf_api/__init__.py,sha256=B8xq2bcHRV-z-pluFbo3Ow-FBAYO4jG5hq6wll9FhqE,1150
+otf_api/__init__.py,sha256=gg7_JvYNdWRmAD0TtUR0XXwcfH_erUEdWRYsPfH2un0,1150
 otf_api/cache.py,sha256=m4xUBhaS0MkcZypon1jjfhJzZPRQBE5Fto4_RMReXZ0,4311
 otf_api/exceptions.py,sha256=b6ZdH1dtYyUfXSupdVGGni6d66qqhzD0SGzyuty89gM,2174
 otf_api/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 otf_api/api/__init__.py,sha256=YC6bwYiAN6BSE7sqbxr2l34vwSdZPMGDv3wwPIQ_S3o,51
 otf_api/api/_compat.py,sha256=vWMC6kC3tK9qoXwnU5AgbQie-dhYAkkUI-hmq_4nf7Q,3512
 otf_api/api/api.py,sha256=yda8RObOMd_lBWbqlAAUZ3dTwGhdEYsfGQJDiphp6Eo,2633
-otf_api/api/client.py,sha256=o533H-nYweRcosBEYmm3U-7ivkpGDF7jElM9iIhK7E8,8157
+otf_api/api/client.py,sha256=wiHryvEWgdZVfjvX_3WoBCYe_OAf_kOj6yk1S4lxUcM,8734
 otf_api/api/utils.py,sha256=mNDUrz6d8b8KqTaUr5fGGGejy6bq9HHLs9LS8ibbfW4,9926
 otf_api/api/bookings/__init__.py,sha256=ocHSZXV4nnkZhjpjBX76iKHCQ21_ZL5hV4YKgUR0Wwg,62
 otf_api/api/bookings/booking_api.py,sha256=VdEX5WF3TIV5V_1n2DclBPqxhXQysNqyxJxeR6c05UM,22124
 otf_api/api/bookings/booking_client.py,sha256=qYaEomexGWP2N_WdWePMN7BR8aIKEpO71g5J0KaVfAc,4244
 otf_api/api/members/__init__.py,sha256=6mkeMiATRsNQTYQ37P7k6SWf9RZ9T5QY9-_r1sS1-vY,59
 otf_api/api/members/member_api.py,sha256=5FAPbRN_Dv3mLC1AmxcErqF1d1ypL5zVEn1u7gDbTZA,7232
-otf_api/api/members/member_client.py,sha256=d6v6vexmtiedw1-lXLMobbN8tTmRT-zHq7VzdJb5wDE,4528
+otf_api/api/members/member_client.py,sha256=-rEGnbBrwbeZf41bIxwj0sdciMmTi90qJHfvrD_npXI,4515
 otf_api/api/studios/__init__.py,sha256=vPir509hQOJL7uJ4yBCOHgooXVnXs6N5RlWs1tN54Uk,59
 otf_api/api/studios/studio_api.py,sha256=sn1jBebe4u-wE1VkHp2W96beUOFwHdiGzVSmnGNPZjg,6813
-otf_api/api/studios/studio_client.py,sha256=KWc4CF-2387WDUFYpMtIUm3KyPj1JdelCn7RH9L1FBI,4707
+otf_api/api/studios/studio_client.py,sha256=7KFbf3n7_zUKdmSZevYeYch9ovJ21pXIamvNMrlSyH4,4694
 otf_api/api/workouts/__init__.py,sha256=GPb2cEsAxoaiJIP8Finrk9x9vUeL9NXA9iz2eya9HYk,62
-otf_api/api/workouts/workout_api.py,sha256=prlqDFF4VXuzC7SRPpmD_yeZuMAp-dnfaSa52I4SM9o,13215
-otf_api/api/workouts/workout_client.py,sha256=MvEWbhPXywpQMSdsBSGCYBIR65TJjdqgb7T0Bc8H8D0,6762
+otf_api/api/workouts/workout_api.py,sha256=6h-ErOOfGRcQ2a3NaOyvbCYfWuHCteugaqNWQwZ7UHY,13374
+otf_api/api/workouts/workout_client.py,sha256=GA8jYCOp313uDMwhNUvmX5G4NMS3LarJCPkM-dJbTuM,6736
 otf_api/auth/__init__.py,sha256=uTvFTNQVvfogTN_-4D3bRSundAf2enUJ5ji6PwO0xm8,104
-otf_api/auth/auth.py,sha256=67-RXDla3rVm5vmlyPEOqPOF5kecJiEznVmq0gSz3M0,16448
+otf_api/auth/auth.py,sha256=N7PRohWlN4TMf6xWkD3LQTPaztGYqPFKgpHYMhzfZV8,17031
 otf_api/auth/user.py,sha256=OChRG0EhZ63vEtT_Sg3NmXWmOD1vZcILuHhRXlwaG0I,2156
 otf_api/auth/utils.py,sha256=YBxqg2h59u4V1ij5kgqJDyKh0gtgOlXrSZdpAKSdelY,3954
 otf_api/models/__init__.py,sha256=P0IjxRUPIfzUI-e3VDMpipJeWr6BlypWAED2ovnZw-w,1545
@@ -52,8 +52,8 @@ otf_api/models/workouts/out_of_studio_workout_history.py,sha256=-BKp-MnIp_5-U2KW
 otf_api/models/workouts/performance_summary.py,sha256=R1p-g1L4XWWEiBYaGtX5JFvoS3Z6bzQrsPIfTQHDqyM,3464
 otf_api/models/workouts/telemetry.py,sha256=rKsmLbsOVUxUXH439zxxIEWYGTohKNemqrtMamAl5cA,3465
 otf_api/models/workouts/workout.py,sha256=_KIR9fLI4LgUlS02Q5SNauHd4aOqJI9H4uaFKf-xJOU,4580
-otf_api-0.13.2.dist-info/licenses/LICENSE,sha256=UaPT9ynYigC3nX8n22_rC37n-qmTRKLFaHrtUwF9ktE,1071
-otf_api-0.13.2.dist-info/METADATA,sha256=ABfr_86oOf234G_WSqA06txQG2igkLdAx7KaNhabQFs,2162
-otf_api-0.13.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-otf_api-0.13.2.dist-info/top_level.txt,sha256=KAhYg1X2YG0LkTuVRhUV1I_AReNZUVNdEan7cp0pEE4,8
-otf_api-0.13.2.dist-info/RECORD,,
+otf_api-0.14.0.dist-info/licenses/LICENSE,sha256=UaPT9ynYigC3nX8n22_rC37n-qmTRKLFaHrtUwF9ktE,1071
+otf_api-0.14.0.dist-info/METADATA,sha256=Ut4IOCHQto6qCo9cd-bmUpbmw-dqHLXDss4gQU8cqSU,2162
+otf_api-0.14.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+otf_api-0.14.0.dist-info/top_level.txt,sha256=KAhYg1X2YG0LkTuVRhUV1I_AReNZUVNdEan7cp0pEE4,8
+otf_api-0.14.0.dist-info/RECORD,,