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