o j@sddlmZddlZddlmZddlmZmZmZm Z m Z m Z Gdddej Z Gdd d ej ZGd d d ejZGd d d e ejZGddde ejZdddZeedS)) annotationsN)Any) IDNAError_unicode_dots_realabeldecodeencodeulabelc@s(eZdZdZdddd Zddd d ZdS)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`. strictdatastrerrorsreturntuple[bytes, int]cC4|dkrtd|ddd|sdSt|t|fS)Nr Unsupported error handling ""unsupported_errorscoder)rr lenselfr rrH/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/idna/codec.pyr  z Codec.encodebytestuple[str, int]cCr)Nr rrrrr)rrrrrrrrrz Codec.decodeN)r )r rrrrr)r r rrrr!)__name__ __module__ __qualname____doc__r rrrrrr s  r c@seZdZdZd d d Zd S) 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. r rrfinalboolrrc Cs|dkrtd|ddd|sdSt|}d}|r.|ds%d }|d=n |s.|d=|r.d }g}d }|D]}|t||rC|d 7}|t|7}q4d ||} |t|7}| |fS) Nr rrrrrr.rr)rrsplitappendrrjoin) rr rr)labels trailing_dotresultsizelabel result_bytesrrr_buffer_encode3s0  z!IncrementalEncoder._buffer_encodeN)r rrrr)r*rr)r$r%r&r'r6rrrrr('s r(c@seZdZdZd d d Zd S)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 rrrr)r*rr!c Cs|dkrtd|ddd|sdSt|ts2zt|d}Wnty1}ztdd d|d}~wwt|}d }|rN|d sEd }|d =n |sN|d =|rNd }g}d }|D]} |t| |rc|d7}|t| 7}qTd ||} |t|7}| |fS)Nr rrrrr"asciizInvalid ASCII in A-label invalid_asciir#r+.rr) r isinstancerUnicodeDecodeErrorrr-r.r rr/) rr rr)errr0r1r2r3r4 result_strrrr_buffer_decode]s>   z!IncrementalDecoder._buffer_decodeN)r rrrr)r*rr!)r$r%r&r'r?rrrrr7Ss r7c@ eZdZdS) StreamWriterNr$r%r&rrrrrArAc@r@) StreamReaderNrBrrrrrDrCrDnamerrcodecs.CodecInfo | Nonec Cs,|dkrdStj|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``. idna2008N)rEr rincrementalencoderincrementaldecoder streamwriter streamreader) codecs CodecInfor r rr(r7rArD)rErrrsearch_functions rN)rErrrF) __future__rrLtypingrcorerrrrr r r BufferedIncrementalEncoderr(BufferedIncrementalDecoderr7rArDrNregisterrrrrs   ,0