CODEWITHSUNDEEP
pythonparsingif-statementoperatorsmathematical-expressions
binhex
binhex

Reputation: 394

Python if-statement with variable mathematical operator

I'm trying to insert a variable mathematical operator into a if statement, an example of what I'm trying to achieve in parsing user-supplied mathematical expressions:

maths_operator = "=="

if "test" maths_operator "test":
       print "match found"

maths_operator = "!="

if "test" maths_operator "test":
       print "match found"
else:
       print "match not found"

obviously the above fails with SyntaxError: invalid syntax. I've tried using exec and eval but neither work in an if statement, what options do I have to get around this?

Upvotes: 18

Views: 4446

Answers (3)

w-m
w-m

Reputation: 11232

I've tried using exec and eval but neither work in an if statement

For the sake of completeness it should be mentioned that they do work, even if the posted answers provide a better solution. You'll have to eval() the whole comparison, not just the operator:

maths_operator = "=="

if eval('"test"' + maths_operator '"test"'):
       print "match found"

or exec the line:

exec 'if "test"' + maths_operator + '"test": print "match found"'

Upvotes: 1

Mark Byers
Mark Byers

Reputation: 837906

Use the operator module:

import operator
op = operator.eq

if op("test", "test"):
   print "match found"

Upvotes: 15

Nathan
Nathan

Reputation: 4937

Use the operator package together with a dictionary to look up the operators according to their text equivalents. All of these must be either unary or binary operators to work consistently.

import operator
ops = {'==' : operator.eq,
       '!=' : operator.ne,
       '<=' : operator.le,
       '>=' : operator.ge,
       '>'  : operator.gt,
       '<'  : operator.lt}

maths_operator = "=="

if ops[maths_operator]("test", "test"):
    print "match found"

maths_operator = "!="

if ops[maths_operator]("test", "test"):
    print "match found"
else:
    print "match not found"

Upvotes: 21

Related Questions