| Current Path : /opt/alt/python37/lib/python3.7/site-packages/libfuturize/fixes/ |
| Current File : //opt/alt/python37/lib/python3.7/site-packages/libfuturize/fixes/fix_xrange_with_import.py |
"""
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