42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
r"""Fixer for unicode.
|
|
|
|
* Changes unicode to str and unichr to chr.
|
|
|
|
* If "...\u..." is not unicode literal change it into "...\\u...".
|
|
|
|
* Change u"..." into "...".
|
|
|
|
"""
|
|
|
|
from ..pgen2 import token
|
|
from .. import fixer_base
|
|
|
|
_mapping = {"unichr" : "chr", "unicode" : "str"}
|
|
|
|
class FixUnicode(fixer_base.BaseFix):
|
|
BM_compatible = True
|
|
PATTERN = "STRING | 'unicode' | 'unichr'"
|
|
|
|
def start_tree(self, tree, filename):
|
|
super(FixUnicode, self).start_tree(tree, filename)
|
|
self.unicode_literals = 'unicode_literals' in tree.future_features
|
|
|
|
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:
|
|
val = node.value
|
|
if not self.unicode_literals and val[0] in '\'"' and '\\' in val:
|
|
val = r'\\'.join([
|
|
v.replace('\\u', r'\\u').replace('\\U', r'\\U')
|
|
for v in val.split(r'\\')
|
|
])
|
|
if val[0] in 'uU':
|
|
val = val[1:]
|
|
if val == node.value:
|
|
return node
|
|
new = node.clone()
|
|
new.value = val
|
|
return new
|