"new commit"

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)

libreremo_assets.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+INTRO = """
+Ciao, questo è il nuovo bot telegram di <b>Libreremo</b>, un progetto politico che da anni il collettivo universitario fiorentino porta avanti <b>distribuendo ogni anno centinaia di testi universitari GRATUITAMENTE</b> 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...
+<a href="https://www.facebook.com/collettivodateneofirenze">Facebook</a>
+<a href="https://www.instagram.com/collettivo_ateneo_firenze">Instagram</a>
+<a href="mailto:colpol@inventati.org">Mail ColPol</a>
+<a href="   mailto:collettivostumunifi@autistici.org">Mail Krisis</a>
+
+<b>Per cominciare a cercare</b> 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 <b>Libreremo</b>, un progetto politico che da anni il collettivo universitario fiorentino porta avanti <b>distribuendo ogni anno centinaia di testi universitari GRATUITAMENTE</b> per contrastare la mercificazione della cultura e il carolibri.
+<b>Per cominciare a cercare</b> 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"

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()

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
-