From e0d0dba566a0bac93a42576b56b3337885a8b24c Mon Sep 17 00:00:00 2001 From: campi Date: Tue, 15 Mar 2022 00:31:09 +0100 Subject: [PATCH] "new commit" --- database_lib.py | 17 ++--- libreremo_assets.py | 36 +++++++++++ libreremo_bot.py | 147 +++++++++++++++++++++----------------------- libreremo_res.py | 15 +++-- 4 files changed, 124 insertions(+), 91 deletions(-) create mode 100644 libreremo_assets.py diff --git a/database_lib.py b/database_lib.py index 5774de2..080a856 100644 --- a/database_lib.py +++ b/database_lib.py @@ -6,7 +6,8 @@ import sqlite3 as sql3 logger = logging.getLogger(__name__) -class Database(): + +class Database: def __init__(self, filename): self.filename = filename @@ -17,7 +18,7 @@ class Database(): result_list = None try: self.con = sql3.connect(self.filename) - self.con.row_factory = factory + self.con.row_factory = factory self.cur = self.con.cursor() logging.info("Connection to db open") self.cur.execute(sql) @@ -26,8 +27,8 @@ class Database(): logging.info("an error occured", error) finally: if self.con: - self.con.close() - logging.info("Connection to db closed") + self.con.close() + logging.info("Connection to db closed") return result_list def select_books_by_title(self, query): @@ -36,17 +37,17 @@ class Database(): SELECT title, author_sort, path FROM books WHERE title LIKE '{query}'; """ - return self._execute(sql, res.book_factory) - + return self._execute(sql, res.book_factory) + def select_books_by_author(self, query): query = self.__format_query(query) sql = f""" SELECT title, author_sort, path FROM books WHERE books.author_sort LIKE '%' || (SELECT sort FROM authors - WHERE sort = '{query}') || '%'; + WHERE sort LIKE '%{query}%') || '%'; """ - return self._execute(sql, res.book_factory) + return self._execute(sql, res.book_factory) def select_authors(self, query): query = self.__format_query(query) diff --git a/libreremo_assets.py b/libreremo_assets.py new file mode 100644 index 0000000..3d16282 --- /dev/null +++ b/libreremo_assets.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +INTRO = """ +Ciao, questo è il nuovo bot telegram di Libreremo, un progetto politico che da anni il collettivo universitario fiorentino porta avanti distribuendo ogni anno centinaia di testi universitari GRATUITAMENTE per contrastare la mercificazione della cultura e il carolibri. + +Il carolibri è da sempre, e soprattutto negli ultimi anni, un peso non indifferente per le tasche degli studenti e delle loro famiglie. Ma la nostra critica non è solo economica, pur costituendo già questa una difficoltà per tanti studenti costretti ad abbandonare gli studi. È anche politica, verso una concezione della cultura che vede quest’ultima come un bene acquistabile da chi può pagarselo, invece che un mezzo a disposizione di tutti e tutte per la propria crescita e quella della società nel suo complesso. +Una concezione questa che, purtroppo, permea l’intero mondo dell’istruzione e si fa particolarmente pesante in università. + +Qua i nostri riferimenti social, stay tuned... +Facebook +Instagram +Mail ColPol +Mail Krisis + +Per cominciare a cercare scrivi il titolo del libro oppure usa il menu comandi a fianco della chat o scrivi uno dei seguenti comandi: + +/search Per la ricerca generica +/search_by_title Per la ricerca per titolo +/search_by_author Per la ricerca per autore +/help Se hai bisogno di aiuto +""" + +HELP = """ +Ciao, questo è il nuovo bot telegram di Libreremo, un progetto politico che da anni il collettivo universitario fiorentino porta avanti distribuendo ogni anno centinaia di testi universitari GRATUITAMENTE per contrastare la mercificazione della cultura e il carolibri. +Per cominciare a cercare scrivi il titolo del libro oppure usa il menu comandi a fianco della chat o scrivi uno dei seguenti comandi: + +/help Visualizza questo aiuto +/search Per la ricerca generica +/search_by_title Per la ricerca per titolo +/search_by_author Per la ricerca per autore +""" + +FILENOTFOUND = "Mi dispiace non ho trovato niente, prova a fare una ricerca più generica o a cercare per autore" + +SHORTQUERY = "Scusa non ho capito cosa vuoi cercare, prova ad inserire quello che stai cercando per esteso" \ No newline at end of file diff --git a/libreremo_bot.py b/libreremo_bot.py index 2728877..13b53e4 100644 --- a/libreremo_bot.py +++ b/libreremo_bot.py @@ -1,54 +1,44 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- -from telegram.ext.updater import Updater -from telegram.update import Update +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.messagehandler import MessageHandler from telegram.ext.conversationhandler import ConversationHandler -from telegram.ext.callbackqueryhandler import CallbackQueryHandler -from telegram.callbackquery import CallbackQuery from telegram.ext.filters import Filters -from libreremo_conf import TOKEN, CALIBRE_DIR, MIN_QUERY_LENGTH -import glob +from telegram.ext.messagehandler import MessageHandler +from telegram.ext.updater import Updater +from telegram.update import Update + import database_lib as DB -import logging +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) - + 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(""" - Ciao, questo è il nuovo Bot di libreremo :), - al momento non funzionante e' attivo solo a scopo - di testing, ma restate connessi per aggiornamenti!! - """) -# qua puoi scaricare centinaia di testi -# universitari; /help se vuoi v.edere i -# comandi che puoi usare.""") + update.message.reply_text(txt.INTRO, parse_mode=ParseMode.HTML) def help(update: Update, context: CallbackContext): - update.message.reply_text("""Available Commands : - /search - /search_by_author - /search_by_class - /""") - - + 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="Corso di studio", callback_data="cds")]] + [InlineKeyboardButton(text="Autore", callback_data="autore")]] reply_markup = InlineKeyboardMarkup(keyboard) update.message.reply_text( - text="In base a cosa vuoi cercare il tuo libro?", + text="In base a cosa vuoi cercare il tuo libro?", reply_markup=reply_markup) return GET_RESOURCE_TYPE @@ -57,53 +47,49 @@ 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"])) + 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(""" - scusa non ho capito cosa vuoi cercare, prova - ad inserire il titolo del libro per esteso, - cosa stai cercando?""") + 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"): + + 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"): + + 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) + 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)) + .format(res=context.user_data["resource"], txt=resource)) if not result_list: logging.info("cannot found result for {}".format(resource)) - update.message.reply_text(""" - Mi dispiace non ho trovato niente, controlla....""") + 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, + text=text, reply_markup=reply_markup) - return handler - + return handler + def _get_author(update: Update, context: CallbackContext): query = update.callback_query @@ -111,41 +97,40 @@ def _get_author(update: Update, context: CallbackContext): 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") + 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 + 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) + # result_list = db.select_books_by_cds(query) context.bot.send_message( - chat_id = query.message.chat_id, - text = "scegli uno di questi libri", - reply_markup = reply_markup) - return GET_BOOK + 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! :)') + '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: + 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") + "Mi dispiace non rieco a trovare il file, prova piu' tardi") return ConversationHandler.END @@ -159,15 +144,18 @@ def search_by_title(update: Update, context: CallbackContext): 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 + 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" @@ -182,21 +170,26 @@ def unknown(update: Update, context: CallbackContext): 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]) + 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 = { + 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)], @@ -207,12 +200,12 @@ cnv_handler = ConversationHandler( MessageHandler(Filters.command, _cnv_stop)], ) - -updater.dispatcher.add_handler(cnv_handler) updater.dispatcher.add_handler(CommandHandler('start', start)) updater.dispatcher.add_handler(CommandHandler('help', help)) -updater.dispatcher.add_handler(MessageHandler(Filters.text, unknown)) +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() diff --git a/libreremo_res.py b/libreremo_res.py index 1038f3b..e9fa5a9 100644 --- a/libreremo_res.py +++ b/libreremo_res.py @@ -10,11 +10,12 @@ class ResourceInterface: callable(subclass.format_data) or NotImplemented) + class Author(ResourceInterface): def __init__(self, author, sort): self.author = author - self.sort = sort + self.sort = sort def format_text(self): return self.author @@ -22,12 +23,13 @@ class Author(ResourceInterface): def format_data(self): return self.sort + class Book(ResourceInterface): def __init__(self, title, author, file): self.title = title self.author = author - self.file = file + self.file = file def format_text(self): return self.title + " - " + self.author @@ -35,11 +37,12 @@ class Book(ResourceInterface): def format_data(self): return self.file + class CorsoDiStudio(ResourceInterface): def __init__(self, cds): self.cds = cds - + def format_text(self): return self.cds @@ -48,11 +51,11 @@ class CorsoDiStudio(ResourceInterface): def book_factory(cursor, row): - return Book(row[0], row[1], row[2]) + return Book(row[0], row[1], row[2]) + def author_factory(cursor, row): return Author(row[0], row[1]) -#def cds_factory(cursor, row): +# def cds_factory(cursor, row): # pass -