def test_interpret(self): sys_platform = sys.platform version = sys.version.split()[0] os_name = os.name platform_version = platform.version() platform_machine = platform.machine() platform_python_implementation = python_implementation() self.assertTrue(interpret("sys.platform == '%s'" % sys_platform)) self.assertTrue(interpret( "sys.platform == '%s' and python_full_version == '%s'" % (sys_platform, version))) self.assertTrue(interpret("'%s' == sys.platform" % sys_platform)) self.assertTrue(interpret('os.name == "%s"' % os_name)) self.assertTrue(interpret( 'platform.version == "%s" and platform.machine == "%s"' % (platform_version, platform_machine))) self.assertTrue(interpret('platform.python_implementation == "%s"' % platform_python_implementation)) # stuff that need to raise a syntax error ops = ('os.name == os.name', 'os.name == 2', "'2' == '2'", 'okpjonon', '', 'os.name ==', 'python_version == 2.4') for op in ops: self.assertRaises(SyntaxError, interpret, op) # combined operations OP = 'os.name == "%s"' % os_name FALSEOP = 'os.name == "buuuu"' AND = ' and ' OR = ' or ' self.assertTrue(interpret(OP + AND + OP)) self.assertTrue(interpret(OP + AND + OP + AND + OP)) self.assertTrue(interpret(OP + OR + OP)) self.assertTrue(interpret(OP + OR + FALSEOP)) self.assertTrue(interpret(OP + OR + OP + OR + FALSEOP)) self.assertTrue(interpret(OP + OR + FALSEOP + OR + FALSEOP)) self.assertTrue(interpret(FALSEOP + OR + OP)) self.assertFalse(interpret(FALSEOP + AND + FALSEOP)) self.assertFalse(interpret(FALSEOP + OR + FALSEOP)) # other operators self.assertTrue(interpret("os.name != 'buuuu'")) self.assertTrue(interpret("python_version > '1.0'")) self.assertTrue(interpret("python_version < '5.0'")) self.assertTrue(interpret("python_version <= '5.0'")) self.assertTrue(interpret("python_version >= '1.0'")) self.assertTrue(interpret("'%s' in os.name" % os_name)) self.assertTrue(interpret("'buuuu' not in os.name")) self.assertTrue(interpret( "'buuuu' not in os.name and '%s' in os.name" % os_name)) # execution context self.assertTrue(interpret('python_version == "0.1"', {'python_version': '0.1'}))
def _operate(operation, x, y): return _OPERATORS[operation](x, y) # restricted set of variables _VARS = {'sys.platform': sys.platform, 'python_version': '%s.%s' % sys.version_info[:2], # FIXME parsing sys.platform is not reliable, but there is no other # way to get e.g. 2.7.2+, and the PEP is defined with sys.version 'python_full_version': sys.version.split(' ', 1)[0], 'os.name': os.name, 'platform.version': platform.version(), 'platform.machine': platform.machine(), 'platform.python_implementation': python_implementation(), } class _Operation(object): def __init__(self, execution_context=None): self.left = None self.op = None self.right = None if execution_context is None: execution_context = {} self.execution_context = execution_context def _get_var(self, name): if name in self.execution_context: