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)
    for feed in feeds:
        feed.run_check()


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

        bot = TeleBot(config.feed_bot_token)
        for last_id in feed.config.execute(bot, feed.chat_id, feed.last_id):
            if last_id:
                feed.last_id = last_id
                feed.save()
        bot.stop_bot()
        feed.last_check = timezone.now()
        feed.save()
    finally:
        if feed:
            feed.lock = False
            feed.save()