|
|
|
@ -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 <b>{}</b> 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 <b>{}</b>".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: <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"
|
|
|
|
|
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()
|
|
|
|
|