3
\� � @ s< d Z ddlmZ ddlmZ ddd�ZG dd� dej�Zd S )
z�Fixer for unicode.
* Changes unicode to str and unichr to chr.
* If "...\u..." is not unicode literal change it into "...\\u...".
* Change u"..." into "...".
� )�token)�
fixer_base�chr�str)ZunichrZunicodec s, e Zd ZdZdZ� fdd�Zdd� Z� ZS )�
FixUnicodeTz STRING | 'unicode' | 'unichr'c s" t t| �j||� d|jk| _d S )N�unicode_literals)�superr �
start_treeZfuture_featuresr )�selfZtree�filename)� __class__� �1/usr/lib64/python3.6/lib2to3/fixes/fix_unicode.pyr s zFixUnicode.start_treec C s� |j tjkr$|j� }t|j |_|S |j tjkr�|j}| j rl|d dkrld|krldjdd� |j d�D ��}|d dkr�|dd � }||jkr�|S |j� }||_|S d S ) N� z'"�\z\\c S s g | ]}|j d d�j dd��qS )z\uz\\uz\Uz\\U)�replace)�.0�vr
r
r �
<listcomp>! s z(FixUnicode.transform.<locals>.<listcomp>ZuU� )
�typer �NAMEZclone�_mapping�value�STRINGr �join�split)r
ZnodeZresults�new�valr
r
r � transform s"
zFixUnicode.transform)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr r �
__classcell__r
r
)r r r s r N)�__doc__Zpgen2r � r r ZBaseFixr r
r
r
r �<module> s