From d692a5271bd3512aecf810bb7058afd47bda7763 Mon Sep 17 00:00:00 2001
From: mutantmonkey <mutantmonkey@mutantmonkey.in>
Date: Tue, 5 Nov 2013 09:19:02 -0800
Subject: [PATCH] move calculator functionality to ddg

This may or may not be temporary, but it's better than the calculator
being totally broken.
---
 modules/calc.py           | 30 +++++++++++++++---------------
 modules/test/test_calc.py |  1 +
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/modules/calc.py b/modules/calc.py
index 128cc55..412a417 100644
--- a/modules/calc.py
+++ b/modules/calc.py
@@ -29,25 +29,25 @@ subs = [
 
 
 def c(phenny, input):
-    """Google calculator."""
+    """DuckDuckGo calculator."""
     if not input.group(2):
         return phenny.reply("Nothing to calculate.")
     q = input.group(2)
-    q = q.replace('\xcf\x95', 'phi')  # utf-8 U+03D5
-    q = q.replace('\xcf\x80', 'pi')  # utf-8 U+03C0
-    uri = 'http://www.google.com/ig/calculator?q='
-    bytes = web.get(uri + web.quote(q))
-    parts = bytes.split('",')
-    answer = [p for p in parts if p.startswith('rhs: "')][0][6:]
+
+    try:
+        r = web.get(
+            'https://api.duckduckgo.com/?q={}&format=json&no_html=1'
+            '&t=mutantmonkey/phenny'.format(web.quote(q)))
+    except web.HTTPError:
+        raise GrumbleError("Couldn't parse the result from DuckDuckGo.")
+
+    data = web.json(r)
+    if data['AnswerType'] == 'calc':
+        answer = data['Answer'].split('=')[-1].strip()
+    else:
+        answer = None
+
     if answer:
-        #answer = ''.join(chr(ord(c)) for c in answer)
-        #answer = answer.decode('utf-8')
-        answer = answer.replace('\\x26#215;', '*')
-        answer = answer.replace('\\x3c', '<')
-        answer = answer.replace('\\x3e', '>')
-        answer = answer.replace('<sup>', '^(')
-        answer = answer.replace('</sup>', ')')
-        answer = web.decode(answer)
         phenny.say(answer)
     else:
         phenny.reply('Sorry, no result.')
diff --git a/modules/test/test_calc.py b/modules/test/test_calc.py
index 081bccf..80eb61d 100644
--- a/modules/test/test_calc.py
+++ b/modules/test/test_calc.py
@@ -18,6 +18,7 @@ class TestCalc(unittest.TestCase):
 
         self.phenny.say.assert_called_once_with('25')
 
+    @unittest.skip('Not supported with DuckDuckGo')
     def test_c_scientific(self):
         input = Mock(group=lambda x: '2^64')
         c(self.phenny, input)