Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
markusressel committed Sep 28, 2024
1 parent 430192b commit dc0c9bb
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 10 deletions.
3 changes: 1 addition & 2 deletions keel_telegram_bot/bot/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,7 @@ def filter_tracked_images_by(resources: List[TrackedImage], glob: str or None) -

return result

items = self._api_client.get_tracked()
items = self._api_client.get_tracked_images()
filtered_items = filter_tracked_images_by(items, glob)

formatted_message = "\n\n".join(
Expand Down Expand Up @@ -364,7 +364,6 @@ async def execute(update: Update, context: ContextTypes.DEFAULT_TYPE, item: Reso
if schedule is not None:
self._api_client.set_schedule(
identifier=item.identifier,
provider=Provider.Kubernetes,
schedule=schedule,
)

Expand Down
23 changes: 15 additions & 8 deletions keel_telegram_bot/client/api_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,14 +53,25 @@ def get_resource(self, identifier: str) -> Resource or None:
return resource
return None

def get_tracked(self) -> List[TrackedImage]:
def get_tracked_images(self) -> List[TrackedImage]:
"""
Returns a list of all tracked images
"""
response = self._do_request(HttpMethod.GET, self._base_url + "/v1/tracked")
result = [TrackedImage.from_dict(tracked) for tracked in response]
return result

def get_tracked_image(self, identifier: str) -> TrackedImage or None:
"""
Returns a list of all tracked images
"""
trecked_images = self.get_tracked_images()
for tracked_image in trecked_images:
if tracked_image.image == identifier:
return tracked_image
return None


def set_tracked(self, identifier: str, provider: Provider, trigger: Trigger,
schedule: PollSchedule or None) -> None:
"""
Expand Down Expand Up @@ -105,18 +116,14 @@ def set_policy(self, identifier: str, provider: Provider, policy: Policy) -> Non
"policy": policy.value,
})

def set_schedule(self, identifier: str, provider: Provider, schedule: PollSchedule) -> None:
def set_schedule(self, identifier: str, schedule: PollSchedule) -> None:
"""
Set the polling schedule for an image
:param identifier: the identifier of the image
:param provider: the provider of the image
:param schedule: the schedule of the image
"""
self._do_request(HttpMethod.PUT, self._base_url + "/v1/schedule", json={
"identifier": identifier,
"provider": provider.value,
"schedule": schedule.value,
})
tracked_image = self.get_tracked_image(identifier)
self.set_tracked(identifier, tracked_image.provider, tracked_image.trigger, schedule)

def set_trigger(self, identifier: str, provider: Provider, trigger: Trigger) -> None:
"""
Expand Down

0 comments on commit dc0c9bb

Please sign in to comment.