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.
21 lines
479 B
21 lines
479 B
4 years ago
|
"""
|
||
|
For the ``future`` package.
|
||
|
|
||
|
Turns any xrange calls into range calls and adds this import line:
|
||
|
|
||
|
from builtins import range
|
||
|
|
||
|
at the top.
|
||
|
"""
|
||
|
|
||
|
from lib2to3.fixes.fix_xrange import FixXrange
|
||
|
|
||
|
from libfuturize.fixer_util import touch_import_top
|
||
|
|
||
|
|
||
|
class FixXrangeWithImport(FixXrange):
|
||
|
def transform(self, node, results):
|
||
|
result = super(FixXrangeWithImport, self).transform(node, results)
|
||
|
touch_import_top('builtins', 'range', node)
|
||
|
return result
|