otf-api 0.11.0rc1__py3-none-any.whl → 0.11.1__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.11.0-rc1"
+__version__ = "0.11.1"
 
 
 __all__ = ["Otf", "OtfUser", "models"]

otf_api/api.py

@@ -18,7 +18,7 @@ from otf_api import exceptions as exc
 from otf_api import filters, models
 from otf_api.auth import OtfUser
 from otf_api.models.enums import HISTORICAL_BOOKING_STATUSES
-from otf_api.utils import ensure_date, ensure_datetime, ensure_list, get_booking_uuid, get_class_uuid  # get_booking_id
+from otf_api.utils import ensure_date, ensure_datetime, ensure_list, get_booking_id, get_booking_uuid, get_class_uuid
 
 API_BASE_URL = "api.orangetheory.co"
 API_IO_BASE_URL = "api.orangetheory.io"
@@ -69,7 +69,7 @@ class Otf:
         return hash(self.member_uuid)
 
     @retry(
-        retry=retry_if_exception_type(exc.OtfRequestError),
+        retry=retry_if_exception_type((exc.OtfRequestError, httpx.HTTPStatusError)),
         stop=stop_after_attempt(3),
         wait=wait_exponential(multiplier=1, min=4, max=10),
         reraise=True,
@@ -123,8 +123,11 @@ class Otf:
             raise
 
         if not response.text:
-            # insanely enough, at least one endpoint (get perf summary) returns None without error instead of 404
-            raise exc.OtfRequestError("Empty response", None, response=response, request=request)
+            if method == "GET":
+                raise exc.OtfRequestError("Empty response", None, response=response, request=request)
+
+            LOGGER.debug(f"Request {method!r} to {full_url!r} returned no content")
+            return None
 
         try:
             resp = response.json()
@@ -144,10 +147,15 @@ class Otf:
         return resp
 
     def _classes_request(
-        self, method: str, url: str, params: dict[str, Any] | None = None, headers: dict[str, Any] | None = None
+        self,
+        method: str,
+        url: str,
+        params: dict[str, Any] | None = None,
+        headers: dict[str, Any] | None = None,
+        **kwargs: Any,
     ) -> Any:
         """Perform an API request to the classes API."""
-        return self._do(method, API_IO_BASE_URL, url, params, headers=headers)
+        return self._do(method, API_IO_BASE_URL, url, params, headers=headers, **kwargs)
 
     def _default_request(
         self,
@@ -175,7 +183,7 @@ class Otf:
 
         return self._do(method, API_IO_BASE_URL, url, params, headers=headers)
 
-    def _get_classes_raw(self, studio_uuids: list[str] | None) -> dict:
+    def _get_classes_raw(self, studio_uuids: list[str]) -> dict:
         """Retrieve raw class data."""
         return self._classes_request("GET", "/v1/classes", params={"studio_ids": studio_uuids})
 
@@ -203,6 +211,11 @@ class Otf:
             raise exc.OtfException(f"Error booking class {class_uuid}: {e}")
         return resp
 
+    def _book_class_new_raw(self, body: dict[str, str | bool]) -> dict:
+        """Book a class by class_id."""
+
+        return self._classes_request("POST", "/v1/bookings/me", json=body)
+
     def _get_booking_raw(self, booking_uuid: str) -> dict:
         """Retrieve raw booking data."""
         return self._default_request("GET", f"/member/members/{self.member_uuid}/bookings/{booking_uuid}")
@@ -240,15 +253,7 @@ class Otf:
 
     def _cancel_booking_new_raw(self, booking_id: str) -> dict:
         """Cancel a booking by booking_id."""
-        return self._classes_request("DELETE", f"/v1/bookings/me/{booking_id}", headers={"SIGV4AUTH_REQUIRED": "true"})
-
-    def _get_booking_new_raw(self, booking_id: str) -> dict:
-        """Retrieve raw booking data."""
-        return self._classes_request("GET", f"/v1/bookings/me/{booking_id}", headers={"SIGV4AUTH_REQUIRED": "true"})
-
-    def _book_class_new_raw(self, class_id: str) -> dict:
-        """Book a class by class_id."""
-        return self._classes_request("POST", f"/v1/bookings/me/{class_id}", headers={"SIGV4AUTH_REQUIRED": "true"})
+        return self._classes_request("DELETE", f"/v1/bookings/me/{booking_id}")
 
     def _get_member_detail_raw(self) -> dict:
         """Retrieve raw member details."""
@@ -434,6 +439,9 @@ class Otf:
         end_date = pendulum.today().start_of("day").add(days=45)
         return self.get_bookings_new(start_date, end_date, exclude_canceled=False)
 
+    def _get_app_config_raw(self) -> dict[str, Any]:
+        return self._default_request("GET", "/member/app-configurations", headers={"SIGV4AUTH_REQUIRED": "true"})
+
     def get_bookings_new(
         self,
         start_dtme: datetime | str | None = None,
@@ -474,10 +482,13 @@ class Otf:
 
         return [models.BookingV2(**b) for b in bookings_resp["items"]]
 
-    # def get_booking_new(self, booking_id: str) -> models.BookingV2:
-    #     """Get a booking by ID."""
-    #     booking_resp = self._get_booking_new_raw(booking_id)
-    #     return models.BookingV2(**booking_resp)
+    def get_booking_new(self, booking_id: str) -> models.BookingV2:
+        """Get a booking by ID."""
+        all_bookings = self._get_all_bookings_new()
+        booking = next((b for b in all_bookings if b.booking_id == booking_id), None)
+        if not booking:
+            raise exc.ResourceNotFoundError(f"Booking with ID {booking_id} not found")
+        return booking
 
     def get_classes(
         self,
@@ -493,11 +504,11 @@ class Otf:
         UUIDs are provided, it will default to the user's home studio.
 
         Args:
-            start_date (date | None): The start date for the classes. Default is None.
-            end_date (date | None): The end date for the classes. Default is None.
+            start_date (date | str | None): The start date for the classes. Default is None.
+            end_date (date | str | None): The end date for the classes. Default is None.
             studio_uuids (list[str] | None): The studio UUIDs to get the classes for. Default is None, which will\
             default to the user's home studio only.
-            include_home_studio (bool): Whether to include the home studio in the classes. Default is True.
+            include_home_studio (bool | None): Whether to include the home studio in the classes. Default is True.
             filters (list[ClassFilter] | ClassFilter | None): A list of filters to apply to the classes, or a single\
             filter. Filters are applied as an OR operation. Default is None.
 
@@ -726,6 +737,26 @@ class Otf:
 
         return booking
 
+    def book_class_new(self, class_id: str) -> models.BookingV2:
+        """Book a class by providing the class_id.
+
+        Args:
+            class_id (str): The class ID to book.
+
+        Returns:
+            BookingV2: The booking.
+        """
+        if not class_id:
+            raise ValueError("class_id is required")
+
+        body = {"class_id": class_id, "confirmed": False, "waitlist": False}
+
+        resp = self._book_class_new_raw(body)
+
+        new_booking = models.BookingV2(**resp)
+
+        return new_booking
+
     def _check_class_already_booked(self, class_uuid: str) -> None:
         """Check if the class is already booked.
 
@@ -779,17 +810,14 @@ class Otf:
             ValueError: If booking_uuid is None or empty string
             BookingNotFoundError: If the booking does not exist.
         """
-        # if isinstance(booking, models.BookingV2):
-        #     LOGGER.warning("BookingV2 object provided, using the new cancel booking endpoint (`cancel_booking_new`)")
-        #     self.cancel_booking_new(booking)
+        if isinstance(booking, models.BookingV2):
+            LOGGER.warning("BookingV2 object provided, using the new cancel booking endpoint (`cancel_booking_new`)")
+            self.cancel_booking_new(booking)
 
         booking_uuid = get_booking_uuid(booking)
 
         if booking == booking_uuid:  # ensure this booking exists by calling the booking endpoint
-            try:
-                self.get_booking(booking_uuid)
-            except Exception:
-                raise exc.BookingNotFoundError(f"Booking {booking_uuid} does not exist.")
+            _ = self.get_booking(booking_uuid)  # allow the exception to be raised if it doesn't exist
 
         resp = self._cancel_booking_raw(booking_uuid)
         if resp["code"] == "NOT_AUTHORIZED" and resp["message"].startswith("This class booking has"):
@@ -797,25 +825,27 @@ class Otf:
                 f"Booking {booking_uuid} is already cancelled.", booking_uuid=booking_uuid
             )
 
-    # def cancel_booking_new(self, booking: str | models.BookingV2) -> None:
-    #     """Cancel a booking by providing either the booking_id or the BookingV2 object.
-    #     Args:
-    #         booking (str | BookingV2): The booking ID or the BookingV2 object to cancel.
-    #     Raises:
-    #         ValueError: If booking_id is None or empty string
-    #         BookingNotFoundError: If the booking does not exist.
-    #     """
+    def cancel_booking_new(self, booking: str | models.BookingV2) -> None:
+        """Cancel a booking by providing either the booking_id or the BookingV2 object.
+
+        Args:
+            booking (str | BookingV2): The booking ID or the BookingV2 object to cancel.
+
+        Raises:
+            ValueError: If booking_id is None or empty string
+            BookingNotFoundError: If the booking does not exist.
+        """
 
-    #     if isinstance(booking, models.Booking):
-    #         LOGGER.warning("Booking object provided, using the old cancel booking endpoint (`cancel_booking`)")
-    #         self.cancel_booking(booking)
+        if isinstance(booking, models.Booking):
+            LOGGER.warning("Booking object provided, using the old cancel booking endpoint (`cancel_booking`)")
+            self.cancel_booking(booking)
 
-    #     booking_id = get_booking_id(booking)
+        booking_id = get_booking_id(booking)
 
-    #     if booking == booking_id:  # ensure this booking exists by calling the booking endpoint
-    #         self.get_booking_new(booking_id)
+        if booking == booking_id:
+            _ = self.get_booking_new(booking_id)  # allow the exception to be raised if it doesn't exist
 
-    #     self._cancel_booking_new_raw(booking_id)
+        self._cancel_booking_new_raw(booking_id)
 
     def get_bookings(
         self,
@@ -1483,32 +1513,31 @@ class Otf:
                 raise exc.AlreadyRatedError(f"Workout {performance_summary_id} is already rated.") from None
             raise
 
-    # def get_workout_from_booking(self, booking: str | models.BookingV2) -> models.Workout:
-    #     """Get a workout for a specific booking.
+    def get_workout_from_booking(self, booking: str | models.BookingV2) -> models.Workout:
+        """Get a workout for a specific booking.
+
+        Args:
+            booking (str | Booking): The booking ID or BookingV2 object to get the workout for.
 
-    #     Args:
-    #         booking_id (str | Booking): The booking ID or Booking object to get the workout for.
+        Returns:
+            Workout: The member's workout.
 
-    #     Returns:
-    #         Workout: The member's workout.
+        Raises:
+            BookingNotFoundError: If the booking does not exist.
+            ResourceNotFoundError: If the workout does not exist.
+        """
+        booking_id = get_booking_id(booking)
 
-    #     Raises:
-    #         BookingNotFoundError: If the booking does not exist.
-    #         ResourceNotFoundError: If the workout does not exist.
-    #     """
-    #     booking_id = booking if isinstance(booking, str) else booking.booking_id
+        booking = self.get_booking_new(booking_id)
 
-    #     booking = self.get_booking_new(booking_id)
-    #     if not booking:
-    #         raise exc.BookingNotFoundError(f"Booking {booking_id} not found.")
+        if not booking.workout or not booking.workout.performance_summary_id:
+            raise exc.ResourceNotFoundError(f"Workout for booking {booking_id} not found.")
 
-    #     if not booking.workout or not booking.workout.performance_summary_id:
-    #         raise exc.ResourceNotFoundError(f"Workout for booking {booking_id} not found.")
+        perf_summary = self._get_performance_summary_raw(booking.workout.performance_summary_id)
+        telemetry = self.get_telemetry(booking.workout.performance_summary_id)
+        workout = models.Workout(**perf_summary, v2_booking=booking, telemetry=telemetry)
 
-    #     perf_summary = self._get_performance_summary_raw(booking.workout.performance_summary_id)
-    #     telemetry = self.get_telemetry(booking.workout.performance_summary_id)
-    #     workout = models.Workout(**perf_summary, v2_booking=booking, telemetry=telemetry)
-    #     return workout
+        return workout
 
     def get_workouts(
         self, start_date: date | str | None = None, end_date: date | str | None = None
@@ -1516,8 +1545,8 @@ class Otf:
         """Get the member's workouts, using the new bookings endpoint and the performance summary endpoint.
 
         Args:
-            start_date (date | None): The start date for the workouts. If None, defaults to 30 days ago.
-            end_date (date | None): The end date for the workouts. If None, defaults to today.
+            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.
 
         Returns:
             list[Workout]: The member's workouts.
@@ -1621,19 +1650,7 @@ class Otf:
 
         self.rate_class(workout.class_uuid, workout.performance_summary_id, class_rating, coach_rating)
 
-        # TODO: use this once we get it working, getting workout list is substitute for now
-        # return self.get_workout_from_booking(workout.booking_id)
-
-        workout_date = pendulum.instance(workout.otf_class.starts_at).start_of("day")
-        workouts = self.get_workouts(workout_date.subtract(days=1), workout_date.add(days=1))
-
-        selected_workout = next(
-            (w for w in workouts if w.performance_summary_id == workout.performance_summary_id), None
-        )
-
-        assert selected_workout is not None, "Workout not found in the list of workouts"
-
-        return selected_workout
+        return self.get_workout_from_booking(workout.booking_id)
 
     # the below do not return any data for me, so I can't test them
 

otf_api/auth/auth.py

@@ -29,7 +29,6 @@ LOGGER = getLogger(__name__)
 CLIENT_ID = "1457d19r0pcjgmp5agooi0rb1b"  # from android app
 USER_POOL_ID = "us-east-1_dYDxUeyL1"
 REGION = "us-east-1"
-COGNITO_IDP_URL = f"https://cognito-idp.{REGION}.amazonaws.com/"
 
 ID_POOL_ID = "us-east-1:4943c880-fb02-4fd7-bc37-2f4c32ecb2a3"
 PROVIDER_KEY = f"cognito-idp.{REGION}.amazonaws.com/{USER_POOL_ID}"
@@ -296,6 +295,8 @@ class HttpxCognitoAuth(httpx.Auth):
 
         request.headers[self.http_header] = self.http_header_prefix + token
 
+        # If the request has the SIGV4AUTH_REQUIRED header, sign the request
+        # used by very few endpoints, but I expect that may change in the future
         if request.headers.get("SIGV4AUTH_REQUIRED"):
             del request.headers["SIGV4AUTH_REQUIRED"]
             yield from self.sign_httpx_request(request)

{otf_api-0.11.0rc1.dist-info → otf_api-0.11.1.dist-info}/METADATA

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

{otf_api-0.11.0rc1.dist-info → otf_api-0.11.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
-otf_api/__init__.py,sha256=9EKuuZV9IGzs0IR8i13YCi-tO5qYQ9rFVKqvxyleKuo,209
-otf_api/api.py,sha256=XbPj4MR0spEoN_1rfh7TjiTI9j5HjSVgqX3bhQGtxns,67643
+otf_api/__init__.py,sha256=VlLjNep7gdvPfnIDl1YbY8uY5eRM1vE4XJvNhI4mIB8,205
+otf_api/api.py,sha256=95qIszuc2zjneEk-VhS9e-F5f2EnUs4lHGehL7rU2G4,67511
 otf_api/exceptions.py,sha256=GISekwF5dPt0Ol0WCU55kE5ODc5VxicNEEhmlguuE0U,1815
 otf_api/filters.py,sha256=fk2bFGi3srjS96qZlaDx-ARZRaj93NUTUdMJ01TX420,3702
 otf_api/logging.py,sha256=PRZpCaJ1F1Xya3L9Efkt3mKS5_QNr3sXjEUERSxYjvE,563
 otf_api/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 otf_api/utils.py,sha256=TATPgb5-t3nEAqo_iGY2p2PKzJA564alN1skb23OCMQ,4817
 otf_api/auth/__init__.py,sha256=PuhtiZ02GVP8zVSn1O-fhaYm7JM_tq5yUFATk_-8upk,132
-otf_api/auth/auth.py,sha256=69iZIGFn9l470PrUfLvwdMTIL0tHYJ3tvuKOxy7NSyU,13185
+otf_api/auth/auth.py,sha256=5cPELC9Soif5knDDHm55ii1OMEPkJlGUphAdbOEmaRo,13278
 otf_api/auth/user.py,sha256=XlK3nbqJA4fF5UFmw2tt0eAle4QOQd6trnW72QrBsx4,2681
 otf_api/auth/utils.py,sha256=vc2pEyU-3yKdv0mR6r_zSHZyMw92j5UeIN_kzcb6TeE,2909
 otf_api/models/__init__.py,sha256=MSinaMQaBTGscL-YRKJ3axFiItQ1HoH62wC2xdaBMgk,1876
@@ -31,8 +31,8 @@ otf_api/models/studio_detail.py,sha256=2gq0A27NOZGz_PTBvsB-dkzm01nYc9FHmx1NON6xp
 otf_api/models/studio_services.py,sha256=aGLQMQmjGVpI6YxzAl-mcp3Y9cHPXuH9dIqrl6E-78E,1665
 otf_api/models/telemetry.py,sha256=PQ_CbADW5-t-U2iEQJGugNy-c4rD0q76TfyIqeFnTho,3170
 otf_api/models/workout.py,sha256=P3xVTvcYrm_RdU6qi3Xm2BXTxxvhvF0dgoEcODY41AA,3678
-otf_api-0.11.0rc1.dist-info/licenses/LICENSE,sha256=UaPT9ynYigC3nX8n22_rC37n-qmTRKLFaHrtUwF9ktE,1071
-otf_api-0.11.0rc1.dist-info/METADATA,sha256=au0J0vMB2Edln-n_Swag6fh2Zneyw1nnle6jdTcJu9k,2148
-otf_api-0.11.0rc1.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-otf_api-0.11.0rc1.dist-info/top_level.txt,sha256=KAhYg1X2YG0LkTuVRhUV1I_AReNZUVNdEan7cp0pEE4,8
-otf_api-0.11.0rc1.dist-info/RECORD,,
+otf_api-0.11.1.dist-info/licenses/LICENSE,sha256=UaPT9ynYigC3nX8n22_rC37n-qmTRKLFaHrtUwF9ktE,1071
+otf_api-0.11.1.dist-info/METADATA,sha256=BEppb5v_YlMUirZck7aKOtVkTemE7Uvantq2Gk8_00Y,2145
+otf_api-0.11.1.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+otf_api-0.11.1.dist-info/top_level.txt,sha256=KAhYg1X2YG0LkTuVRhUV1I_AReNZUVNdEan7cp0pEE4,8
+otf_api-0.11.1.dist-info/RECORD,,