1 files changed, 49 insertions, 0 deletions
diff --git a/modules/books.py b/modules/books.py
new file mode 100644
index 00000000..c78e381f
--- /dev/null
+++ b/modules/books.py
@@ -0,0 +1,49 @@
+import json, re
+import Utils
+
+URL_GOOGLEBOOKS = "https://www.googleapis.com/books/v1/volumes"
+URL_BOOKINFO = "https://books.google.co.uk/books?id=%s"
+REGEX_BOOKID = re.compile("id=([\w\-]+)")
+
+class Module(object):
+    _name = "ISBN"
+    def __init__(self, bot):
+        self.bot = bot
+        bot.events.on("received").on("command").on("isbn").hook(
+            self.isbn, help="Get book information from a provided ISBN",
+            min_args=1)
+        bot.events.on("received").on("command").on("book").hook(
+            self.book, help="Get book information from a provided title",
+            min_args=1)
+
+    def get_book(self, query, event):
+        page = Utils.get_url(URL_GOOGLEBOOKS, get_params={
+            "q": query, "country": "us"}, json=True)
+        if page:
+            if page["totalItems"] > 0:
+                book = page["items"][0]["volumeInfo"]
+                print(book)
+                title = book["title"]
+                sub_title = (", %s" % book.get("subtitle")
+                    ) if book.get("subtitle") else ""
+                authors = ", ".join(book["authors"])
+                date = book["publishedDate"]
+                rating = book["averageRating"]
+                id = re.search(REGEX_BOOKID, book["infoLink"]).group(1)
+                info_link = URL_BOOKINFO % id
+                event["stdout"].write("%s - %s (%s)%s %s (%s/5.0)" % (
+                    title, authors, date, sub_title, info_link, rating))
+            else:
+                event["stderr"].write("Unable to find book")
+        else:
+            event["stderr"].write("Failed to load results")
+
+    def isbn(self, event):
+        isbn = event["args_split"][0]
+        if len(isbn) == 10:
+            isbn = "978%s" % isbn
+        isbn = isbn.replace("-", "")
+        self.get_book("isbn:%s" % isbn, event)
+
+    def book(self, event):
+        self.get_book(event["args"], event)