campi
/
libreremo_telegram_bot
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

"new commit"

Browse Source
deploy
campi 3 years ago
parent b1784d04d8
commit e0d0dba566
4 changed files with 124 additions and 91 deletions
Show Stats Download Patch File Download Diff File

17
database_lib.py
Unescape Escape View File

@ -6,7 +6,8 @@ import sqlite3 as sql3
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class Database():
class Database:
def __init__(self, filename): def __init__(self, filename):
self.filename = filename self.filename = filename
@ -17,7 +18,7 @@ class Database():
result_list = None result_list = None
try: try:
self.con = sql3.connect(self.filename) self.con = sql3.connect(self.filename)
self.con.row_factory = factory self.con.row_factory = factory
self.cur = self.con.cursor() self.cur = self.con.cursor()
logging.info("Connection to db open") logging.info("Connection to db open")
self.cur.execute(sql) self.cur.execute(sql)
@ -26,8 +27,8 @@ class Database():
logging.info("an error occured", error) logging.info("an error occured", error)
finally: finally:
if self.con: if self.con:
self.con.close() self.con.close()
logging.info("Connection to db closed") logging.info("Connection to db closed")
return result_list return result_list
def select_books_by_title(self, query): def select_books_by_title(self, query):
@ -36,17 +37,17 @@ class Database():
SELECT title, author_sort, path FROM books SELECT title, author_sort, path FROM books
WHERE title LIKE '{query}'; 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): def select_books_by_author(self, query):
query = self.__format_query(query) query = self.__format_query(query)
sql = f""" sql = f"""
SELECT title, author_sort, path FROM books SELECT title, author_sort, path FROM books
WHERE books.author_sort LIKE '%' || WHERE books.author_sort LIKE '%' ||
(SELECT sort FROM authors (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): def select_authors(self, query):
query = self.__format_query(query) query = self.__format_query(query)

36
libreremo_assets.py
Unescape Escape View File

@ -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"

147
libreremo_bot.py
Unescape Escape View File

@ -1,54 +1,44 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from telegram.ext.updater import Updater import glob
from telegram.update import Update import logging
from telegram import InlineKeyboardButton, InlineKeyboardMarkup from telegram import InlineKeyboardButton, InlineKeyboardMarkup
from telegram import ParseMode
from telegram.ext.callbackcontext import CallbackContext from telegram.ext.callbackcontext import CallbackContext
from telegram.ext.callbackqueryhandler import CallbackQueryHandler
from telegram.ext.commandhandler import CommandHandler from telegram.ext.commandhandler import CommandHandler
from telegram.ext.messagehandler import MessageHandler
from telegram.ext.conversationhandler import ConversationHandler from telegram.ext.conversationhandler import ConversationHandler
from telegram.ext.callbackqueryhandler import CallbackQueryHandler
from telegram.callbackquery import CallbackQuery
from telegram.ext.filters import Filters from telegram.ext.filters import Filters
from libreremo_conf import TOKEN, CALIBRE_DIR, MIN_QUERY_LENGTH from telegram.ext.messagehandler import MessageHandler
import glob from telegram.ext.updater import Updater
from telegram.update import Update
import database_lib as DB 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', 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) updater = Updater(TOKEN, use_context=True, arbitrary_callback_data=True)
db = DB.Database(CALIBRE_DIR + "metadata.db") db = DB.Database(CALIBRE_DIR + "metadata.db")
def start(update: Update, context: CallbackContext): def start(update: Update, context: CallbackContext):
update.message.reply_text(""" update.message.reply_text(txt.INTRO, parse_mode=ParseMode.HTML)
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.""")
def help(update: Update, context: CallbackContext): def help(update: Update, context: CallbackContext):
update.message.reply_text("""Available Commands : update.message.reply_text(txt.HELP, parse_mode=ParseMode.HTML)
/search
/search_by_author
/search_by_class
/""")
def _ask_resource_type(update, context): def _ask_resource_type(update, context):
keyboard = [[InlineKeyboardButton(text="Titolo", callback_data="titolo")], keyboard = [[InlineKeyboardButton(text="Titolo", callback_data="titolo")],
[InlineKeyboardButton(text="Autore", callback_data="autore")], [InlineKeyboardButton(text="Autore", callback_data="autore")]]
[InlineKeyboardButton(text="Corso di studio", callback_data="cds")]]
reply_markup = InlineKeyboardMarkup(keyboard) reply_markup = InlineKeyboardMarkup(keyboard)
update.message.reply_text( 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) reply_markup=reply_markup)
return GET_RESOURCE_TYPE return GET_RESOURCE_TYPE
@ -57,53 +47,49 @@ def _get_resource_type(update, context):
query = update.callback_query query = update.callback_query
context.user_data["resource"] = query.data context.user_data["resource"] = query.data
context.bot.send_message( context.bot.send_message(
chat_id = query.message.chat_id, chat_id=query.message.chat_id,
text = "quale {} stai cercando?" text="quale {} stai cercando?".format(context.user_data["resource"]))
.format(context.user_data["resource"]))
return GET_RESOURCE return GET_RESOURCE
def _get_resource(update, context): def _get_resource(update, context):
resource = update.message.text resource = update.message.text
result_list = []; result_list = []
if (len(resource) <= MIN_QUERY_LENGTH): if len(resource) <= MIN_QUERY_LENGTH:
update.message.reply_text(""" update.message.reply_text(txt.SHORTQUERY)
scusa non ho capito cosa vuoi cercare, prova update.message.reply_text("Cosa stai cercando?")
ad inserire il titolo del libro per esteso,
cosa stai cercando?""")
return GET_RESOURCE return GET_RESOURCE
if (context.user_data["resource"] == "titolo"): if context.user_data["resource"] == "titolo":
result_list = db.select_books_by_title(resource) result_list = db.select_books_by_title(resource)
text = "Scegli uno di questi libri" text = "Scegli uno di questi libri"
handler = GET_BOOK handler = GET_BOOK
elif (context.user_data["resource"] == "autore"): elif context.user_data["resource"] == "autore":
result_list = db.select_authors(resource) result_list = db.select_authors(resource)
text = "Scegli uno di questi autori" text = "Scegli uno di questi autori"
handler = GET_AUTHOR handler = GET_AUTHOR
elif (context.user_data["resource"] == "cds"): elif context.user_data["resource"] == "cds":
# result_list = db.select_class(resource) # result_list = db.select_class(resource)
text = "Scegli uno di questi corsi di studio" text = "Scegli uno di questi corsi di studio"
handler = GET_CDS handler = GET_CDS
logging.info("asked resource {res}: {txt}" 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: if not result_list:
logging.info("cannot found result for {}".format(resource)) logging.info("cannot found result for {}".format(resource))
update.message.reply_text(""" update.message.reply_text(txt.FILENOTFOUND, parse_mode=ParseMode.HTML)
Mi dispiace non ho trovato niente, controlla....""")
return ConversationHandler.END return ConversationHandler.END
reply_markup = __format_reply_markup(result_list) reply_markup = __format_reply_markup(result_list)
update.message.reply_text( update.message.reply_text(
text=text, text=text,
reply_markup=reply_markup) reply_markup=reply_markup)
return handler return handler
def _get_author(update: Update, context: CallbackContext): def _get_author(update: Update, context: CallbackContext):
query = update.callback_query query = update.callback_query
@ -111,41 +97,40 @@ def _get_author(update: Update, context: CallbackContext):
if not result_list: if not result_list:
logging.info("cannot find books for requested author") logging.info("cannot find books for requested author")
context.bot.send_message( context.bot.send_message(
chat_id = query.message.chat_id, chat_id=query.message.chat_id,
text = "Mi dispiace ma non riesco a trovare libri per questo autore") text="Mi dispiace ma non riesco a trovare libri per questo autore")
return ConversationHandler.END return ConversationHandler.END
context.bot.send_message( context.bot.send_message(
chat_id = query.message.chat_id, chat_id=query.message.chat_id,
text = "scegli uno di questi libri", text="scegli uno di questi libri",
reply_markup = __format_reply_markup(result_list)) reply_markup=__format_reply_markup(result_list))
return GET_BOOK return GET_BOOK
def _get_cds(update: Update, context: CallbackContext): def _get_cds(update: Update, context: CallbackContext):
query = update.callback_query query = update.callback_query
# result_list = db.select_books_by_cds(query) # result_list = db.select_books_by_cds(query)
context.bot.send_message( context.bot.send_message(
chat_id = query.message.chat_id, chat_id=query.message.chat_id,
text = "scegli uno di questi libri", text="scegli uno di questi libri")
reply_markup = reply_markup) return GET_BOOK
return GET_BOOK
def _get_book(update: Update, context: CallbackContext): def _get_book(update: Update, context: CallbackContext):
query = update.callback_query query = update.callback_query
chat_id = query.message.chat_id chat_id = query.message.chat_id
query.message.reply_text( 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: try:
filepath = CALIBRE_DIR + query.data + "/*.pdf" filepath = CALIBRE_DIR + query.data + "/*.pdf"
for file in glob.glob(filepath): 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) context.bot.send_document(chat_id, doc_pdf)
logging.info("Sharing {}".format(file)) logging.info("Sharing {}".format(file))
except FileNotFoundError: except FileNotFoundError:
logging.info("cannot send {}".format(filepath)) logging.info("cannot send {}".format(filepath))
query.message.reply_text( 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 return ConversationHandler.END
@ -159,15 +144,18 @@ def search_by_title(update: Update, context: CallbackContext):
context.user_data["resource"] = "titolo" context.user_data["resource"] = "titolo"
return GET_RESOURCE return GET_RESOURCE
def search_by_title_default(update: Update, context: CallbackContext): def search_by_title_default(update: Update, context: CallbackContext):
context.user_data["resource"] = "titolo" context.user_data["resource"] = "titolo"
return GET_RESOURCE return _get_resource(update, context)
def search_by_author(update: Update, context: CallbackContext): def search_by_author(update: Update, context: CallbackContext):
update.message.reply_text("di quale autore stai cercando i libri?") update.message.reply_text("di quale autore stai cercando i libri?")
context.user_data["resource"] = "autore" context.user_data["resource"] = "autore"
return GET_RESOURCE return GET_RESOURCE
def search_by_class(update: Update, context: CallbackContext): def search_by_class(update: Update, context: CallbackContext):
update.message.reply_text("di quale corso di studio stai cercando i libri?") update.message.reply_text("di quale corso di studio stai cercando i libri?")
context.user_data["resource"] = "cds" context.user_data["resource"] = "cds"
@ -182,21 +170,26 @@ def unknown(update: Update, context: CallbackContext):
def __format_reply_markup(list_of_res): def __format_reply_markup(list_of_res):
return InlineKeyboardMarkup([ return InlineKeyboardMarkup([
[InlineKeyboardButton( [InlineKeyboardButton(
text = res.format_text(), text=res.format_text(),
callback_data = res.format_data())] callback_data=res.format_data())]
for res in list_of_res]) 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) GET_RESOURCE_TYPE, ASK_RESOURCE, GET_RESOURCE, GET_BOOK, GET_AUTHOR, GET_CDS = range(6)
cnv_handler = ConversationHandler( cnv_handler = ConversationHandler(
entry_points = [CommandHandler('search', _ask_resource_type), entry_points=[CommandHandler('search', _ask_resource_type),
CommandHandler('search_by_title', search_by_title), CommandHandler('search_by_title', search_by_title),
CommandHandler('search_by_author', search_by_author), CommandHandler('search_by_author', search_by_author),
CommandHandler('search_by_class', search_by_class), CommandHandler('search_by_class', search_by_class),
MessageHandler(Filters.text, search_by_title_default)], MessageHandler(Filters.text, search_by_title_default)],
states = { states={
GET_RESOURCE_TYPE: [CallbackQueryHandler(_get_resource_type)], GET_RESOURCE_TYPE: [CallbackQueryHandler(_get_resource_type)],
GET_RESOURCE: [MessageHandler(Filters.text, _get_resource)], GET_RESOURCE: [MessageHandler(Filters.text, _get_resource)],
GET_AUTHOR: [CallbackQueryHandler(_get_author)], GET_AUTHOR: [CallbackQueryHandler(_get_author)],
@ -207,12 +200,12 @@ cnv_handler = ConversationHandler(
MessageHandler(Filters.command, _cnv_stop)], MessageHandler(Filters.command, _cnv_stop)],
) )
updater.dispatcher.add_handler(cnv_handler)
updater.dispatcher.add_handler(CommandHandler('start', start)) updater.dispatcher.add_handler(CommandHandler('start', start))
updater.dispatcher.add_handler(CommandHandler('help', help)) 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_handler(MessageHandler(Filters.command, unknown)) # Filters out unknown commands
updater.dispatcher.add_error_handler(handle_error)
updater.start_polling() updater.start_polling()
updater.idle() updater.idle()

15
libreremo_res.py
Unescape Escape View File

@ -10,11 +10,12 @@ class ResourceInterface:
callable(subclass.format_data) or callable(subclass.format_data) or
NotImplemented) NotImplemented)
class Author(ResourceInterface): class Author(ResourceInterface):
def __init__(self, author, sort): def __init__(self, author, sort):
self.author = author self.author = author
self.sort = sort self.sort = sort
def format_text(self): def format_text(self):
return self.author return self.author
@ -22,12 +23,13 @@ class Author(ResourceInterface):
def format_data(self): def format_data(self):
return self.sort return self.sort
class Book(ResourceInterface): class Book(ResourceInterface):
def __init__(self, title, author, file): def __init__(self, title, author, file):
self.title = title self.title = title
self.author = author self.author = author
self.file = file self.file = file
def format_text(self): def format_text(self):
return self.title + " - " + self.author return self.title + " - " + self.author
@ -35,11 +37,12 @@ class Book(ResourceInterface):
def format_data(self): def format_data(self):
return self.file return self.file
class CorsoDiStudio(ResourceInterface): class CorsoDiStudio(ResourceInterface):
def __init__(self, cds): def __init__(self, cds):
self.cds = cds self.cds = cds
def format_text(self): def format_text(self):
return self.cds return self.cds
@ -48,11 +51,11 @@ class CorsoDiStudio(ResourceInterface):
def book_factory(cursor, row): 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): def author_factory(cursor, row):
return Author(row[0], row[1]) return Author(row[0], row[1])
#def cds_factory(cursor, row): # def cds_factory(cursor, row):
# pass # pass

Write Preview
Loading…
Cancel
Save