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.

220 lines
7.6 KiB
Python

3 years ago
#!/usr/bin/env python
# -*- coding: utf-8 -*-
3 years ago
import glob
import logging
3 years ago
from telegram import InlineKeyboardButton, InlineKeyboardMarkup
3 years ago
from telegram import ParseMode
3 years ago
from telegram.ext.callbackcontext import CallbackContext
3 years ago
from telegram.ext.callbackqueryhandler import CallbackQueryHandler
3 years ago
from telegram.ext.commandhandler import CommandHandler
from telegram.ext.conversationhandler import ConversationHandler
from telegram.ext.filters import Filters
3 years ago
from telegram.ext.messagehandler import MessageHandler
from telegram.ext.updater import Updater
from telegram.update import Update
3 years ago
import database_lib as DB
3 years ago
import libreremo_assets as txt
from libreremo_conf import TOKEN, CALIBRE_DIR, MIN_QUERY_LENGTH
3 years ago
3 years ago
logging.basicConfig(
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
)
3 years ago
3 years ago
db = DB.Database(CALIBRE_DIR + "metadata.db")
3 years ago
def start(update: Update, context: CallbackContext):
3 years ago
update.message.reply_text(txt.INTRO, parse_mode=ParseMode.HTML)
3 years ago
3 years ago
def show_help(update: Update, context: CallbackContext):
3 years ago
update.message.reply_text(txt.HELP, parse_mode=ParseMode.HTML)
3 years ago
def _ask_resource_type(update, context):
keyboard = [[InlineKeyboardButton(text="Titolo", callback_data="titolo")],
3 years ago
[InlineKeyboardButton(text="Autore", callback_data="autore")],
[InlineKeyboardButton(text="\u274C Annulla \u274C", callback_data="stop")]]
3 years ago
reply_markup = InlineKeyboardMarkup(keyboard)
update.message.reply_text(
3 years ago
text="In base a cosa vuoi cercare il tuo libro?",
3 years ago
reply_markup=reply_markup)
return GET_RESOURCE_TYPE
def _get_resource_type(update, context):
query = update.callback_query
3 years ago
query.answer()
if query.data == "stop":
query.edit_message_text("Ricerca annullata", None)
return ConversationHandler.END
3 years ago
context.user_data["resource"] = query.data
3 years ago
query.edit_message_text(
text="quale <b>{}</b> stai cercando?".format(query.data),
parse_mode=ParseMode.HTML
)
3 years ago
return GET_RESOURCE
def _get_resource(update, context):
resource = update.message.text
3 years ago
result_list = []
if len(resource) <= MIN_QUERY_LENGTH:
update.message.reply_text(txt.SHORTQUERY)
update.message.reply_text("Cosa stai cercando?")
3 years ago
return GET_RESOURCE
3 years ago
if context.user_data["resource"] == "titolo":
3 years ago
result_list = db.select_books_by_title(resource)
text = "Scegli uno di questi libri"
handler = GET_BOOK
3 years ago
elif context.user_data["resource"] == "autore":
3 years ago
result_list = db.select_authors(resource)
text = "Scegli uno di questi autori"
handler = GET_AUTHOR
logging.info("asked resource {res}: {txt}"
3 years ago
.format(res=context.user_data["resource"], txt=resource))
3 years ago
if not result_list:
logging.info("cannot found result for {}".format(resource))
3 years ago
update.message.reply_text(txt.FILENOTFOUND, parse_mode=ParseMode.HTML)
3 years ago
return ConversationHandler.END
reply_markup = __format_reply_markup(result_list)
update.message.reply_text(
3 years ago
text=text,
3 years ago
reply_markup=reply_markup)
3 years ago
return handler
3 years ago
def _get_author(update: Update, context: CallbackContext):
query = update.callback_query
3 years ago
query.answer()
if query.data == "stop":
query.edit_message_text("Ricerca annullata", None)
return ConversationHandler.END
book_list = db.select_books_by_author(query.data)
if not book_list:
logging.info(f"cannot find books for author: {query.data}")
query.edit_message_text(
3 years ago
text="Mi dispiace ma non riesco a trovare libri per questo autore")
3 years ago
return ConversationHandler.END
3 years ago
query.edit_message_text(
text="Questi sono i libri che abbiamo per <b>{}</b>".format(
"".join(query.data.split(",")[::-1])),
reply_markup=__format_reply_markup(book_list),
parse_mode=ParseMode.HTML
)
3 years ago
return GET_BOOK
3 years ago
def _get_book(update: Update, context: CallbackContext):
query = update.callback_query
3 years ago
query.answer()
if query.data == "stop":
query.edit_message_text("Ricerca annulata", None)
return ConversationHandler.END
3 years ago
chat_id = query.message.chat_id
3 years ago
query.edit_message_reply_markup()
3 years ago
query.message.reply_text(
3 years ago
"Hai scelto: <b>{}</b>".format("(".join(
query.data.split("/", 1)[1]
.split("(")[:-1])),
parse_mode=ParseMode.HTML
)
query.message.reply_text(
'Ti invio immediatamente il file, sii paziente se il tuo libro non arriva')
filepath = CALIBRE_DIR + query.data + "/*.pdf"
3 years ago
try:
for file in glob.glob(filepath):
3 years ago
with open(file, 'rb') as doc_pdf:
3 years ago
context.bot.send_document(chat_id, doc_pdf)
3 years ago
logging.info("Sharing {}".format(file))
3 years ago
except FileNotFoundError:
logging.info("cannot send {}".format(filepath))
query.message.reply_text(
3 years ago
"Mi dispiace non rieco a trovare il file, prova piu' tardi")
3 years ago
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
3 years ago
3 years ago
def search_by_title_default(update: Update, context: CallbackContext):
context.user_data["resource"] = "titolo"
3 years ago
return _get_resource(update, context)
3 years ago
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
3 years ago
3 years ago
def unknown(update: Update, context: CallbackContext):
update.message.reply_text(
3 years ago
"{} non è un comando valido, prova con /help".format(update.message.text))
3 years ago
def __format_reply_markup(list_of_res):
3 years ago
inline_keyboard = [[InlineKeyboardButton(
text=res.format_text(), callback_data=res.format_data())]
for res in list_of_res]
inline_keyboard.append([InlineKeyboardButton(
text="\u274C Annulla \u274C", callback_data="stop")])
return InlineKeyboardMarkup(inline_keyboard)
3 years ago
3 years ago
3 years ago
# 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...")
3 years ago
3 years ago
GET_RESOURCE_TYPE, ASK_RESOURCE, GET_RESOURCE, GET_BOOK, GET_AUTHOR = range(5)
3 years ago
cnv_handler = ConversationHandler(
3 years ago
entry_points=[CommandHandler('search', _ask_resource_type),
CommandHandler('search_by_title', search_by_title),
CommandHandler('search_by_author', search_by_author),
MessageHandler(Filters.text, search_by_title_default)],
states={
3 years ago
GET_RESOURCE_TYPE: [CallbackQueryHandler(_get_resource_type)],
GET_RESOURCE: [MessageHandler(Filters.text, _get_resource)],
GET_AUTHOR: [CallbackQueryHandler(_get_author)],
GET_BOOK: [CallbackQueryHandler(_get_book)],
},
fallbacks=[CommandHandler('stop', _cnv_stop),
MessageHandler(Filters.command, _cnv_stop)],
)
3 years ago
3 years ago
def main():
updater = Updater(TOKEN, use_context=True, arbitrary_callback_data=True)
updater.dispatcher.add_handler(CommandHandler('start', start))
updater.dispatcher.add_handler(CommandHandler('help', show_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()
3 years ago
3 years ago
if __name__ == '__main__':
main()