telegram_bots/bots/tasks.py

import json
import logging
import os
import traceback
from time import sleep

from telegram import Bot, InlineKeyboardMarkup, InlineKeyboardButton
from telegram.error import RetryAfter

from config.celery import app
from djconfig import config


@app.task(bind=True)
def upload_image_rpc(self, queued_image_pk, fpath, caption):
    from .modules.channel_helper import QueuedItem
    config._reload()
    try:
        logging.warning(f'Processing upload task for queued item #{queued_image_pk}')
        qi = QueuedItem.objects.get(pk=queued_image_pk)
        bot = qi.config.bot.get_bot()
        if qi.config.queued:
            m = bot.send_photo(
                config.tmp_uploads_chat_id,
                open(fpath, 'rb'),
                caption=caption,
                reply_markup=InlineKeyboardMarkup([[
                    InlineKeyboardButton('Delete', callback_data=f'del {queued_image_pk}')
                ]]),
            )
            qi.message_id = m.message_id
            qi.args = json.dumps([m.photo[-1].file_id])
            qi.save()
        else:
            bot.send_photo(
                qi.config.chat_id,
                open(fpath, 'rb'),
                caption=caption,
            )
            qi.processed = True
            qi.save()
        os.unlink(fpath)
        sleep(.3)
    except RetryAfter as e:
        raise self.retry(exc=e, countdown=int(e.retry_after))
    except Exception as e:
        traceback.print_exc()
        raise self.retry(exc=e, countdown=5)


@app.task(bind=True)
def upload_animation_rpc(self, queued_image_pk, fpath, caption):
    from .modules.channel_helper import QueuedItem
    config._reload()
    try:
        logging.warning(f'Processing upload task for queued animation item #{queued_image_pk}')
        qi = QueuedItem.objects.get(pk=queued_image_pk)
        bot = qi.config.bot.get_bot()
        if qi.config.queued:
            m = bot.send_animation(
                config.tmp_uploads_chat_id,
                open(fpath, 'rb'),
                caption=caption,
                reply_markup=InlineKeyboardMarkup([[
                    InlineKeyboardButton('Delete', callback_data=f'del {queued_image_pk}')
                ]]),
            )
            qi.message_id = m.message_id
            qi.args = json.dumps([m.animation.file_id])
            qi.save()
        else:
            bot.send_animation(
                qi.config.chat_id,
                open(fpath, 'rb'),
                caption=caption,
            )
            qi.processed = True
            qi.save()
        os.unlink(fpath)
    except RetryAfter as e:
        raise self.retry(exc=e, countdown=int(e.retry_after))
    except Exception as e:
        traceback.print_exc()
        raise self.retry(exc=e, countdown=5)