116 lines
4.8 KiB
Python
116 lines
4.8 KiB
Python
"""Looking for books"""
|
|
|
|
# PYTHON STUFFS #######################################################
|
|
|
|
import urllib
|
|
|
|
from nemubot import context
|
|
from nemubot.exception import IMException
|
|
from nemubot.hooks import hook
|
|
from nemubot.tools import web
|
|
|
|
from nemubot.module.more import Response
|
|
|
|
|
|
# LOADING #############################################################
|
|
|
|
def load(context):
|
|
if not context.config or "goodreadskey" not in context.config:
|
|
raise ImportError("You need a Goodreads API key in order to use this "
|
|
"module. Add it to the module configuration file:\n"
|
|
"<module name=\"books\" goodreadskey=\"XXXXXX\" />\n"
|
|
"Get one at https://www.goodreads.com/api/keys")
|
|
|
|
|
|
# MODULE CORE #########################################################
|
|
|
|
def get_book(title):
|
|
"""Retrieve a book from its title"""
|
|
response = web.getXML("https://www.goodreads.com/book/title.xml?key=%s&title=%s" %
|
|
(context.config["goodreadskey"], urllib.parse.quote(title)))
|
|
if response is not None and len(response.getElementsByTagName("book")):
|
|
return response.getElementsByTagName("book")[0]
|
|
else:
|
|
return None
|
|
|
|
|
|
def search_books(title):
|
|
"""Get a list of book matching given title"""
|
|
response = web.getXML("https://www.goodreads.com/search.xml?key=%s&q=%s" %
|
|
(context.config["goodreadskey"], urllib.parse.quote(title)))
|
|
if response is not None and len(response.getElementsByTagName("search")):
|
|
return response.getElementsByTagName("search")[0].getElementsByTagName("results")[0].getElementsByTagName("work")
|
|
else:
|
|
return []
|
|
|
|
|
|
def search_author(name):
|
|
"""Looking for an author"""
|
|
response = web.getXML("https://www.goodreads.com/api/author_url/%s?key=%s" %
|
|
(urllib.parse.quote(name), context.config["goodreadskey"]))
|
|
if response is not None and len(response.getElementsByTagName("author")) and response.getElementsByTagName("author")[0].hasAttribute("id"):
|
|
response = web.getXML("https://www.goodreads.com/author/show/%s.xml?key=%s" %
|
|
(urllib.parse.quote(response.getElementsByTagName("author")[0].getAttribute("id")), context.config["goodreadskey"]))
|
|
if response is not None and len(response.getElementsByTagName("author")):
|
|
return response.getElementsByTagName("author")[0]
|
|
return None
|
|
|
|
|
|
# MODULE INTERFACE ####################################################
|
|
|
|
@hook.command("book",
|
|
help="Get information about a book from its title",
|
|
help_usage={
|
|
"TITLE": "Get information about a book titled TITLE"
|
|
})
|
|
def cmd_book(msg):
|
|
if not len(msg.args):
|
|
raise IMException("please give me a title to search")
|
|
|
|
book = get_book(" ".join(msg.args))
|
|
if book is None:
|
|
raise IMException("unable to find book named like this")
|
|
res = Response(channel=msg.channel)
|
|
res.append_message("%s, written by %s: %s" % (book.getElementsByTagName("title")[0].firstChild.nodeValue,
|
|
book.getElementsByTagName("author")[0].getElementsByTagName("name")[0].firstChild.nodeValue,
|
|
web.striphtml(book.getElementsByTagName("description")[0].firstChild.nodeValue if book.getElementsByTagName("description")[0].firstChild else "")))
|
|
return res
|
|
|
|
|
|
@hook.command("search_books",
|
|
help="Search book's title",
|
|
help_usage={
|
|
"APPROX_TITLE": "Search for a book approximately titled APPROX_TITLE"
|
|
})
|
|
def cmd_books(msg):
|
|
if not len(msg.args):
|
|
raise IMException("please give me a title to search")
|
|
|
|
title = " ".join(msg.args)
|
|
res = Response(channel=msg.channel,
|
|
title="%s" % (title),
|
|
count=" (%d more books)")
|
|
|
|
for book in search_books(title):
|
|
res.append_message("%s, writed by %s" % (book.getElementsByTagName("best_book")[0].getElementsByTagName("title")[0].firstChild.nodeValue,
|
|
book.getElementsByTagName("best_book")[0].getElementsByTagName("author")[0].getElementsByTagName("name")[0].firstChild.nodeValue))
|
|
return res
|
|
|
|
|
|
@hook.command("author_books",
|
|
help="Looking for books writen by a given author",
|
|
help_usage={
|
|
"AUTHOR": "Looking for books writen by AUTHOR"
|
|
})
|
|
def cmd_author(msg):
|
|
if not len(msg.args):
|
|
raise IMException("please give me an author to search")
|
|
|
|
name = " ".join(msg.args)
|
|
ath = search_author(name)
|
|
if ath is None:
|
|
raise IMException("%s does not appear to be a published author." % name)
|
|
return Response([b.getElementsByTagName("title")[0].firstChild.nodeValue for b in ath.getElementsByTagName("book")],
|
|
channel=msg.channel,
|
|
title=ath.getElementsByTagName("name")[0].firstChild.nodeValue)
|