3
\� � @ s� d Z ddlZdd� Zdd� Zdd� Zd d
� Zd d � Zd
Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZG d d � d ej�ZG d d � d ej�ZG d d!� d!ej�ZG d"d#� d#eej�ZG d$d%� d%eej�Zd&d'� ZdS )(uY Codec for the Punicode encoding, as specified in RFC 3492
Written by Martin v. Löwis.
� Nc C sT t � }t� }x2| D ]*}t|�dk r2|jt|�� q|j|� qW t|�}t|�|fS )z 3.1 Basic code point segregation� )� bytearray�set�ord�append�add�sorted�bytes)�str�base�extended�c� r �*/usr/lib64/python3.6/encodings/punycode.py� segregate
s
r c C s* d}x | D ]}t |�|k r
|d7 }q
W |S )z@Return the length of str, considering only characters below max.r � )r )r
�max�resr
r r r �
selective_len s
r c C sR t | �}xD|d7 }||kr dS | | }||kr:|d |fS ||k r
|d7 }q
W dS )a Return a pair (index, pos), indicating the next occurrence of
char in str. index is the position of the character considering
only ordinals up to and including char, and pos is the position in
the full string. index/pos is the starting position in the full
string.r N���r )r r )�len)r
�char�index�pos�lr
r r r �selective_find s r c C s� d}g }d}x~|D ]v}d }}t |�}t| |�} | d || }
xBt| |||�\}}|dkr^P |
|| 7 }
|j|
d � |}d}
qBW |}qW |S )z3.2 Insertion unsort codingr r r r r r )r r r r ) r
r Zoldchar�resultZoldindexr
r r r Zcurlen�deltar r r �insertion_unsort0 s$
r c C s, d| d | }|dk r dS |dkr(dS |S )N�$ r � r )�j�biasr r r r �TF s r# s$ abcdefghijklmnopqrstuvwxyz0123456789c C sr t � }d}xbt||�}| |k r4|jt| � t|�S |jt|| | d| � | | d| } |d7 }q W dS )z(3.3 Generalized variable-length integersr r r N)r r# r �digitsr )�Nr" r r! �tr r r � generate_generalized_integerN s
r' c C s\ |r| d } n| d } | | | 7 } d}x | dkrB| d } |d7 }q(W |d| | d }|S )Ni� � r i� �# r �& r )r �first�numcharsZ divisionsr" r r r �adapt[ s
r- c C sT t � }d}x@t|�D ]4\}}t||�}|j|� t||dk| | d �}qW t|�S )z3.4 Bias adaptation�H r r )r � enumerater' �extendr- r )Zbaselen�deltasr r" Zpointsr �sr r r �generate_integersj s
r3 c C s8 t | �\}}t| |�}tt|�|�}|r4|d | S |S )N� -)r r r3 r )�textr r r1 r r r �punycode_encodeu s
r6 c
C s� d}d}d}x�yt | | �}W n, tk
rJ |dkr>td��|d dfS X |d7 }d| kofdkn rv|d }nDd| ko�d kn r�|d
}n"|dkr�td | | ��n|dfS t||�} ||| 7 }|| k r�||fS |d | }|d7 }qW dS )
z(3.3 Generalized variable-length integersr r �strictzincomplete punicode stringN�A �Z �0 �9 � z Invalid extended code point '%s'r )r �
IndexError� UnicodeErrorr# )
r �extposr" �errorsr �wr! r �digitr& r r r �decode_generalized_number s2
rC c C s� d}d
}d}d}x�|t |�k r�t||||�\}}|dkr<| S ||d 7 }||t | �d 7 }|dkr�|dkrxtd| ��td �}|t | �d }| d|� t|� | |d� } t||dkt | ��}|}qW | S ) z3.2 Insertion unsort codingr r r. r Ni�� r7 zInvalid character U+%x�?r )r rC r> r �chrr- ) r r r@ r r r" r? Znewposr r r r �insertion_sort� s( rF c C s� t | t�r| jd�} t | t�r&t| �} | jd�}|dkrLd}t| d�j� }n.t| d |� d|�}t| |d d � d�j� }t|||�S )N�asciir4 r � r )�
isinstancer
�encode�
memoryviewr �rfind�upperrF )r5 r@ r r r r r r �punycode_decode� s
rN c @ s e Zd Zddd�Zddd�ZdS ) �Codecr7 c C s t |�}|t|�fS )N)r6 r )�self�inputr@ r r r r rJ � s z Codec.encodec C s* |dkrt d| ��t||�}|t|�fS )Nr7 �replace�ignorezUnsupported error handling )r7 rR rS )r> rN r )rP rQ r@ r r r r �decode� s
z Codec.decodeN)r7 )r7 )�__name__�
__module__� __qualname__rJ rT r r r r rO � s
rO c @ s e Zd Zddd�ZdS )�IncrementalEncoderFc C s t |�S )N)r6 )rP rQ �finalr r r rJ � s zIncrementalEncoder.encodeN)F)rU rV rW rJ r r r r rX � s rX c @ s e Zd Zddd�ZdS )�IncrementalDecoderFc C s$ | j dkrtd| j ��t|| j �S )Nr7 rR rS zUnsupported error handling )r7 rR rS )r@ r> rN )rP rQ rY r r r rT � s
zIncrementalDecoder.decodeN)F)rU rV rW rT r r r r rZ � s rZ c @ s e Zd ZdS )� StreamWriterN)rU rV rW r r r r r[ � s r[ c @ s e Zd ZdS )� StreamReaderN)rU rV rW r r r r r\ � s r\ c C s t jdt� jt� jttttd�S )NZpunycode)�namerJ rT �incrementalencoder�incrementaldecoder� streamwriter� streamreader) �codecs� CodecInforO rJ rT rX rZ r[ r\ r r r r � getregentry� s rd )�__doc__rb r r r r r# r$ r' r- r3 r6 rC rF rN rO rX rZ r[ r\ rd r r r r �<module> s(