| 1 | # Copyright 2006 Google, Inc. All Rights Reserved.
|
|---|
| 2 | # Licensed to PSF under a Contributor Agreement.
|
|---|
| 3 |
|
|---|
| 4 | """Fixer for execfile.
|
|---|
| 5 |
|
|---|
| 6 | This converts usages of the execfile function into calls to the built-in
|
|---|
| 7 | exec() function.
|
|---|
| 8 | """
|
|---|
| 9 |
|
|---|
| 10 | from .. import fixer_base
|
|---|
| 11 | from ..fixer_util import (Comma, Name, Call, LParen, RParen, Dot, Node,
|
|---|
| 12 | ArgList, String, syms)
|
|---|
| 13 |
|
|---|
| 14 |
|
|---|
| 15 | class FixExecfile(fixer_base.BaseFix):
|
|---|
| 16 | BM_compatible = True
|
|---|
| 17 |
|
|---|
| 18 | PATTERN = """
|
|---|
| 19 | power< 'execfile' trailer< '(' arglist< filename=any [',' globals=any [',' locals=any ] ] > ')' > >
|
|---|
| 20 | |
|
|---|
| 21 | power< 'execfile' trailer< '(' filename=any ')' > >
|
|---|
| 22 | """
|
|---|
| 23 |
|
|---|
| 24 | def transform(self, node, results):
|
|---|
| 25 | assert results
|
|---|
| 26 | filename = results["filename"]
|
|---|
| 27 | globals = results.get("globals")
|
|---|
| 28 | locals = results.get("locals")
|
|---|
| 29 |
|
|---|
| 30 | # Copy over the prefix from the right parentheses end of the execfile
|
|---|
| 31 | # call.
|
|---|
| 32 | execfile_paren = node.children[-1].children[-1].clone()
|
|---|
| 33 | # Construct open().read().
|
|---|
| 34 | open_args = ArgList([filename.clone()], rparen=execfile_paren)
|
|---|
| 35 | open_call = Node(syms.power, [Name(u"open"), open_args])
|
|---|
| 36 | read = [Node(syms.trailer, [Dot(), Name(u'read')]),
|
|---|
| 37 | Node(syms.trailer, [LParen(), RParen()])]
|
|---|
| 38 | open_expr = [open_call] + read
|
|---|
| 39 | # Wrap the open call in a compile call. This is so the filename will be
|
|---|
| 40 | # preserved in the execed code.
|
|---|
| 41 | filename_arg = filename.clone()
|
|---|
| 42 | filename_arg.prefix = u" "
|
|---|
| 43 | exec_str = String(u"'exec'", u" ")
|
|---|
| 44 | compile_args = open_expr + [Comma(), filename_arg, Comma(), exec_str]
|
|---|
| 45 | compile_call = Call(Name(u"compile"), compile_args, u"")
|
|---|
| 46 | # Finally, replace the execfile call with an exec call.
|
|---|
| 47 | args = [compile_call]
|
|---|
| 48 | if globals is not None:
|
|---|
| 49 | args.extend([Comma(), globals.clone()])
|
|---|
| 50 | if locals is not None:
|
|---|
| 51 | args.extend([Comma(), locals.clone()])
|
|---|
| 52 | return Call(Name(u"exec"), args, prefix=node.prefix)
|
|---|
