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 = None
    try:
        feed = Feed.objects.get(pk=feed_pk)

        # if not feed.lock:
        #     feed.lock = True
        #     feed.save()
        # else:
        #     raise Reject('Lock', requeue=False)

        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()