from lib2to3.fixer_base import BaseFix
from lib2to3.fixer_util import Name


class FixUgettext(BaseFix):
    """.ugettext -> .gettext"""

    PATTERN = """
    power< before=any+ trailer< '.' name='ugettext' > any* >
    """

    def transform(self, node, results):
        """See `BaseFix.transform()`"""
        name = results['name']
        name.replace(Name('gettext', prefix=name.prefix))