U j@sddlZddlmZmZmZddlmZmZmZm Z m Z m Z Gdddej Z Gdddej ZGd d d ejZGd d d e ejZGd dde ejZeeejdddZeedS)N)AnyOptionalTuple) IDNAError_unicode_dots_realabeldecodeencodeulabelc@sHeZdZdZd eeeeefdddZd eeeeefdddZ dS) CodecafStateless IDNA 2008 codec. Implements the :class:`codecs.Codec` protocol so that the whole-domain encoder (:func:`idna.encode`) and decoder (:func:`idna.decode`) are accessible through the standard codec machinery as ``"idna2008"``. Only the ``"strict"`` error handler is supported; any other handler raises :exc:`~idna.IDNAError`. strict)dataerrorsreturncCs0|dkrtd|d|s dSt|t|fS)Nr Unsupported error handling ""r)rr lenselfrrr./tmp/pip-unpacked-wheel-j_8njjuc/idna/codec.pyr s z Codec.encodecCs0|dkrtd|d|s dSt|t|fS)Nr rrr)rr rrrrrr s z Codec.decodeN)r )r ) __name__ __module__ __qualname____doc__strrbytesintr r rrrrr s  r c@s,eZdZdZeeeeeefdddZ dS)IncrementalEncoderaIncremental IDNA 2008 encoder. Buffers a partial trailing label across calls until either the next label separator is seen or ``final=True``, so that streamed input is encoded one whole label at a time. Any of the four Unicode label separators (``U+002E``, ``U+3002``, ``U+FF0E``, ``U+FF61``) ends a label; the result always uses ``U+002E`` as the separator. Only the ``"strict"`` error handler is supported. rrfinalrc Cs|dkrtd|d|s dSt|}d}|rX|dsFd}|d=n|sX|d=|rXd}g}d}|D]*}|t||r|d 7}|t|7}qdd||} |t|7}| |fS) Nr rrrr.rr)rrsplitappendrrjoin) rrrr%labels trailing_dotresultsizelabelZ result_bytesrrr_buffer_encode1s0  z!IncrementalEncoder._buffer_encodeN) rrrrr boolrr!r"r0rrrrr#%s r#c@s,eZdZdZeeeeeefdddZ dS)IncrementalDecoderaIncremental IDNA 2008 decoder. Buffers a partial trailing label across calls until either the next label separator is seen or ``final=True``, so that streamed input is decoded one whole label at a time. Only the ``"strict"`` error handler is supported. r$c Cs|dkrtd|d|s dSt|ts4t|d}t|}d}|rl|dsZd}|d=n|sl|d=|rld}g}d }|D]*}|t||r|d 7}|t|7}qxd||} |t|7}| |fS) Nr rrrasciirr&.rr) r isinstancer rr(r)r rr*) rrrr%r+r,r-r.r/Z result_strrrr_buffer_decode\s4    z!IncrementalDecoder._buffer_decodeN) rrrrrr r1rr"r6rrrrr2Rs r2c@s eZdZdS) StreamWriterNrrrrrrrr7sr7c@s eZdZdS) StreamReaderNr8rrrrr9sr9)namerc Cs,|dkr dStj|tjtjttttdS)aCodec search function registered with :mod:`codecs`. Returns a :class:`codecs.CodecInfo` for the ``"idna2008"`` codec name so that ``str.encode("idna2008")`` and ``bytes.decode("idna2008")`` invoke the IDNA 2008 codec defined in this module. :param name: The codec name being looked up. :returns: A :class:`codecs.CodecInfo` instance if ``name`` is ``"idna2008"``, otherwise ``None``. Zidna2008N)r:r r incrementalencoderincrementaldecoder streamwriter streamreader) codecs CodecInfor r r r#r2r7r9)r:rrrsearch_functions rA)r?typingrrrcorerrrr r r r BufferedIncrementalEncoderr#BufferedIncrementalDecoderr2r7r9r r@rAregisterrrrrs --