You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

212 lines
7.5 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import glob
import logging
from telegram import InlineKeyboardButton, InlineKeyboardMarkup
from telegram import ParseMode
from telegram.ext.callbackcontext import CallbackContext
from telegram.ext.callbackqueryhandler import CallbackQueryHandler
from telegram.ext.commandhandler import CommandHandler
from telegram.ext.conversationhandler import ConversationHandler
from telegram.ext.filters import Filters
from telegram.ext.messagehandler import MessageHandler
from telegram.ext.updater import Updater
from telegram.update import Update
import database_lib as DB
import libreremo_assets as txt
from libreremo_conf import TOKEN, CALIBRE_DIR, MIN_QUERY_LENGTH
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
updater = Updater(TOKEN, use_context=True, arbitrary_callback_data=True)
db = DB.Database(CALIBRE_DIR + "metadata.db")
def start(update: Update, context: CallbackContext):
update.message.reply_text(txt.INTRO, parse_mode=ParseMode.HTML)
def help(update: Update, context: CallbackContext):
update.message.reply_text(txt.HELP, parse_mode=ParseMode.HTML)
def _ask_resource_type(update, context):
keyboard = [[InlineKeyboardButton(text="Titolo", callback_data="titolo")],
[InlineKeyboardButton(text="Autore", callback_data="autore")]]
reply_markup = InlineKeyboardMarkup(keyboard)
update.message.reply_text(
text="In base a cosa vuoi cercare il tuo libro?",
reply_markup=reply_markup)
return GET_RESOURCE_TYPE
def _get_resource_type(update, context):
query = update.callback_query
context.user_data["resource"] = query.data
context.bot.send_message(
chat_id=query.message.chat_id,
text="quale {} stai cercando?".format(context.user_data["resource"]))
return GET_RESOURCE
def _get_resource(update, context):
resource = update.message.text
result_list = []
if len(resource) <= MIN_QUERY_LENGTH:
update.message.reply_text(txt.SHORTQUERY)
update.message.reply_text("Cosa stai cercando?")
return GET_RESOURCE
if context.user_data["resource"] == "titolo":
result_list = db.select_books_by_title(resource)
text = "Scegli uno di questi libri"
handler = GET_BOOK
elif context.user_data["resource"] == "autore":
result_list = db.select_authors(resource)
text = "Scegli uno di questi autori"
handler = GET_AUTHOR
elif context.user_data["resource"] == "cds":
# result_list = db.select_class(resource)
text = "Scegli uno di questi corsi di studio"
handler = GET_CDS
logging.info("asked resource {res}: {txt}"
.format(res=context.user_data["resource"], txt=resource))
if not result_list:
logging.info("cannot found result for {}".format(resource))
update.message.reply_text(txt.FILENOTFOUND, parse_mode=ParseMode.HTML)
return ConversationHandler.END
reply_markup = __format_reply_markup(result_list)
update.message.reply_text(
text=text,
reply_markup=reply_markup)
return handler
def _get_author(update: Update, context: CallbackContext):
query = update.callback_query
result_list = db.select_books_by_author(query.data)
if not result_list:
logging.info("cannot find books for requested author")
context.bot.send_message(
chat_id=query.message.chat_id,
text="Mi dispiace ma non riesco a trovare libri per questo autore")
return ConversationHandler.END
context.bot.send_message(
chat_id=query.message.chat_id,
text="scegli uno di questi libri",
reply_markup=__format_reply_markup(result_list))
return GET_BOOK
def _get_cds(update: Update, context: CallbackContext):
query = update.callback_query
# result_list = db.select_books_by_cds(query)
context.bot.send_message(
chat_id=query.message.chat_id,
text="scegli uno di questi libri")
return GET_BOOK
def _get_book(update: Update, context: CallbackContext):
query = update.callback_query
chat_id = query.message.chat_id
query.message.reply_text(
'perfetto, ti invio immediatamente il file, abbi un attimo di pazienza se il libro non arriva subito')
try:
filepath = CALIBRE_DIR + query.data + "/*.pdf"
for file in glob.glob(filepath):
with open(file, 'rb') as doc_pdf:
context.bot.send_document(chat_id, doc_pdf)
logging.info("Sharing {}".format(file))
except FileNotFoundError:
logging.info("cannot send {}".format(filepath))
query.message.reply_text(
"Mi dispiace non rieco a trovare il file, prova piu' tardi")
return ConversationHandler.END
def _cnv_stop(update: Update, context: CallbackContext):
update.message.reply_text("ciao")
return ConversationHandler.END
def search_by_title(update: Update, context: CallbackContext):
update.message.reply_text("che libro stai cercando?")
context.user_data["resource"] = "titolo"
return GET_RESOURCE
def search_by_title_default(update: Update, context: CallbackContext):
context.user_data["resource"] = "titolo"
return _get_resource(update, context)
def search_by_author(update: Update, context: CallbackContext):
update.message.reply_text("di quale autore stai cercando i libri?")
context.user_data["resource"] = "autore"
return GET_RESOURCE
def search_by_class(update: Update, context: CallbackContext):
update.message.reply_text("di quale corso di studio stai cercando i libri?")
context.user_data["resource"] = "cds"
return GET_RESOURCE
def unknown(update: Update, context: CallbackContext):
update.message.reply_text(
"Sorry '%s' is not a valid command" % update.message.text)
def __format_reply_markup(list_of_res):
return InlineKeyboardMarkup([
[InlineKeyboardButton(
text=res.format_text(),
callback_data=res.format_data())]
for res in list_of_res])
def handle_error(update: Update, context: CallbackContext):
context.bot.send_message(chat_id=update.callback_query.message.chat_id,
text="Oh no, qualcosa è andato storto, riprova più tardi...")
return ConversationHandler.END
GET_RESOURCE_TYPE, ASK_RESOURCE, GET_RESOURCE, GET_BOOK, GET_AUTHOR, GET_CDS = range(6)
cnv_handler = ConversationHandler(
entry_points=[CommandHandler('search', _ask_resource_type),
CommandHandler('search_by_title', search_by_title),
CommandHandler('search_by_author', search_by_author),
CommandHandler('search_by_class', search_by_class),
MessageHandler(Filters.text, search_by_title_default)],
states={
GET_RESOURCE_TYPE: [CallbackQueryHandler(_get_resource_type)],
GET_RESOURCE: [MessageHandler(Filters.text, _get_resource)],
GET_AUTHOR: [CallbackQueryHandler(_get_author)],
GET_BOOK: [CallbackQueryHandler(_get_book)],
GET_CDS: [CallbackQueryHandler(_get_cds)]
},
fallbacks=[CommandHandler('stop', _cnv_stop),
MessageHandler(Filters.command, _cnv_stop)],
)
updater.dispatcher.add_handler(CommandHandler('start', start))
updater.dispatcher.add_handler(CommandHandler('help', help))
updater.dispatcher.add_handler(cnv_handler)
updater.dispatcher.add_handler(MessageHandler(Filters.command, unknown)) # Filters out unknown commands
updater.dispatcher.add_error_handler(handle_error)
updater.start_polling()
updater.idle()