feat: allow arbitrary kwargs in requests

tryterra · Jun 18, 2024 · 9218b19 · 9218b19
1 parent 3e6f91e
commit 9218b19
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/terra/base_client.py b/terra/base_client.py
@@ -107,6 +107,7 @@ def get_activity_for_user(
         end_date: typing.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: typing.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves workouts/activity data for a given User object. By default, data will be asynchronously sent to registered
@@ -130,6 +131,7 @@ def get_activity_for_user(
             end_date=end_date if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_body_for_user(
@@ -139,6 +141,7 @@ def get_body_for_user(
         end_date: typing.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: typing.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves body metrics data for a given User object. By default, data will be asynchronously sent to registered
@@ -162,6 +165,7 @@ def get_body_for_user(
             end_date=end_date if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_daily_for_user(
@@ -171,6 +175,7 @@ def get_daily_for_user(
         end_date: typing.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: typing.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves daily summary data for a given User object. By default, data will be asynchronously sent to registered
@@ -194,6 +199,7 @@ def get_daily_for_user(
             end_date=end_date if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_sleep_for_user(
@@ -203,6 +209,7 @@ def get_sleep_for_user(
         end_date: typing.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: typing.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves sleep data for a given User object. By default, data will be asynchronously sent to registered
@@ -226,6 +233,7 @@ def get_sleep_for_user(
             end_date=end_date if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_athlete_for_user(
@@ -255,6 +263,7 @@ def get_menstruation_for_user(
         end_date: typing.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: typing.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves daily summary data for a given User object. By default, data will be asynchronously sent to registered
@@ -279,6 +288,7 @@ def get_menstruation_for_user(
             end_date=end_date if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_nutrition_for_user(
@@ -288,6 +298,7 @@ def get_nutrition_for_user(
         end_date: typing.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: typing.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves daily summary data for a given User object. By default, data will be asynchronously sent to registered
@@ -312,6 +323,7 @@ def get_nutrition_for_user(
             end_date=end_date if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def generate_widget_session(

diff --git a/terra/models/user.py b/terra/models/user.py
@@ -86,6 +86,7 @@ def get_activity(
         end_date: t.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: t.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves workouts/activity data for a given User object. By default, data will be asynchronously sent to registered
@@ -113,6 +114,7 @@ def get_activity(
             end_date=int(end_date.timestamp()) if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_body(
@@ -121,6 +123,7 @@ def get_body(
         end_date: t.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: t.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves body metrics data for a given User object. By default, data will be asynchronously sent to registered
@@ -147,6 +150,7 @@ def get_body(
             end_date=int(end_date.timestamp()) if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_nutrition(
@@ -155,6 +159,7 @@ def get_nutrition(
         end_date: t.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: t.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves nutrition data for a given User object. By default, data will be asynchronously sent to registered
@@ -180,6 +185,7 @@ def get_nutrition(
             end_date=int(end_date.timestamp()) if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_daily(
@@ -188,6 +194,7 @@ def get_daily(
         end_date: t.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: t.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves daily summary data for a given User object. By default, data will be asynchronously sent to registered
@@ -213,6 +220,7 @@ def get_daily(
             end_date=int(end_date.timestamp()) if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_sleep(
@@ -221,6 +229,7 @@ def get_sleep(
         end_date: t.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: t.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves sleep data for a given User object. By default, data will be asynchronously sent to registered
@@ -247,6 +256,7 @@ def get_sleep(
             end_date=int(end_date.timestamp()) if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )
 
     def get_athlete(
@@ -274,6 +284,7 @@ def get_menstruation(
         end_date: t.Optional[datetime.datetime] = None,
         to_webhook: bool = True,
         with_samples: bool = True,
+        **kwargs: t.Any,
     ) -> api_responses.TerraApiResponse:
         """
         Retrieves daily summary data for a given User object. By default, data will be asynchronously sent to registered
@@ -299,4 +310,5 @@ def get_menstruation(
             end_date=int(end_date.timestamp()) if end_date is not None else None,
             to_webhook=to_webhook,
             with_samples=with_samples,
+            **kwargs,
         )