|
@@ -0,0 +1,55 @@
|
|
|
+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))
|
|
|
+ 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.log("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()
|