From f921728e87fa5b4f1d6cc2cb6c6b02b3d4276125 Mon Sep 17 00:00:00 2001 From: campi Date: Tue, 15 Mar 2022 17:06:22 +0100 Subject: [PATCH] "bot main" --- libreremo_bot.py | 126 +++++++++++++++++++++++++---------------------- 1 file changed, 67 insertions(+), 59 deletions(-) diff --git a/libreremo_bot.py b/libreremo_bot.py index 13b53e4..95ae5c7 100644 --- a/libreremo_bot.py +++ b/libreremo_bot.py @@ -18,10 +18,10 @@ 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) +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") @@ -29,13 +29,14 @@ def start(update: Update, context: CallbackContext): update.message.reply_text(txt.INTRO, parse_mode=ParseMode.HTML) -def help(update: Update, context: CallbackContext): +def show_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")]] + [InlineKeyboardButton(text="Autore", callback_data="autore")], + [InlineKeyboardButton(text="\u274C Annulla \u274C", callback_data="stop")]] reply_markup = InlineKeyboardMarkup(keyboard) update.message.reply_text( text="In base a cosa vuoi cercare il tuo libro?", @@ -45,10 +46,15 @@ def _ask_resource_type(update, context): def _get_resource_type(update, context): query = update.callback_query + query.answer() + if query.data == "stop": + query.edit_message_text("Ricerca annullata", None) + return ConversationHandler.END 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"])) + query.edit_message_text( + text="quale {} stai cercando?".format(query.data), + parse_mode=ParseMode.HTML + ) return GET_RESOURCE @@ -70,11 +76,6 @@ def _get_resource(update, context): 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)) @@ -93,36 +94,43 @@ def _get_resource(update, context): 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, + 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( 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") + query.edit_message_text( + text="Questi sono i libri che abbiamo per {}".format( + "".join(query.data.split(",")[::-1])), + reply_markup=__format_reply_markup(book_list), + parse_mode=ParseMode.HTML + ) return GET_BOOK def _get_book(update: Update, context: CallbackContext): query = update.callback_query + query.answer() + if query.data == "stop": + query.edit_message_text("Ricerca annulata", None) + return ConversationHandler.END chat_id = query.message.chat_id + query.edit_message_reply_markup() query.message.reply_text( - 'perfetto, ti invio immediatamente il file, abbi un attimo di pazienza se il libro non arriva subito') + "Hai scelto: {}".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" 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) @@ -156,56 +164,56 @@ def search_by_author(update: Update, context: CallbackContext): 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) + "{} non è un comando valido, prova con /help".format(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]) + 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) -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 +# 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...") -GET_RESOURCE_TYPE, ASK_RESOURCE, GET_RESOURCE, GET_BOOK, GET_AUTHOR, GET_CDS = range(6) +GET_RESOURCE_TYPE, ASK_RESOURCE, GET_RESOURCE, GET_BOOK, GET_AUTHOR = range(5) 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) +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() + -updater.start_polling() -updater.idle() +if __name__ == '__main__': + main()