Index: third_party/google-endpoints/libfuturize/fixes/fix_future_standard_library.py |
diff --git a/third_party/google-endpoints/libfuturize/fixes/fix_future_standard_library.py b/third_party/google-endpoints/libfuturize/fixes/fix_future_standard_library.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..501c2a940e814bfc57e9d09cfd073399d3c725fa |
--- /dev/null |
+++ b/third_party/google-endpoints/libfuturize/fixes/fix_future_standard_library.py |
@@ -0,0 +1,26 @@ |
+""" |
+For the ``future`` package. |
+ |
+Changes any imports needed to reflect the standard library reorganization. Also |
+Also adds these import lines: |
+ |
+ from future import standard_library |
+ standard_library.install_aliases() |
+ |
+after any __future__ imports but before any other imports. |
+""" |
+ |
+from lib2to3.fixes.fix_imports import FixImports |
+from libfuturize.fixer_util import touch_import_top |
+ |
+ |
+class FixFutureStandardLibrary(FixImports): |
+ run_order = 8 |
+ |
+ def transform(self, node, results): |
+ result = super(FixFutureStandardLibrary, self).transform(node, results) |
+ # TODO: add a blank line between any __future__ imports and this? |
+ touch_import_top(u'future', u'standard_library', node) |
+ return result |
+ |
+ |