diff options
| -rw-r--r-- | modules/wacalc.py | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/modules/wacalc.py b/modules/wacalc.py new file mode 100644 index 0000000..4d40ade --- /dev/null +++ b/modules/wacalc.py @@ -0,0 +1,72 @@ +info = { + 'author': 'Jon Bergli Heier', + 'title': 'WolframAlpha calculator', + 'description': 'WolframAlpha calculator', +} + +import urllib, urllib2, re +from xml.etree.ElementTree import ElementTree + +whitespace_re = re.compile(r'\s+') + +class Module(object): + config_section = 'module/wacalc' + + def __init__(self, bot): + self.irc = bot + if self.irc: + self.irc.register_keyword('!wacalc', self) + if config.has_option(self.config_section, 'register_gcalc'): + # Register gcalc trigger for the obsolete gcalc module + self.irc.register_keyword('!gcalc', self) + + def keyword(self, nick, channel, kw, msg): + target = channel if not channel == self.irc.nickname else nick.split('!')[0] + if not len(msg.strip()): + self.irc.msg(target, 'Usage: !wacalc query') + return + query = { + 'appid': config.get(self.config_section, 'api_key'), + 'format': 'plaintext', + 'input': msg, + } + if config.has_option(self.config_section, 'location'): + query['location'] = config.get(self.config_section, 'location') + u = urllib2.urlopen('http://api.wolframalpha.com/v2/query?%s' % urllib.urlencode(query)) + xmldoc = ElementTree() + xmldoc.parse(u) + xml = xmldoc.getroot() + if xml.attrib['success'] != 'true': + tips = xml.find('tips') + dym = xml.find('didyoumeans') + message = None + if tips is not None and len(tips): + message = 'Tip: %s' % tips[0].attrib['text'].encode('utf-8') + elif dym is not None and len(dym): + # TODO: sort by score + message = 'Did you mean "%s"?' % dym[0].text.encode('utf-8') + if message: + message = 'No result (%s)' % message + else: + message = 'No result.' + self.irc.msg(target, message) + return + + right = None + left = msg + def pod_plaintext(pod): + result = pod.find('subpod/plaintext') + if result is not None and result.text is not None: + text = result.text + text = whitespace_re.sub(' ', text) + return text.encode('utf-8') + for pod in xml.findall('pod'): + pod_id = pod.attrib['id'] + if pod_id == 'Result': + right = pod_plaintext(pod) + elif pod_id == 'Input': + left = pod_plaintext(pod) + if left and right: + self.irc.msg(target, '%s = %s' % (left, right)) + else: + self.irc.msg(target, 'Something failed.') |