41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from celery import shared_task
|
|
|
|
from feeds.models import Feed
|
|
from feeds.services import fetch_and_archive_feed
|
|
|
|
if TYPE_CHECKING:
|
|
from celery import Task
|
|
|
|
|
|
@shared_task(
|
|
bind=True,
|
|
autoretry_for=(Exception,),
|
|
retry_kwargs={"max_retries": 3, "countdown": 60},
|
|
)
|
|
def archive_feed_task(self: Task, feed_id: int) -> str:
|
|
"""Celery task to fetch and archive a feed by its ID.
|
|
|
|
Args:
|
|
self: The task instance.
|
|
feed_id: The ID of the Feed to archive.
|
|
|
|
Returns:
|
|
A message indicating the result of the archiving process.
|
|
"""
|
|
try:
|
|
feed: Feed = Feed.objects.get(id=feed_id)
|
|
except Feed.DoesNotExist:
|
|
return f"Feed with id {feed_id} does not exist."
|
|
|
|
try:
|
|
new_entries_count: int = fetch_and_archive_feed(feed)
|
|
|
|
# TODO(TheLovinator): Replace with a specific exception type # noqa: TD003
|
|
except ValueError as e:
|
|
raise self.retry(exc=e) from e
|
|
else:
|
|
if new_entries_count > 0:
|
|
return f"Archived {new_entries_count} new entries for {feed.url}"
|
|
return f"No new entries archived for {feed.url}"
|