1"""Fixer that changes input(...) into eval(input(...)).""" 2# Author: Andre Roberge 3 4# Local imports 5from .. import fixer_base 6from ..fixer_util import Call, Name 7from .. import patcomp 8 9 10context = patcomp.compile_pattern("power< 'eval' trailer< '(' any ')' > >") 11 12 13class FixInput(fixer_base.BaseFix): 14 BM_compatible = True 15 PATTERN = """ 16 power< 'input' args=trailer< '(' [any] ')' > > 17 """ 18 19 def transform(self, node, results): 20 # If we're already wrapped in an eval() call, we're done. 21 if context.match(node.parent.parent): 22 return 23 24 new = node.clone() 25 new.prefix = "" 26 return Call(Name("eval"), [new], prefix=node.prefix) 27