Index: third_party/google-endpoints/libpasteurize/fixes/fix_future_builtins.py |
diff --git a/third_party/google-endpoints/libpasteurize/fixes/fix_future_builtins.py b/third_party/google-endpoints/libpasteurize/fixes/fix_future_builtins.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..27339abc3bf6e878f0f6c6964aa49ed2e8c20076 |
--- /dev/null |
+++ b/third_party/google-endpoints/libpasteurize/fixes/fix_future_builtins.py |
@@ -0,0 +1,47 @@ |
+""" |
+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)) |
+ |