diff options
Diffstat (limited to 'Lib/lib2to3/fixes/fix_execfile.py')
-rw-r--r-- | Lib/lib2to3/fixes/fix_execfile.py | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/Lib/lib2to3/fixes/fix_execfile.py b/Lib/lib2to3/fixes/fix_execfile.py new file mode 100644 index 0000000..0e67f09 --- /dev/null +++ b/Lib/lib2to3/fixes/fix_execfile.py @@ -0,0 +1,37 @@ +# Copyright 2006 Google, Inc. All Rights Reserved. +# Licensed to PSF under a Contributor Agreement. + +"""Fixer for execfile. + +This converts usages of the execfile function into calls to the built-in +exec() function. +""" + +from .. import pytree +from . import basefix +from .util import Comma, Name, Call, LParen, RParen, Dot + + +class FixExecfile(basefix.BaseFix): + + PATTERN = """ + power< 'execfile' trailer< '(' arglist< filename=any [',' globals=any [',' locals=any ] ] > ')' > > + | + power< 'execfile' trailer< '(' filename=any ')' > > + """ + + def transform(self, node, results): + assert results + syms = self.syms + filename = results["filename"] + globals = results.get("globals") + locals = results.get("locals") + args = [Name('open'), LParen(), filename.clone(), RParen(), Dot(), + Name('read'), LParen(), RParen()] + args[0].set_prefix("") + if globals is not None: + args.extend([Comma(), globals.clone()]) + if locals is not None: + args.extend([Comma(), locals.clone()]) + + return Call(Name("exec"), args, prefix=node.get_prefix()) |