"""
Fixer that changes os.getcwdu() to os.getcwd(),
and getcwdu(...) into getcwd(...).
"""
# Author: Victor Stinner

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

class FixGetcwdu(fixer_base.BaseFix):

    PATTERN = """
              power< 'os' trailer< dot='.' name='getcwdu' > trailer< '(' [any] ')' > >
              |
              power< name='getcwdu' trailer< '(' [any] ')' > any* >
              """

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