otf-api 0.10.0__py3-none-any.whl → 0.10.2__py3-none-any.whl

otf_api/__init__.py

@@ -4,7 +4,7 @@ from otf_api.api import Otf
 from otf_api import models
 from otf_api.auth import OtfUser
 
-__version__ = "0.10.0"
+__version__ = "0.10.2"
 
 
 __all__ = ["Otf", "OtfUser", "models"]

otf_api/api.py

@@ -1,7 +1,6 @@
 import atexit
 import contextlib
 import functools
-import warnings
 from concurrent.futures import ThreadPoolExecutor
 from copy import deepcopy
 from datetime import date, datetime, timedelta
@@ -68,6 +67,7 @@ class Otf:
         retry=retry_if_exception_type(exc.OtfRequestError),
         stop=stop_after_attempt(3),
         wait=wait_exponential(multiplier=1, min=4, max=10),
+        reraise=True,
     )
     def _do(
         self,
@@ -177,9 +177,10 @@ class Otf:
         """Retrieve raw member membership details."""
         return self._default_request("GET", f"/member/members/{self.member_uuid}/memberships")
 
-    def _get_performance_summaries_raw(self) -> dict:
+    def _get_performance_summaries_raw(self, limit: int | None = None) -> dict:
         """Retrieve raw performance summaries data."""
-        return self._performance_summary_request("GET", "/v1/performance-summaries")
+        params = {"limit": limit} if limit else {}
+        return self._performance_summary_request("GET", "/v1/performance-summaries", params=params)
 
     def _get_performance_summary_raw(self, performance_summary_id: str) -> dict:
         """Retrieve raw performance summary data."""
@@ -321,14 +322,16 @@ class Otf:
             },
         )
 
-    def _rate_class_raw(self, class_uuid: str, class_history_uuid: str, class_rating: int, coach_rating: int) -> dict:
+    def _rate_class_raw(
+        self, class_uuid: str, performance_summary_id: str, class_rating: int, coach_rating: int
+    ) -> dict:
         """Retrieve raw response from rating a class and coach."""
         return self._default_request(
             "POST",
             "/mobile/v1/members/classes/ratings",
             json={
                 "classUUId": class_uuid,
-                "otBeatClassHistoryUUId": class_history_uuid,
+                "otBeatClassHistoryUUId": performance_summary_id,
                 "classRating": class_rating,
                 "coachRating": coach_rating,
             },
@@ -919,6 +922,8 @@ class Otf:
         """Get detailed information about a specific studio. If no studio UUID is provided, it will default to the
         user's home studio.
 
+        If the studio is not found, it will return a StudioDetail object with default values.
+
         Args:
             studio_uuid (str, optional): The studio UUID to get detailed information about.
 
@@ -926,7 +931,11 @@ class Otf:
             StudioDetail: Detailed information about the studio.
         """
         studio_uuid = studio_uuid or self.home_studio_uuid
-        res = self._get_studio_detail_raw(studio_uuid)
+
+        try:
+            res = self._get_studio_detail_raw(studio_uuid)
+        except exc.ResourceNotFoundError:
+            return models.StudioDetail(studioUUId=studio_uuid, studioName="Studio Not Found", studioStatus="Unknown")
 
         return models.StudioDetail(**res["data"])
 
@@ -1103,9 +1112,12 @@ class Otf:
         return models.FitnessBenchmark(**data["Dto"][0])
 
     @cached(cache=TTLCache(maxsize=1024, ttl=600))
-    def get_performance_summaries_dict(self) -> dict[str, models.PerformanceSummary]:
+    def get_performance_summaries_dict(self, limit: int | None = None) -> dict[str, models.PerformanceSummary]:
         """Get a dictionary of performance summaries for the authenticated user.
 
+        Args:
+            limit (int | None): The maximum number of entries to return. Default is None.
+
         Returns:
             dict[str, PerformanceSummary]: A dictionary of performance summaries, keyed by class history UUID.
 
@@ -1115,7 +1127,7 @@ class Otf:
 
         """
 
-        items = self._get_performance_summaries_raw()["items"]
+        items = self._get_performance_summaries_raw(limit=limit)["items"]
 
         distinct_studio_ids = set([rec["class"]["studio"]["id"] for rec in items])
         perf_summary_ids = set([rec["id"] for rec in items])
@@ -1133,7 +1145,7 @@ class Otf:
             item["detail"] = perf_summary_dict[item["id"]]
 
         entries = [models.PerformanceSummary(**item) for item in items]
-        entries_dict = {entry.class_history_uuid: entry for entry in entries}
+        entries_dict = {entry.performance_summary_id: entry for entry in entries}
 
         return entries_dict
 
@@ -1141,7 +1153,7 @@ class Otf:
         """Get a list of all performance summaries for the authenticated user.
 
         Args:
-            limit (int | None): The maximum number of entries to return. Default is None. Deprecated.
+            limit (int | None): The maximum number of entries to return. Default is None.
 
         Returns:
             list[PerformanceSummary]: A list of performance summaries.
@@ -1152,26 +1164,33 @@ class Otf:
 
         """
 
-        if limit:
-            warnings.warn("Limit is deprecated and will be removed in a future version.", DeprecationWarning)
-
-        records = list(self.get_performance_summaries_dict().values())
+        records = list(self.get_performance_summaries_dict(limit=limit).values())
 
         sorted_records = sorted(records, key=lambda x: x.otf_class.starts_at, reverse=True)
 
         return sorted_records
 
-    def get_performance_summary(self, performance_summary_id: str) -> models.PerformanceSummary:
+    def get_performance_summary(
+        self, performance_summary_id: str, limit: int | None = None
+    ) -> models.PerformanceSummary:
         """Get performance summary for a given workout.
 
+        Note: Due to the way the OTF API is set up, we have to call both the list and the get endpoints. By
+        default this will call the list endpoint with no limit, in order to ensure that the performance summary
+        is returned if it exists. This could result in a lot of requests, so you also have the option to provide
+        a limit to only fetch a certain number of performance summaries.
+
         Args:
             performance_summary_id (str): The ID of the performance summary to retrieve.
 
         Returns:
             PerformanceSummary: The performance summary.
+
+        Raises:
+            ResourceNotFoundError: If the performance_summary_id is not in the list of performance summaries.
         """
 
-        perf_summary = self.get_performance_summaries_dict().get(performance_summary_id)
+        perf_summary = self.get_performance_summaries_dict(limit=limit).get(performance_summary_id)
 
         if perf_summary is None:
             raise exc.ResourceNotFoundError(f"Performance summary {performance_summary_id} not found")
@@ -1180,7 +1199,7 @@ class Otf:
 
     @functools.lru_cache(maxsize=1024)
     def _get_performancy_summary_detail(self, performance_summary_id: str) -> dict[str, Any]:
-        """Get the details for a performance summary.
+        """Get the details for a performance summary. Generally should not be called directly. This
 
         Args:
             performance_summary_id (str): The performance summary ID.
@@ -1350,7 +1369,7 @@ class Otf:
     def _rate_class(
         self,
         class_uuid: str,
-        class_history_uuid: str,
+        performance_summary_id: str,
         class_rating: Literal[0, 1, 2, 3],
         coach_rating: Literal[0, 1, 2, 3],
     ) -> models.PerformanceSummary:
@@ -1363,7 +1382,7 @@ class Otf:
 
         Args:
             class_uuid (str): The class UUID.
-            class_history_uuid (str): The performance summary ID.
+            performance_summary_id (str): The performance summary ID.
             class_rating (int): The class rating. Must be 0, 1, 2, or 3.
             coach_rating (int): The coach rating. Must be 0, 1, 2, or 3.
 
@@ -1390,10 +1409,10 @@ class Otf:
         body_coach_rating = COACH_RATING_MAP[coach_rating]
 
         try:
-            self._rate_class_raw(class_uuid, class_history_uuid, body_class_rating, body_coach_rating)
+            self._rate_class_raw(class_uuid, performance_summary_id, body_class_rating, body_coach_rating)
         except exc.OtfRequestError as e:
             if e.response.status_code == 403:
-                raise exc.AlreadyRatedError(f"Performance summary {class_history_uuid} is already rated.") from None
+                raise exc.AlreadyRatedError(f"Performance summary {performance_summary_id} is already rated.") from None
             raise
 
         # we have to clear the cache after rating a class, otherwise we will get back the same data
@@ -1403,7 +1422,7 @@ class Otf:
         # NOTE: the individual perf summary endpoint does not have rating data, so it's cache is not cleared
         self.get_performance_summaries_dict.cache_clear()
 
-        return self.get_performance_summary(class_history_uuid)
+        return self.get_performance_summary(performance_summary_id)
 
     def rate_class_from_performance_summary(
         self,
@@ -1429,18 +1448,20 @@ class Otf:
         """
 
         if perf_summary.is_rated:
-            raise exc.AlreadyRatedError(f"Performance summary {perf_summary.class_history_uuid} is already rated.")
+            raise exc.AlreadyRatedError(f"Performance summary {perf_summary.performance_summary_id} is already rated.")
 
         if not perf_summary.ratable:
-            raise exc.ClassNotRatableError(f"Performance summary {perf_summary.class_history_uuid} is not rateable.")
+            raise exc.ClassNotRatableError(
+                f"Performance summary {perf_summary.performance_summary_id} is not rateable."
+            )
 
         if not perf_summary.otf_class or not perf_summary.otf_class.class_uuid:
             raise ValueError(
-                f"Performance summary {perf_summary.class_history_uuid} does not have an associated class."
+                f"Performance summary {perf_summary.performance_summary_id} does not have an associated class."
             )
 
         return self._rate_class(
-            perf_summary.otf_class.class_uuid, perf_summary.class_history_uuid, class_rating, coach_rating
+            perf_summary.otf_class.class_uuid, perf_summary.performance_summary_id, class_rating, coach_rating
         )
 
     # the below do not return any data for me, so I can't test them

otf_api/models/bookings.py

@@ -36,6 +36,10 @@ class OtfClass(OtfItemBase):
     program_name: str | None = Field(None, alias="programName", exclude=True, repr=False)
     virtual_class: bool | None = Field(None, alias="virtualClass", exclude=True, repr=False)
 
+    def __str__(self) -> str:
+        starts_at_str = self.starts_at.strftime("%a %b %d, %I:%M %p")
+        return f"Class: {starts_at_str} {self.name} - {self.coach.first_name}"
+
 
 class Booking(OtfItemBase):
     booking_uuid: str = Field(alias="classBookingUUId", description="ID used to cancel the booking")

otf_api/models/classes.py

@@ -33,7 +33,9 @@ class OtfClass(OtfItemBase):
     is_home_studio: bool | None = Field(None, description="Custom helper field to determine if at home studio")
 
     # unused fields
-    class_id: str | None = Field(None, alias="id", exclude=True, repr=False, description="Not used by API")
+    class_id: str | None = Field(
+        None, alias="id", exclude=True, repr=False, description="Matches new booking endpoint class id"
+    )
 
     created_at: datetime | None = Field(None, exclude=True, repr=False)
     ends_at_utc: datetime | None = Field(None, alias="ends_at", exclude=True, repr=False)

otf_api/models/enums.py

@@ -8,6 +8,7 @@ class StudioStatus(StrEnum):
     COMING_SOON = "Coming Soon"
     TEMP_CLOSED = "Temporarily Closed"
     PERM_CLOSED = "Permanently Closed"
+    UNKNOWN = "Unknown"
 
 
 class BookingStatus(StrEnum):

otf_api/models/performance_summary.py

@@ -105,7 +105,8 @@ class PerformanceSummary(OtfItemBase):
 
     """
 
-    class_history_uuid: str = Field(..., alias="id")
+    performance_summary_id: str = Field(..., alias="id", description="Unique identifier for this performance summary")
+    class_history_uuid: str = Field(..., alias="id", description="Same as performance_summary_id")
     ratable: bool | None = None
     otf_class: Class | None = Field(None, alias="class")
     coach: str | None = Field(None, alias=AliasPath("class", "coach", "first_name"))

otf_api/models/studio_detail.py

@@ -9,8 +9,8 @@ from otf_api.models.mixins import AddressMixin
 
 class StudioLocation(AddressMixin):
     phone_number: str | None = Field(None, alias=AliasChoices("phone", "phoneNumber"))
-    latitude: float = Field(..., alias=AliasChoices("latitude"))
-    longitude: float = Field(..., alias=AliasChoices("longitude"))
+    latitude: float | None = Field(None, alias=AliasChoices("latitude"))
+    longitude: float | None = Field(None, alias=AliasChoices("longitude"))
 
     physical_region: str | None = Field(None, alias="physicalRegion", exclude=True, repr=False)
     physical_country_id: int | None = Field(None, alias="physicalCountryId", exclude=True, repr=False)
@@ -27,7 +27,7 @@ class StudioDetail(OtfItemBase):
         exclude=True,
         repr=False,
     )
-    location: StudioLocation = Field(..., alias="studioLocation")
+    location: StudioLocation = Field(..., alias="studioLocation", default_factory=StudioLocation)
     name: str | None = Field(None, alias="studioName")
     status: StudioStatus | None = Field(
         None, alias="studioStatus", description="Active, Temporarily Closed, Coming Soon"

otf_api/models/telemetry.py

@@ -49,7 +49,10 @@ class TelemetryItem(OtfItemBase):
 
 class Telemetry(OtfItemBase):
     member_uuid: str = Field(..., alias="memberUuid")
-    class_history_uuid: str = Field(..., alias="classHistoryUuid")
+    performance_summary_id: str = Field(
+        ..., alias="classHistoryUuid", description="The ID of the performance summary this telemetry item belongs to."
+    )
+    class_history_uuid: str = Field(..., alias="classHistoryUuid", description="The same as performance_summary_id.")
     class_start_time: datetime | None = Field(None, alias="classStartTime")
     max_hr: int | None = Field(None, alias="maxHr")
     zones: Zones

{otf_api-0.10.0.dist-info → otf_api-0.10.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: otf-api
-Version: 0.10.0
+Version: 0.10.2
 Summary: Python OrangeTheory Fitness API Client
 License: MIT
 Author: Jessica Smith

{otf_api-0.10.0.dist-info → otf_api-0.10.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-otf_api/__init__.py,sha256=4GiRSBfop-C9ihSYXXzkrtTsM4AVuU3lKe3GI6-s-q0,205
-otf_api/api.py,sha256=TIHqZ2a6DmtpXAwGm28grz0Hy4BXHQ1-k1v_Cz1LmDg,59228
+otf_api/__init__.py,sha256=B_mniIQTSJOh_sn2p57CXH1bN6lFiBpPuP8fqj9itQE,205
+otf_api/api.py,sha256=OmEwG3Zrfq1OQ-WoiaYg1LzsAPx8nsy6ft5ZSWb2xiQ,60334
 otf_api/auth/__init__.py,sha256=Pu6ugbEXhgjJEpYKr7K2CqEac53Wq9FRF3F16GLQ_ww,133
 otf_api/auth/auth.py,sha256=m9qzUn0hdAo6AZHm2Mn054nLqUWPxEyP3y3DOBVkvYA,9010
 otf_api/auth/user.py,sha256=HgggfneeVMnKFWSUMnlikfZyC__HE8vVaKSpBS4_sIg,2256
@@ -10,11 +10,11 @@ otf_api/logging.py,sha256=PRZpCaJ1F1Xya3L9Efkt3mKS5_QNr3sXjEUERSxYjvE,563
 otf_api/models/__init__.py,sha256=Ak1Wph5Yn9Npu3m3vvG-n5CsMH6j7NCuzoOBd0b_OlI,1524
 otf_api/models/base.py,sha256=KJlIxl_sRj6f-g5vKYPw4yV6fGDk-fwZ93EO0JGPYMw,202
 otf_api/models/body_composition_list.py,sha256=jGdR-9ScvIOtULJNB99aYh2INk2ihoHAnTWtbQCIea4,12202
-otf_api/models/bookings.py,sha256=8zgLnQ40ci0c7AimGeoAI-Raw8d2byvKoXkJXyW1xUE,4260
+otf_api/models/bookings.py,sha256=Lj-IHN1k-w4vF-5aIKbsiQ6Uq-I5_ELoPUKXPGkfbgM,4440
 otf_api/models/challenge_tracker_content.py,sha256=zukbda38DloiBHW9HUEKQnumGhgVqpjjbxo7G4bDIc0,2630
 otf_api/models/challenge_tracker_detail.py,sha256=KDfLRDjr8quVhDPVj9x_tuLnkJPua5y8C3WtGIVzXZw,4356
-otf_api/models/classes.py,sha256=DZ7oo_MNK7m3ul2gK929maGBksekG9RtLzYa-Di8skk,3005
-otf_api/models/enums.py,sha256=ssv32lKF7CQb0GFPzYf0mUzdrEZotHtCXIjPA9hGTj4,4580
+otf_api/models/classes.py,sha256=TIbFL-haguBqYSSLUEBzPAdv2B-zHNUvcmxcTSu0o6c,3041
+otf_api/models/enums.py,sha256=VyG3w-kcpDMTsHvX7s6xp-cI61mqvsmN_A76jsnmKj0,4604
 otf_api/models/lifetime_stats.py,sha256=sQw0oaT5uSlCTZyMSEEhYJg7IdOzuxelTI3mzDBHNls,3055
 otf_api/models/member_detail.py,sha256=CDDZg3Ow07U57yRqbm3i-BVha0cvCNOZ8QhN0pliU_A,6598
 otf_api/models/member_membership.py,sha256=jZwHzwtVyMUr8dWGlFbMYj9qveCbiOblWW5szXDUFFo,1338
@@ -22,13 +22,13 @@ otf_api/models/member_purchases.py,sha256=Ne7ByEbGTqTJhuEyCgywWe8I3nc-D46qw09up7
 otf_api/models/mixins.py,sha256=VR5EeweySHhzaiqqnCr853Cpe1uK97cxY0IFEdf5T8w,2262
 otf_api/models/notifications.py,sha256=AkmIfiIiU6wox_7puyenbhCX10SFvBD5eBAovcurRgY,833
 otf_api/models/out_of_studio_workout_history.py,sha256=El5i0K2Td_sMReyfUKP-Iv1L1WgRx0ijjjnHzYvlCeY,1703
-otf_api/models/performance_summary.py,sha256=yC5Inyc1S3Q48DbcCxxfmSSsKh7uZ8PTv9bziGnrf_I,5901
-otf_api/models/studio_detail.py,sha256=tTjKGXIGH-kdUZ1d_H4phwfEcGHAiDO8PEbV2lm3Q5c,3681
+otf_api/models/performance_summary.py,sha256=0pg1TTnYEk7HW6exA3Czclt1YMtYRBusgPQFWN_udGo,6066
+otf_api/models/studio_detail.py,sha256=DZicrQWee8iexyK5I8UN1wBYevTB_1dciz2EKBhPids,3729
 otf_api/models/studio_services.py,sha256=aGLQMQmjGVpI6YxzAl-mcp3Y9cHPXuH9dIqrl6E-78E,1665
-otf_api/models/telemetry.py,sha256=_g-wtJDbaXENk1DqH0bbpJUsGepWDYraRIsIvpvDBU8,2531
+otf_api/models/telemetry.py,sha256=LuoQ-W7wWQ-trjQqfITQsUpkWL1X3TXdsHiDhyIDvUk,2748
 otf_api/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 otf_api/utils.py,sha256=-9JBjRmoS6xpua_lHccJYjEX2tWp2-5QQrbbuOYMyIo,3945
-otf_api-0.10.0.dist-info/LICENSE,sha256=UaPT9ynYigC3nX8n22_rC37n-qmTRKLFaHrtUwF9ktE,1071
-otf_api-0.10.0.dist-info/METADATA,sha256=mMqM2O0X4CcR2ZgytM_WEvVhbo-4hclF9X2TZ0RyPPo,2375
-otf_api-0.10.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-otf_api-0.10.0.dist-info/RECORD,,
+otf_api-0.10.2.dist-info/LICENSE,sha256=UaPT9ynYigC3nX8n22_rC37n-qmTRKLFaHrtUwF9ktE,1071
+otf_api-0.10.2.dist-info/METADATA,sha256=ZAojA_OhEwYb2vHLBDErKWzKgpOXmgbYvQ5TL1PHgt8,2375
+otf_api-0.10.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+otf_api-0.10.2.dist-info/RECORD,,

{otf_api-0.10.0.dist-info → otf_api-0.10.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.1
+Generator: poetry-core 2.1.2
 Root-Is-Purelib: true
 Tag: py3-none-any