1# Copyright 2006 Google, Inc. All Rights Reserved. 2# Licensed to PSF under a Contributor Agreement. 3 4"""Fixer for exec. 5 6This converts usages of the exec statement into calls to a built-in 7exec() function. 8 9exec code in ns1, ns2 -> exec(code, ns1, ns2) 10""" 11 12# Local imports 13from .. import fixer_base 14from ..fixer_util import Comma, Name, Call 15 16 17class FixExec(fixer_base.BaseFix): 18 BM_compatible = True 19 20 PATTERN = """ 21 exec_stmt< 'exec' a=any 'in' b=any [',' c=any] > 22 | 23 exec_stmt< 'exec' (not atom<'(' [any] ')'>) a=any > 24 """ 25 26 def transform(self, node, results): 27 assert results 28 syms = self.syms 29 a = results["a"] 30 b = results.get("b") 31 c = results.get("c") 32 args = [a.clone()] 33 args[0].prefix = "" 34 if b is not None: 35 args.extend([Comma(), b.clone()]) 36 if c is not None: 37 args.extend([Comma(), c.clone()]) 38 39 return Call(Name("exec"), args, prefix=node.prefix) 40