a %jB@sddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z Gdddej Z Gdd d ejZGd d d ejZGd d d e ejZGddde ejZdddddZeedS)) annotationsN)Any) IDNAError_max_domain_length_unicode_dots_realabeldecodeencodeulabelc@s8eZdZdZd ddddddZdddd dd d Zd S)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`. strictstrtuple[bytes, int])dataerrorsreturncCs4|dkrtd|ddd|s$dSt|t|fS)Nr Unsupported error handling ""unsupported_errorscode)r)rr lenselfrrr0E:\soccer\ml_env\Lib\site-packages\idna/codec.pyr s z Codec.encodebytestuple[str, int]cCs4|dkrtd|ddd|s$dSt|t|fS)Nr rrrrr)rr rrrrrr s z Codec.decodeN)r )r )__name__ __module__ __qualname____doc__r r rrrrr s  r csneZdZdZddddfdd Zddfd d Zd dd d Zd ddddZdddddddZZ 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. The 253-octet domain length limit (254 with a trailing dot) that :func:`idna.encode` enforces is applied to the accumulated output, so that streaming a name and encoding it in one shot either both succeed with the same result or both raise :exc:`~idna.IDNAError`. Only the ``"strict"`` error handler is supported. r rNonerrcst|d|_d|_dSNrF)super__init___emitted _trailing_dotrr __class__rrr+8s zIncrementalEncoder.__init__rcstd|_d|_dSr))r*resetr,r-rr/rrr2=s zIncrementalEncoder.resetrcCs |js|jsdS|j|j|jfSNr)bufferr,r-r3rrrgetstateBs zIncrementalEncoder.getstatestatercCs"|r|\|_|_|_n|dSN)r5r,r-r2rr8rrrsetstateGszIncrementalEncoder.setstateboolrrrfinalrc Cs|dkrtd|dddd}d}|rt|}d}|r`|dsNd }|d=n|s`|d=|r`d }g}|D]*} |t| |r|d 7}|t| 7}qhd ||}|t|7}|jt|7_|r|d |_ |j p| } |jt | krtd d d||fS) Nr rrrrrr.rDomain too longdomain_too_long) rrsplitappendrrjoinr,endswithr-r) rrrr>Z result_bytessizelabels trailing_dotresultlabelZmay_end_with_dotrrr_buffer_encodeMs<     z!IncrementalEncoder._buffer_encode)r ) r"r#r$r%r+r2r6r;rL __classcell__rrr/rr&'s r&csneZdZdZddddfdd Zddfd d Zd dd d Zd ddddZdddddddZZ 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. The 254-octet input length limit that :func:`idna.decode` enforces is applied to the accumulated input, so that streaming a name and decoding it in one shot either both succeed with the same result or both raise :exc:`~idna.IDNAError`. Only the ``"strict"`` error handler is supported. r rr'r(cst|d|_dSr4)r*r+ _consumedr.r/rrr+s zIncrementalDecoder.__init__r1cstd|_dSr4)r*r2rOr3r/rrr2s zIncrementalDecoder.resetrcCs |j|jfSr9r5rOr3rrrr6szIncrementalDecoder.getstater7cCs|\|_|_dSr9rPr:rrrr;szIncrementalDecoder.setstaterr<rr=c Cs2|dkrtd|ddd|s$dSt|tspzt|d}Wn2tyn}ztdd d|WYd}~n d}~00|jt|td krtd d dt|}d }|r|dsd}|d=n|s|d=|rd}g}d}|D]*} | t | |r|d 7}|t| 7}qd ||} |t|7}|j|7_| |fS)Nr rrrrr asciizInvalid ASCII in A-labelZ invalid_asciirrArBr!r?.r) r isinstancerUnicodeDecodeErrorrOrrrrCrDr rE) rrrr>errrHrIrJrGrKZ result_strrrr_buffer_decodes@ $   z!IncrementalDecoder._buffer_decode)r ) r"r#r$r%r+r2r6r;rVrMrrr/rrNvs rNc@s eZdZdS) StreamWriterNr"r#r$rrrrrWsrWc@s eZdZdS) StreamReaderNrXrrrrrYsrYrzcodecs.CodecInfo | None)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)rZr r incrementalencoderincrementaldecoder streamwriter streamreader) codecs CodecInfor r r r&rNrWrY)rZrrrsearch_functions ra) __future__rr_typingrcorerrrrr r r r BufferedIncrementalEncoderr&BufferedIncrementalDecoderrNrWrYraregisterrrrrs  $OK