80 lines
2.0 KiB
Python
Executable File
80 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
"""
|
|
calc.py - Phenny Calculator Module
|
|
Copyright 2014, sfan5
|
|
Licensed under GNU General Public License v2.0
|
|
"""
|
|
|
|
import math
|
|
import random
|
|
import struct
|
|
import multiprocessing
|
|
from functools import reduce
|
|
|
|
class SomeObject(object):
|
|
pass
|
|
|
|
env = {
|
|
"abs": abs, "all": all, "any": any,
|
|
"ascii": ascii, "bin": bin, "bool": bool,
|
|
"bytearray": bytearray, "bytes": bytes,
|
|
"callable": callable, "chr": chr,
|
|
"complex": complex, "dict": dict,
|
|
"divmod": divmod, "enumerate": enumerate,
|
|
"filter": filter, "float": float,
|
|
"format": format, "frozenset": frozenset,
|
|
"hex": hex, "int": int,
|
|
"iter": iter, "len": len, "list": list,
|
|
"map": map, "max": max, "min": min,
|
|
"next": next,"oct": oct,
|
|
"ord": ord, "pow": pow, "range": range,
|
|
"reduce": reduce, "repr": repr,
|
|
"reversed": reversed, "round": round,
|
|
"set": set, "slice": slice,
|
|
"sorted": sorted, "str": str,
|
|
"sum": sum, "tuple": tuple,
|
|
"type": type, "zip": zip,
|
|
}
|
|
|
|
libs = [
|
|
'math', 'random', 'struct'
|
|
]
|
|
|
|
for lib in libs:
|
|
env[lib] = SomeObject()
|
|
for funcn in dir(globals()[lib]):
|
|
if funcn.startswith("_"):
|
|
continue
|
|
setattr(env[lib], funcn, getattr(globals()[lib], funcn))
|
|
|
|
def c(phenny, input):
|
|
if not input.group(2):
|
|
return phenny.reply("Nothing to calculate.")
|
|
q = input.group(2)
|
|
if '__' in q:
|
|
return phenny.reply("Sorry, double underscores are not allowed.")
|
|
log.log("event", "%s calculated '%s'" % (log.fmt_user(input), q), phenny)
|
|
o = multiprocessing.Queue()
|
|
def get_result(o, q):
|
|
try:
|
|
o.put(repr(eval(q, {'__builtins__': env}, {})))
|
|
except Exception as e:
|
|
o.put(type(e).__name__ + ": " + str(e))
|
|
proc = multiprocessing.Process(target=get_result, args=(o,q))
|
|
proc.start()
|
|
proc.join(3.0)
|
|
if proc.is_alive():
|
|
proc.terminate()
|
|
log.log("action", "Calculation by %s timed out." % log.fmt_user(input), phenny)
|
|
phenny.reply("Took to long to calculate")
|
|
else:
|
|
phenny.say(o.get())
|
|
|
|
|
|
c.commands = ['c']
|
|
c.example = '.c 5 + 3'
|
|
|
|
if __name__ == '__main__':
|
|
print(__doc__.strip())
|