| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- """
- Adds this import line:
- from builtins import XYZ
- for each of the functions XYZ that is used in the module.
- """
- 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_builtins = '''filter map zip
- ascii chr hex input next oct open round super
- bytes dict int range str'''.split()
- expression = '|'.join(["name='{0}'".format(name) for name in replaced_builtins])
- class FixFutureBuiltins(fixer_base.BaseFix):
- BM_compatible = True
- run_order = 9
- # Currently we only match uses as a function. This doesn't match e.g.:
- # if isinstance(s, str):
- # ...
- PATTERN = """
- power<
- ({0}) trailer< '(' args=[any] ')' >
- rest=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))
|
