69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
from constant import consts
|
|
from simpleeval import EvalWithCompoundTypes, DEFAULT_FUNCTIONS, DEFAULT_NAMES, DEFAULT_OPERATORS
|
|
import ast
|
|
import sys
|
|
import operator as op
|
|
import math
|
|
import statistics
|
|
import cmath
|
|
import decimal
|
|
import fractions
|
|
|
|
funcs = {}
|
|
named_funcs = {}
|
|
|
|
|
|
def add_func(module):
|
|
for name in dir(module):
|
|
item = getattr(module, name)
|
|
if not name.startswith('_') and callable(item):
|
|
funcs[name] = item
|
|
|
|
|
|
def add_named_func(module):
|
|
named_funcs[module.__name__] = {}
|
|
for name in dir(module):
|
|
item = getattr(module, name)
|
|
if not name.startswith('_') and callable(item):
|
|
named_funcs[module.__name__][name] = item
|
|
|
|
|
|
add_func(math)
|
|
add_func(statistics)
|
|
add_named_func(cmath)
|
|
add_named_func(decimal)
|
|
add_named_func(fractions)
|
|
|
|
s_eval = EvalWithCompoundTypes(
|
|
operators={
|
|
**DEFAULT_OPERATORS,
|
|
ast.BitOr: op.or_,
|
|
ast.BitAnd: op.and_,
|
|
ast.BitXor: op.xor,
|
|
ast.Invert: op.invert,
|
|
},
|
|
functions={**funcs, **DEFAULT_FUNCTIONS, 'complex': complex,
|
|
'divmod': divmod, 'len': len, 'round': round},
|
|
names={
|
|
**DEFAULT_NAMES, **consts, **named_funcs,
|
|
'pi': math.pi,
|
|
'e': math.e,
|
|
'tau': math.tau,
|
|
'inf': math.inf, 'nan': math.nan,
|
|
'cmath': {
|
|
'pi': cmath.pi,
|
|
'e': cmath.e,
|
|
'tau': cmath.tau,
|
|
'inf': cmath.inf,
|
|
'infj': cmath.infj,
|
|
'nan': cmath.nan,
|
|
'nanj': cmath.nanj,
|
|
**named_funcs['cmath'],
|
|
},
|
|
}, )
|
|
|
|
try: # rina's rina lazy solution :rina:
|
|
sys.stdout.write('Result ' + str(s_eval.eval(' '.join(sys.argv[1:]))))
|
|
except Exception as e:
|
|
sys.stdout.write('Failed ' + str(e))
|
|
sys.exit()
|