From 0ebecf3c90349a4319f498684170ffaefc1b3c12 Mon Sep 17 00:00:00 2001
From: Sfan5 <sfan5@live.de>
Date: Mon, 6 Jan 2014 16:01:54 +0100
Subject: [PATCH] [Calc] Make it work again

---
 COMMANDS.md |  2 +-
 calc.py     | 95 +++++++++++++++++++++++++++--------------------------
 2 files changed, 49 insertions(+), 48 deletions(-)

diff --git a/COMMANDS.md b/COMMANDS.md
index 80067be..d8fa8db 100644
--- a/COMMANDS.md
+++ b/COMMANDS.md
@@ -17,7 +17,7 @@ Required arguments are enclosed in { and }, optional arguments are enclosed in \
     <tr> <td>!ignore {user}</td>                     <td>Add user to ignore list</td>                         <td><i>Admins</i></td> </tr>
     <tr> <td>!unignore {user}</td>                   <td>Remove user from ignore list</td>                    <td><i>Admins</i></td> </tr>
     <tr> <td><b>calc.py</b></td>                     <td></td>                                                <td></td>              </tr>
-    <tr> <td>!c {expression}</td>                    <td>Calculate expression using Google Calculator</td>    <td>Anyone</td>        </tr>
+    <tr> <td>!c {expression}</td>                    <td>Calculate expression</td>                            <td>Anyone</td>        </tr>
     <tr> <td><b>chop.py</b></td>                     <td></td>                                                <td></td>              </tr>
     <tr> <td>!op [nick] ...</td>                     <td>Op nick</td>                                         <td><i>Admins</i></td> </tr>
     <tr> <td>!deop [nick] ...</td>                   <td>DeOp nick</td>                                       <td><i>Admins</i></td> </tr>
diff --git a/calc.py b/calc.py
index e262439..e5457b7 100755
--- a/calc.py
+++ b/calc.py
@@ -2,58 +2,59 @@
 # coding=utf-8
 """
 calc.py - Phenny Calculator Module
-Copyright 2008, Sean B. Palmer, inamidst.com
-Modified by Sfan5 2012
-Licensed under the Eiffel Forum License 2.
-
-http://inamidst.com/phenny/
+Copyright 2014, sfan5
 """
 
-import re
-import web
+import ast
+import operator as op
+import math
 
-r_result = re.compile(r'(?i)<A NAME=results>(.*?)</A>')
-r_tag = re.compile(r'<\S+.*?>')
+# http://stackoverflow.com/questions/2371436/evaluating-a-mathematical-expression-in-a-string/9558001#9558001
+operators = {ast.Add: op.add, ast.Sub: op.sub, ast.Mult: op.mul, ast.Pow: op.pow,
+			 ast.Div: op.truediv, ast.BitXor: op.xor, ast.Mod: op.mod}
 
-subs = [
-   (' in ', ' -> '), 
-   (' over ', ' / '), 
-   (u'£', 'GBP '), 
-   (u'€', 'EUR '), 
-   ('\$', 'USD '), 
-   (r'\bKB\b', 'kilobytes'), 
-   (r'\bMB\b', 'megabytes'), 
-   (r'\bGB\b', 'kilobytes'), 
-   ('kbps', '(kilobits / second)'), 
-   ('mbps', '(megabits / second)')
-]
+funcs = {"bin": bin, "abs": abs, "oct": oct}
+
+for funcn in dir(math):
+	if funcn.startswith("__"):
+		continue
+	funcs[funcn] = getattr(math, funcn)
+
+def getfunc(fn):
+	if fn in funcs:
+		return funcs[fn]
+	else:
+		raise ValueError("Function not allowed: '" + str(fn) + "'")
+
+def eval_expr(expr):
+	return eval_(ast.parse(expr).body[0].value) # Module(body=[Expr(value=...)])
+
+def eval_(node):
+	if isinstance(node, ast.Num): # <number>
+		return node.n
+	elif isinstance(node, ast.operator): # <operator>
+		return operators[type(node)]
+	elif isinstance(node, ast.BinOp): # <left> <operator> <right>
+		return eval_(node.op)(eval_(node.left), eval_(node.right))
+	elif isinstance(node, ast.Call): # <func> ( <args> )
+		return getfunc(node.func.id)(*(eval_(e) for e in node.func.args))
+	else:
+		raise TypeError("AST node type not allowed: '" + type(node).__name__ + "'")
+
+def c(phenny, input):
+	for x in phenny.bot.commands["high"].values():
+		if x[0].__name__ == "aa_hook":
+			if x[0](phenny, input):
+				return # Abort function
+	if not input.group(2):
+		return phenny.reply("Nothing to calculate.")
+	q = input.group(2).encode('utf-8')
+	print("[LOG]: %s calculated '%s'" % (input.nick,q))
+	try:
+		phenny.say(eval_expr(q))
+   	except Exception as e:
+   		phenny.say("Exception: " + str(e))
 
-def c(phenny, input): 
-   """Google calculator."""
-   for x in phenny.bot.commands["high"].values():
-      if x[0].__name__ == "aa_hook":
-         if x[0](phenny, input):
-            return # Abort function
-   if not input.group(2):
-      return phenny.reply("Nothing to calculate.")
-   q = input.group(2).encode('utf-8')
-   q = q.replace('\xcf\x95', 'phi') # utf-8 U+03D5
-   q = q.replace('\xcf\x80', 'pi') # utf-8 U+03C0
-   print("[LOG]: %s calculated '%s'" % (input.nick,q))
-   uri = 'http://www.google.com/ig/calculator?q='
-   bytes = web.get(uri + web.urllib.quote(q))
-   parts = bytes.split('",')
-   answer = [p for p in parts if p.startswith('rhs: "')][0][6:]
-   if answer: 
-      answer = answer.decode('unicode-escape')
-      answer = ''.join(chr(ord(c)) for c in answer)
-      answer = answer.decode('utf-8')
-      answer = answer.replace(u'\xc2\xa0', ',')
-      answer = answer.replace('<sup>', '^(')
-      answer = answer.replace('</sup>', ')')
-      answer = web.decode(answer)
-      phenny.say(answer)
-   else: phenny.say('Sorry, no result.')
 c.commands = ['c']
 c.example = '.c 5 + 3'