1 files changed, 42 insertions, 0 deletions
diff --git a/modules/wolframalpha.py b/modules/wolframalpha.py
new file mode 100644
index 00000000..534187dc
--- /dev/null
+++ b/modules/wolframalpha.py
@@ -0,0 +1,42 @@
+import re
+import Utils
+
+URL_WA = "http://api.wolframalpha.com/v2/query"
+REGEX_CHARHEX = re.compile("\\\\:(\S{4})")
+
+class Module(object):
+    _name = "Wolfram|Alpha"
+    def __init__(self, bot):
+        bot.events.on("received").on("command").on("wolframalpha", "wa"
+            ).hook(self.wa, min_args=1, help=
+            "Evauate a given string on Wolfram|Alpha")
+        self.bot = bot
+
+    def wa(self, event):
+        soup = Utils.get_url(URL_WA, get_params={"input": event["args"],
+            "appid": self.bot.config["wolframalpha-api-key"],
+            "format": "plaintext", "reinterpret": "true"}, soup=True)
+
+        if soup:
+            if int(soup.find("queryresult").get("numpods")) > 0:
+                input = soup.find(id="Input").find("subpod").find("plaintext"
+                    ).text
+                for pod in soup.find_all("pod"):
+                    if pod.get("primary") == "true":
+                        answer = pod.find("subpod").find("plaintext")
+                        text = "(%s) %s" % (input.replace(" | ", ": "),
+                            answer.text.strip().replace(" | ", ": "
+                            ).replace("\n", " | ").replace("\r", ""))
+                        while True:
+                            match = re.search(REGEX_CHARHEX, text)
+                            if match:
+                                text = re.sub(REGEX_CHARHEX, chr(int(
+                                    match.group(1), 16)), text)
+                            else:
+                                break
+                        event["stdout"].write(text)
+                        break
+            else:
+                event["stderr"].write("No results found")
+        else:
+            event["stderr"].write("Failed to load results")