use get_resource(identifier) instead of tracked images

markusressel · Sep 28, 2024 · c3f5d49 · c3f5d49
1 parent 83463e6
commit c3f5d49
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/keel_telegram_bot/client/api_client.py b/keel_telegram_bot/client/api_client.py
@@ -96,10 +96,10 @@ def set_required_approvals_count(
         :param identifier: the identifier of the image
         :param votes_required: the required approvals count
         """
-        tracked_image = self.get_tracked_image(identifier)
+        resource = self.get_resource(identifier)
         self._do_request(HttpMethod.PUT, self._base_url + "/v1/approvals", json={
             "identifier": identifier,
-            "provider": tracked_image.provider.value,
+            "provider": resource.provider.value,
             "votesRequired": votes_required,
         })
 
@@ -109,10 +109,10 @@ def set_policy(self, identifier: str, policy: Policy) -> None:
         :param identifier: the identifier of the image
         :param policy: the policy of the image
         """
-        tracked_image = self.get_tracked_image(identifier)
+        resource = self.get_resource(identifier)
         self._do_request(HttpMethod.PUT, self._base_url + "/v1/policies", json={
             "identifier": identifier,
-            "provider": tracked_image.provider.value,
+            "provider": resource.provider.value,
             "policy": policy.value,
         })
 
@@ -122,17 +122,17 @@ def set_schedule(self, identifier: str, schedule: PollSchedule) -> None:
         :param identifier: the identifier of the image
         :param schedule: the schedule of the image
         """
-        tracked_image = self.get_tracked_image(identifier)
-        self.set_tracked(identifier, tracked_image.provider, tracked_image.trigger, schedule)
+        resource = self.get_resource(identifier)
+        self.set_tracked(identifier, resource.provider, resource.trigger, schedule)
 
     def set_trigger(self, identifier: str, trigger: Trigger) -> None:
         """
         Set the trigger for an image
         :param identifier: the identifier of the image
         :param trigger: the trigger of the image
         """
-        tracked_image = self.get_tracked_image(identifier)
-        self.set_tracked(identifier, tracked_image.provider, trigger, tracked_image.schedule)
+        resource = self.get_resource(identifier)
+        self.set_tracked(identifier, resource.provider, trigger, resource.schedule)
 
     def get_approvals(self, rejected: bool = None, archived: bool = None) -> List[Approval]:
         """