# Based on fix_xrange.py
# 3to2 modification by Paul Kippes

"""
range(...) -> xrange(...)
"""

from ..fixer_common import *

class FixRange(fixer_base.BaseFix):

    PATTERN = """
              power<
                 (name='range') trailer< '(' args=any ')' >
              rest=any* >
              """

    def transform(self, node, results):
        name = results["name"]
        name.replace(Name("xrange", prefix=name.get_prefix()))
        return node