Merge pull request #89 from NodeJSmith/fix/add_date_annotations

Fix/add date annotations

Author: NodeJSmith

.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "0.12.0"
+current_version = "0.12.1"
 
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)(?:-(?P<rc_l>rc)(?P<rc>0|[1-9]\\d*))?"
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "otf-api"
-version = "0.12.0"
+version = "0.12.1"
 description = "Python OrangeTheory Fitness API Client"
 authors = [{ name = "Jessica Smith", email = "[email protected]" }]
 requires-python = ">=3.11"

src/otf_api/__init__.py

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

src/otf_api/api.py

@@ -444,8 +444,8 @@ def _get_app_config_raw(self) -> dict[str, Any]:
 
     def get_bookings_new(
         self,
-        start_date: datetime | str | None = None,
-        end_date: datetime | str | None = None,
+        start_date: datetime | date | str | None = None,
+        end_date: datetime | date | str | None = None,
         exclude_cancelled: bool = True,
     ) -> list[models.BookingV2]:
         """Get the bookings for the user. If no dates are provided, it will return all bookings

src/otf_api/utils.py

@@ -60,7 +60,7 @@ def ensure_list(obj: list | Any | None) -> list:
     return obj
 
 
-def ensure_datetime(date_str: str | datetime | None, combine_with: time = MIN_TIME) -> datetime | None:
+def ensure_datetime(date_str: str | datetime | date | None, combine_with: time = MIN_TIME) -> datetime | None:
     if not date_str:
         return None