Merge pull request #51 from gingrspacecadet/patch-1

Implement math functions in expression evaluator

Author: youngcoder45

cogs/counting.py

@@ -13,6 +13,7 @@
 )
 import ast
 import operator
+import math
 import asyncio
 import time
 from typing import Optional
@@ -273,6 +274,36 @@ def safe_eval(self, expr):
             ast.USub: operator.neg
         }
 
+        constants = {
+            "e": math.e,
+            "pi": math.pi,
+            "tau": math.tau,
+        }
+
+        functions = {
+            "log": math.log,
+            "log10": math.log10,
+            "sqrt": math.sqrt,
+            "factorial": math.factorial,
+            "ceil": math.ceil,
+            "floor": math.floor,
+            "trunc": math.trunc,
+            "sin": math.sin,
+            "cos": math.cos,
+            "tan": math.tan,
+            "asin": math.asin,
+            "acos": math.acos
+            "atan": math.atan,
+            "degrees": math.degrees,
+            "radians": math.radians,
+            "sinh": math.sinh,
+            "cosh": math.cosh,
+            "tanh": math.tanh,
+            "asinh": math.asinh,
+            "acosh": math.acosh,
+            "atanh": math.atanh,
+        }
+
         def eval_node(node):
             if isinstance(node, ast.Constant):
                 if isinstance(node.value, (int, float)):
@@ -291,6 +322,16 @@ def eval_node(node):
                 op = type(node.op)
                 if op in operators:
                     return operators[op](eval_node(node.operand))
+            elif isinstance(node, ast.Call):
+                if isinstance(node.func, ast.Name):
+                    if node.func.id in functions:
+                        args = [eval_node(a) for a in node.args]
+                        return functions[node.func.id](*args)
+                    
+            elif isinstance(node, ast.Name):
+                if node.id in constants:
+                    return constants[node.id]
+                raise TypeError(f"Unknown identifier {node.id}")
             raise TypeError("Unsupported type")
 
         try: