Staging
v0.8.1
https://github.com/python/cpython
Raw File
Tip revision: 65b2eb9dfebe873e8ee5821f13d5a8c07d9c89c2 authored by Georg Brandl on 05 December 2010, 11:42:38 UTC
Fix title.
Tip revision: 65b2eb9
fix_unicode.py
"""Fixer that changes unicode to str, unichr to chr, and u"..." into "...".

"""

import re
from ..pgen2 import token
from .. import fixer_base

_mapping = {"unichr" : "chr", "unicode" : "str"}
_literal_re = re.compile(r"[uU][rR]?[\'\"]")

class FixUnicode(fixer_base.BaseFix):
    BM_compatible = True
    PATTERN = "STRING | 'unicode' | 'unichr'"

    def transform(self, node, results):
        if node.type == token.NAME:
            new = node.clone()
            new.value = _mapping[node.value]
            return new
        elif node.type == token.STRING:
            if _literal_re.match(node.value):
                new = node.clone()
                new.value = new.value[1:]
                return new
back to top