fix api calls

markusressel · Sep 28, 2024 · 1688230 · 1688230
1 parent 1e3081d
commit 1688230
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/keel_telegram_bot/client/api_client.py b/keel_telegram_bot/client/api_client.py
@@ -61,7 +61,8 @@ def get_tracked(self) -> List[TrackedImage]:
         result = [TrackedImage.from_dict(tracked) for tracked in response]
         return result
 
-    def set_tracked(self, identifier: str, provider: Provider, trigger: Trigger, schedule: str or None) -> None:
+    def set_tracked(self, identifier: str, provider: Provider, trigger: Trigger,
+                    schedule: PollSchedule or None) -> None:
         """
         Set the tracking properties for an image
         :param identifier: the identifier of the image
@@ -71,9 +72,9 @@ def set_tracked(self, identifier: str, provider: Provider, trigger: Trigger, sch
         """
         self._do_request(HttpMethod.PUT, self._base_url + "/v1/tracked", json={
             "identifier": identifier,
-            "provider": provider,
-            "trigger": trigger,
-            "schedule": schedule,
+            "provider": provider.value,
+            "trigger": trigger.value,
+            "schedule": schedule.value,
         })
 
     def set_required_approvals_count(
@@ -87,7 +88,7 @@ def set_required_approvals_count(
         """
         self._do_request(HttpMethod.PUT, self._base_url + "/v1/approvals", json={
             "identifier": identifier,
-            "provider": provider,
+            "provider": provider.value,
             "votesRequired": votes_required,
         })
 
@@ -100,8 +101,8 @@ def set_policy(self, identifier: str, provider: Provider, policy: Policy) -> Non
         """
         self._do_request(HttpMethod.PUT, self._base_url + "/v1/policies", json={
             "identifier": identifier,
-            "provider": provider,
-            "policy": policy,
+            "provider": provider.value,
+            "policy": policy.value,
         })
 
     def set_schedule(self, identifier: str, provider: Provider, schedule: PollSchedule) -> None:
@@ -113,8 +114,8 @@ def set_schedule(self, identifier: str, provider: Provider, schedule: PollSchedu
         """
         self._do_request(HttpMethod.PUT, self._base_url + "/v1/schedule", json={
             "identifier": identifier,
-            "provider": provider,
-            "schedule": schedule,
+            "provider": provider.value,
+            "schedule": schedule.value,
         })
 
     def set_trigger(self, identifier: str, provider: Provider, trigger: Trigger) -> None:
@@ -181,7 +182,7 @@ def _run_approval_action(self, id: str, identifier: str, voter: str, action: Act
             "id": id,
             "identifier": identifier,
             "voter": voter,
-            "action": action,
+            "action": action.value,
         })
 
     def get_stats(self) -> DailyStats:

diff --git a/keel_telegram_bot/client/types.py b/keel_telegram_bot/client/types.py
@@ -27,6 +27,9 @@ def from_value(value: str):
         duration = parse_golang_duration(value)
         return PollSchedule(duration)
 
+    def value(self):
+        return self.__str__()
+
     def __str__(self):
         return f"@every {timedelta_to_golang_duration(self.interval)}"