telegram_bots/feeds/tasks.py

import traceback

import sentry_sdk
from celery.exceptions import Reject
from celery_once import QueueOnce
from django.utils import timezone
from telebot import TeleBot
from djconfig import config

from config.celery import app


@app.task(base=QueueOnce, once={'graceful': True})
def check_feeds():
    from feeds.models import Feed

    feeds = Feed.objects.filter(lock=False)
    enqueued = []
    for feed in feeds:
        if feed.run_check():
            enqueued.append(str(feed))
    return f'Following tasks were enqueued: {enqueued}'


@app.task()
def execute_feed(feed_pk):
    from feeds.models import Feed

    config._reload_maybe()

    feed = Feed.objects.get(pk=feed_pk)

    with sentry_sdk.configure_scope() as scope:
        scope.set_extra('feed', str(feed))
        try:
            bot = TeleBot(config.feed_bot_token, threaded=False)
            print(f'Last ID for "{feed}" = "{feed.last_id}"')
            for last_id in feed.config.execute(bot, feed.chat_id, feed.last_id):
                if last_id:
                    feed.last_id = last_id
                    feed.save()
                    print(f'Saved last ID for "{feed}" = "{feed.last_id}"')
            feed.last_check = timezone.now()
            feed.save()
        except Exception as e:
            sentry_sdk.capture_exception(e)
            traceback.print_exc()
        finally:
            if feed:
                feed.lock = False
                feed.save()