telegram-video-downloader/src/main.py

import logging
import os

from telegram import InlineKeyboardMarkup
from telegram.ext import Updater, CallbackQueryHandler, MessageHandler, Filters

from vid_utils import Video, BadLink

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
logger = logging.getLogger(__name__)


def get_format(update, context):
    logger.info("from {}: {}".format(update.message.chat_id, update.message.text))  # "history"

    try:
        video = Video(link=update.message.text, init_keyboard=True)
    except BadLink as e:
        update.message.reply_text("Bad link: {}".format(e))
    except Exception as e:
        update.message.reply_text("Error: {}".format(e))
    else:
        reply_markup = InlineKeyboardMarkup(video.keyboard)
        update.message.reply_text('Choose format:', reply_markup=reply_markup)


def download_choosen_format(update, context):
    query = update.callback_query
    video_index = query.data

    context.bot.edit_message_text(text="Downloading...",
                                  chat_id=query.message.chat_id,
                                  message_id=query.message.message_id)

    video = Video(vid=video_index)
    video.download()

    with video.send() as files:
        for f in files:
            logger.info("Sending... {} ".format(f))
            context.bot.send_document(chat_id=query.message.chat_id, document=open(f, 'rb'))

    context.bot.delete_message(chat_id=query.message.chat_id,
                               message_id=query.message.message_id)

TOKEN = None
with open(os.path.join(os.environ['CONF_FOLDER'], 'bot.token')) as f:
    TOKEN = f.read().strip()

updater = Updater(token=TOKEN)

updater.dispatcher.add_handler(MessageHandler(Filters.text, get_format))
updater.dispatcher.add_handler(CallbackQueryHandler(download_choosen_format))


updater.start_polling()
updater.idle()