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 ejZG d d d ejZG d d! d!ejZG d"d# d#eejZG d$d% d%eejZd&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 ) bytearraysetordappendaddsortedbytes)strbaseextendedc 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
maxresr
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 Nr )r r )len)r
charindexposlr
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 ZoldcharresultZoldindexr
r r r Zcurlendeltar r r insertion_unsort0 s$
r c C s, d| d | }|dk r dS |dkr(dS |S )N$ r r )jbiasr 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 firstnumcharsZ 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 adaptationH r r )r enumerater' extendr- r )Zbaselendeltasr 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}xyt | | }W n, tk
rJ |dkr>td|d dfS X |d7 }d| kofdkn rv|d }nDd| kod kn r|d
}n"|dkrtd | | n|dfS t||} ||| 7 }|| k r||fS |d | }|d7 }qW dS )
z(3.3 Generalized variable-length integersr r strictzincomplete punicode stringNA 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 rt||||\}}|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 | tr| jd} t | tr&t| } | jd}|dkrLd}t| dj }n.t| d | d|}t| |d d dj }t|||S )Nasciir4 r r )
isinstancer
encode
memoryviewr rfindupperrF )r5 r@ r r r r r r punycode_decode s
rN c @ s e Zd ZdddZdddZdS ) Codecr7 c C s t |}|t|fS )N)r6 r )selfinputr@ r r r r rJ s z Codec.encodec C s* |dkrt d| t||}|t|fS )Nr7 replaceignorezUnsupported 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 ZdddZdS )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 ZdddZdS )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 jttttdS )NZpunycode)namerJ rT incrementalencoderincrementaldecoder 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 s(