| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859 |
- """
- 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))
|
