summaryrefslogtreecommitdiffstats
path: root/Lib/lib2to3/fixes/fix_raw_input.py
blob: d9130d01245406046d8ddd09023dabe1dba9ba8c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
"""Fixer that changes raw_input(...) into input(...)."""
# Author: Andre Roberge

# Local imports
from .. import fixer_base
from ..fixer_util import Name

class FixRawInput(fixer_base.BaseFix):

    PATTERN = """
              power< name='raw_input' trailer< '(' [any] ')' > >
              """

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