|
|
@ -107,6 +107,60 @@ class MUCBot(slixmpp.ClientXMPP): |
|
|
|
f.write(message+'\n') |
|
|
|
f.close() |
|
|
|
|
|
|
|
if '/book' in msg['body']: # Check if this is a book |
|
|
|
|
|
|
|
self.send_message(mto=self.room, |
|
|
|
mbody="Oh a book, that's cool! Thanks {}!".format(msg['mucnick']), |
|
|
|
mtype='groupchat') |
|
|
|
|
|
|
|
|
|
|
|
from bs4 import BeautifulSoup |
|
|
|
import requests |
|
|
|
import re |
|
|
|
|
|
|
|
book = msg['body'].replace('@bot', '').replace('/book', '') |
|
|
|
book = re.sub(' +', ' ', book) # remove double spaces |
|
|
|
book = book.lstrip().rstrip() # remove spaces at the beginning and at the end |
|
|
|
book = book.replace(' ', '+').lower() # turn space into + and lowercase |
|
|
|
|
|
|
|
page_link = 'https://www.worldcat.org/search?q={}&qt=results_page'.format(book) |
|
|
|
|
|
|
|
page_response = requests.get(page_link, timeout=5) |
|
|
|
|
|
|
|
page_content = BeautifulSoup(page_response.content, "html.parser") |
|
|
|
|
|
|
|
try: |
|
|
|
book_title = page_content.findAll("div", {"class": "name"})[0].text |
|
|
|
book_author = page_content.findAll("div", {"class": "author"})[0].text |
|
|
|
book_publisher = page_content.findAll("div", {"class": "publisher"})[0].text |
|
|
|
|
|
|
|
response = '<b>BOOK</b>: ' + book_title + ' ' + book_author + ' ' + book_publisher |
|
|
|
|
|
|
|
book_found = True |
|
|
|
|
|
|
|
except IndexError: |
|
|
|
|
|
|
|
book_found = False |
|
|
|
|
|
|
|
if book_found: |
|
|
|
|
|
|
|
# Add message to log |
|
|
|
message = '<b>BOOK</b>: ' + book_title + ' ' + book_author + ' ' + book_publisher |
|
|
|
log = 'log.html' |
|
|
|
log_path = os.path.join(self.output, log) |
|
|
|
f = open(log_path, 'a+') |
|
|
|
f.write(message+'\n') |
|
|
|
f.close() |
|
|
|
|
|
|
|
self.send_message(mto=self.room, mbody='Hope this was the book you were looking for: ' + book_title + ' ' + book_author + ' ' + book_publisher, mtype='groupchat') |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
self.send_message(mto=self.room, mbody='Sorry, no book found!', mtype='groupchat') |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
|
# Setup the command line arguments. |
|
|
|
parser = ArgumentParser() |
|
|
@ -163,4 +217,3 @@ if __name__ == '__main__': |
|
|
|
# Connect to the XMPP server and start processing XMPP stanzas. |
|
|
|
xmpp.connect() |
|
|
|
xmpp.process() |
|
|
|
|
|
|
|