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