"""
For the ``future`` package.
Adds this import line::
from builtins import XYZ
for each of the functions XYZ that is used in the module.
Adds these imports after any other imports (in an initial block of them).
"""
from __future__ import unicode_literals
from lib2to3 import fixer_base
from lib2to3.pygram import python_symbols as syms
from lib2to3.fixer_util import Name, Call, in_special_context
from libfuturize.fixer_util import touch_import_top
# All builtins are:
# from future.builtins.iterators import (filter, map, zip)
# from future.builtins.misc import (ascii, chr, hex, input, isinstance, oct, open, round, super)
# from future.types import (bytes, dict, int, range, str)
# We don't need isinstance any more.
replaced_builtin_fns = '''filter map zip
ascii chr hex input next oct
bytes range str raw_input'''.split()
# This includes raw_input as a workaround for the
# lib2to3 fixer for raw_input on Py3 (only), allowing
# the correct import to be included. (Py3 seems to run
# the fixers the wrong way around, perhaps ignoring the
# run_order class attribute below ...)
expression = '|'.join(["name='{0}'".format(name) for name in replaced_builtin_fns])
class FixFutureBuiltins(fixer_base.BaseFix):
BM_compatible = True
run_order = 7
# Currently we only match uses as a function. This doesn't match e.g.:
# if isinstance(s, str):
# ...
PATTERN = """
power<
({0}) trailer< '(' [arglist=any] ')' >
rest=any* >
|
power<
'map' trailer< '(' [arglist=any] ')' >
>
""".format(expression)
def transform(self, node, results):
name = results["name"]
touch_import_top(u'builtins', name.value, node)
# name.replace(Name(u"input", prefix=name.prefix))