fix_getcwd.py

from lib2to3 import fixer_base
from lib2to3.pytree import Node
from lib2to3.pygram import python_symbols as syms
from lib2to3.fixer_util import Name, ArgList
import libmodernize


class FixGetcwd(fixer_base.ConditionalFix):

    BM_compatible = True
    skip_on = "six.moves.getcwd"

    PATTERN = """
              power< 'os' trailer< dot='.' name='getcwdu' > any* >
              """

    def transform(self, node, results):
        new = Node(syms.power, [Name("getcwd"), ArgList([])], prefix="")
        new.prefix = node.prefix
        libmodernize.touch_import(u'six.moves', u'getcwd', node)
        return new