You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
1001 B
29 lines
1001 B
5 years ago
|
"""
|
||
|
|
For the ``future`` package.
|
||
|
|
|
||
|
|
A special fixer that ensures that these lines have been added::
|
||
|
|
|
||
|
|
from future import standard_library
|
||
|
|
standard_library.install_hooks()
|
||
|
|
|
||
|
|
even if the only module imported was ``urllib``, in which case the regular fixer
|
||
|
|
wouldn't have added these lines.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
from lib2to3.fixes.fix_urllib import FixUrllib
|
||
|
|
from libfuturize.fixer_util import touch_import_top, find_root
|
||
|
|
|
||
|
|
|
||
|
|
class FixFutureStandardLibraryUrllib(FixUrllib): # not a subclass of FixImports
|
||
|
|
run_order = 8
|
||
|
|
|
||
|
|
def transform(self, node, results):
|
||
|
|
# transform_member() in lib2to3/fixes/fix_urllib.py breaks node so find_root(node)
|
||
|
|
# no longer works after the super() call below. So we find the root first:
|
||
|
|
root = find_root(node)
|
||
|
|
result = super(FixFutureStandardLibraryUrllib, self).transform(node, results)
|
||
|
|
# TODO: add a blank line between any __future__ imports and this?
|
||
|
|
touch_import_top(u'future', u'standard_library', root)
|
||
|
|
return result
|