3
�Qg�# � @ s� d dl Z d dlZd dlZd dlmZ ejd�ZdZdZdd� Z dd � Z
d
d � Z G d d
� d
ej �Z G dd� dej
�ZG dd� dej�ZG dd� de ej�ZG dd� de ej�Zdd� ZdS )� N)� ucd_3_2_0u [.。．｡]s xn--zxn--c C s g }x(| D ] }t j|�rq
|jt j|�� q
W dj|�} tjd| �} xr| D ]j}t j|�s�t j|�s�t j |�s�t j
|�s�t j |�s�t j |�s�t j
|�s�t j|�s�t j|�rJtd| ��qJW dd� | D �}t|��rtdd� | D ��r�td��|d p�|d �rtd ��| S )
N� �NFKCzInvalid character %rc S s g | ]}t j|��qS � )�
stringprepZ in_table_d1)�.0�xr r �&/usr/lib64/python3.6/encodings/idna.py�
<listcomp>) s z nameprep.<locals>.<listcomp>c s s | ]}t j|�V qd S )N)r Z in_table_d2)r r r r r � <genexpr>1 s znameprep.<locals>.<genexpr>z Violation of BIDI requirement 2r � z Violation of BIDI requirement 3���)r Z in_table_b1�appendZ map_table_b2�join� unicodedata� normalizeZ in_table_c12Z in_table_c22Z in_table_c3Z in_table_c4Z in_table_c5Z in_table_c6Z in_table_c7Z in_table_c8Z in_table_c9� UnicodeError�any)�labelZnewlabel�cZRandALr r r �nameprep s2
r c C s� y| j d�} W n tk
r" Y n*X dt| � k o:dk n rD| S td��t| �} y| j d�} W n tk
rv Y n*X dt| � k o�dk n r�| S td��| jt�r�td��| j d�} t| } dt| � k o�dk n r�| S td��d S )N�asciir �@ zlabel empty or too longz Label starts with ACE prefix�punycode)�encoder �lenr �
startswith� sace_prefix�
ace_prefix)r r r r �ToASCII< s,
r c C s� t | t�rd}n,y| jd�} d}W n tk
r: d}Y nX |stt| �} y| jd�} W n tk
rr td��Y nX | jt�s�t| d�S | tt�d � }|j d�}t
|�}t| d�j � t|d�kr�td| |��|S )NTr Fz Invalid character in IDN labelr zIDNA does not round-trip) �
isinstance�bytesr r r r r �strr �decoder �lower)r Z
pure_asciiZlabel1�resultZlabel2r r r � ToUnicodeg s*
r& c @ s e Zd Zddd�Zddd�ZdS ) �Codec�strictc C s |dkrt d| ��|s d S y|jd�}W n tk
r> Y njX |jd�}x8|d d � D ](}dt|� k ordk n sXt d ��qXW t|d
�dkr�t d
��|t|�fS t� }tj|�}|r�|d r�d}|d= nd}x(|D ] }|r�|jd� |jt|�� q�W t || �t|�fS )Nr( zunsupported error handling � r r � .r r zlabel empty or too longzlabel too long)r) r r
r
r
r
)
r r �UnicodeEncodeError�splitr � bytearray�dots�extendr r! )�self�input�errorsr% �labelsr � trailing_dotr r r r � s4
z Codec.encodec C s� |dkrt d| ��|s d S t|t�s.t|�}t|kr^y|jd�t|�fS tk
r\ Y nX |jd�}|r�t|d
�dkr�d}|d = nd}g }x|D ]}|jt |�� q�W dj
|�| t|�fS ) Nr( zUnsupported error handling r r r r* r �.)r r r
r
) r r r! r r# r �UnicodeDecodeErrorr, r r& r )r0 r1 r2 r3 r4 r% r r r r r# � s(
z Codec.decodeN)r( )r( )�__name__�
__module__� __qualname__r r# r r r r r' � s
%r' c @ s e Zd Zdd� ZdS )�IncrementalEncoderc C s� |dkrt d| ��|s dS tj|�}d}|rT|d sBd}|d = n|sT|d
= |rTd}t� }d}x<|D ]4}|r~|jd� |d7 }|jt|�� |t|�7 }qdW ||7 }|t|�7 }t|�|fS ) Nr( zunsupported error handling r) r r r* )r) r r
r
r
)r r. r, r- r/ r r r! ) r0 r1 r2 �finalr3 r4 r% �sizer r r r �_buffer_encode� s2
z!IncrementalEncoder._buffer_encodeN)r7 r8 r9 r= r r r r r: � s r: c @ s e Zd Zdd� ZdS )�IncrementalDecoderc C s� |dkrt d| ��|s dS t|t�r2tj|�}nt|d�}|jd�}d}|rt|d sbd}|d
= n|st|d = |rtd}g }d}x2|D ]*}|jt|�� |r�|d7 }|t|�7 }q�W dj|�| }|t|�7 }||fS ) Nr( zUnsupported error handling r r r r5 r )r r r
r
r
) r r r" r. r, r r& r r ) r0 r1 r2 r; r3 r4 r% r< r r r r �_buffer_decode� s6
z!IncrementalDecoder._buffer_decodeN)r7 r8 r9 r? r r r r r> � s r> c @ s e Zd ZdS )� StreamWriterN)r7 r8 r9 r r r r r@ " s r@ c @ s e Zd ZdS )� StreamReaderN)r7 r8 r9 r r r r rA % s rA c C s t jdt� jt� jttttd�S )NZidna)�namer r# �incrementalencoder�incrementaldecoder� streamwriter� streamreader) �codecs� CodecInfor' r r# r: r> r@ rA r r r r � getregentry* s rI )r �rerG r r �compiler. r r r r r& r' �BufferedIncrementalEncoderr: �BufferedIncrementalDecoderr> r@ rA rI r r r r �<module> s
.+)H#'