diff --git a/bot_sample.xml b/bot_sample.xml
index 0867bb6..ce821d2 100644
--- a/bot_sample.xml
+++ b/bot_sample.xml
@@ -11,9 +11,7 @@
-->
diff --git a/modules/ddg/__init__.py b/modules/ddg/__init__.py
index 63ef340..75ee56a 100644
--- a/modules/ddg/__init__.py
+++ b/modules/ddg/__init__.py
@@ -13,17 +13,6 @@ from more import Response
from . import DDGSearch
from . import UrbanDictionnary
-from . import WFASearch
-
-def load(context):
- WFASearch.CONF = context.config
-
-def reload():
- imp.reload(DDGSearch)
- imp.reload(UrbanDictionnary)
- imp.reload(WFASearch)
- imp.reload(Wikipedia)
-
@hook("cmd_hook", "define")
def define(msg):
@@ -76,22 +65,3 @@ def udsearch(msg):
res.append_message(d)
return res
-
-
-@hook("cmd_hook", "calculate")
-def calculate(msg):
- if len(msg.cmds) <= 1:
- return Response("Indicate a calcul to compute",
- msg.channel, nick=msg.nick)
-
- s = WFASearch.WFASearch(' '.join(msg.cmds[1:]))
-
- if s.success:
- res = Response(channel=msg.channel, nomore="No more results")
- for result in s.nextRes:
- res.append_message(result)
- if (len(res.messages) > 0):
- res.messages.pop(0)
- return res
- else:
- return Response(s.error, msg.channel)
diff --git a/modules/ddg/WFASearch.py b/modules/wolframalpha.py
similarity index 60%
rename from modules/ddg/WFASearch.py
rename to modules/wolframalpha.py
index f20bbe2..f3bc072 100644
--- a/modules/ddg/WFASearch.py
+++ b/modules/wolframalpha.py
@@ -2,22 +2,32 @@
from urllib.parse import quote
+from nemubot import context
+from nemubot.exception import IRCException
+from nemubot.hooks import hook
from nemubot.tools import web
+nemubotversion = 4.0
+
+from more import Response
+
+URL_API = "http://api.wolframalpha.com/v2/query?input=%%s&appid=%s"
+
+def load(context):
+ global URL_API
+ if not context.config or not context.config.hasAttribute("apikey"):
+ raise ImportError ("You need a Wolfram|Alpha API key in order to use "
+ "this module. Add it to the module configuration: "
+ "\n\n"
+ "Register at http://products.wolframalpha.com/api/")
+ URL_API = URL_API % quote(context.config["apikey"]).replace("%", "%%")
+
class WFASearch:
def __init__(self, terms):
self.terms = terms
- try:
- url = ("http://api.wolframalpha.com/v2/query?input=%s&appid=%s" %
- (quote(terms), CONF.getNode("wfaapi")["key"]))
- self.wfares = web.getXML(url)
- except (TypeError, KeyError):
- print ("You need a Wolfram|Alpha API key in order to use this "
- "module. Add it to the module configuration file:\n\nRegister at "
- "http://products.wolframalpha.com/api/")
- self.wfares = None
+ self.wfares = web.getXML(URL_API % quote(terms))
@property
def success(self):
@@ -69,3 +79,21 @@ class WFASearch:
subnode.getFirstNode("plaintext").getContent())
except IndexError:
pass
+
+
+@hook("cmd_hook", "calculate")
+def calculate(msg):
+ if not len(msg.args):
+ raise IRCException("Indicate a calcul to compute")
+
+ s = WFASearch(' '.join(msg.args))
+
+ if s.success:
+ res = Response(channel=msg.channel, nomore="No more results")
+ for result in s.nextRes:
+ res.append_message(result)
+ if (len(res.messages) > 0):
+ res.messages.pop(0)
+ return res
+ else:
+ return Response(s.error, msg.channel)