Source code for monty.operator

"""
Useful additional functions for operators
"""
import operator


[docs]def operator_from_str(op): """ Return the operator associated to the given string `op`. raises: `KeyError` if invalid string. """ d = { "==": operator.eq, "!=": operator.ne, ">": operator.gt, ">=": operator.ge, "<": operator.lt, "<=": operator.le, "+": operator.add, "-": operator.sub, "*": operator.mul, "%": operator.mod, "^": operator.xor, } try: d["/"] = operator.truediv except AttributeError: pass return d[op]