o TjB@sddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z Gdddej Z Gdd d ejZGd d d ejZGd d d e ejZGddde ejZdddZeedS)) annotationsN)Any) IDNAError_max_domain_length_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_errorscode)r)rr lenselfrrrM/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/idna/codec.pyr  z Codec.encodebytestuple[str, int]cCr)Nr rrrrr)rr rrrrrr rz Codec.decodeNr )rrrrrr)rr rrrr!)__name__ __module__ __qualname____doc__r r rrrrr s  r csPeZdZdZddfdd Zdfd d Zdd d ZdddZdddZZ 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 rrrNonecst|d|_d|_dSNrF)super__init___emitted _trailing_dotrr __class__rrr-8s  zIncrementalEncoder.__init__cstd|_d|_dSr+)r,resetr.r/rr1rrr3=s  zIncrementalEncoder.resetrcCs |js|jsdS|j|j|jfSNr)bufferr.r/r4rrrgetstateBs zIncrementalEncoder.getstatestatecCs$|r |\|_|_|_dS|dSN)r6r.r/r3rr8rrrsetstateGs zIncrementalEncoder.setstaterfinalboolrc Cs|dkrtd|dddd}d}|rWt|}d}|r0|ds'd }|d=n |s0|d=|r0d }g}|D]} |t| |rC|d 7}|t| 7}q4d ||}|t|7}|jt|7_|rh|d |_ |j pm| } |jt | kr{td d d||fS) Nr rrrrrr.rDomain too longdomain_too_long) rrsplitappendrrjoinr.endswithr/r) rrrr< result_bytessizelabels trailing_dotresultlabelmay_end_with_dotrrr_buffer_encodeMs<     z!IncrementalEncoder._buffer_encoder$rrrr*rr*)rr)r8rrr*)rrrrr<r=rr) r%r&r'r(r-r3r7r;rM __classcell__rrr1rr)'s  r)csPeZdZdZddfdd Zdfd d Zdd d ZdddZdddZZ 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 rrrr*cst|d|_dSr5)r,r- _consumedr0r1rrr-s  zIncrementalDecoder.__init__cstd|_dSr5)r,r3rRr4r1rrr3s  zIncrementalDecoder.resetrcCs |j|jfSr9r6rRr4rrrr7s zIncrementalDecoder.getstater8cCs|\|_|_dSr9rSr:rrrr;szIncrementalDecoder.setstaterrr<r=r!c Cs&|dkrtd|ddd|sdSt|ts2zt|d}Wnty1}ztdd d|d}~ww|jt|td krCtd d dt|}d }|r_|dsVd}|d=n |s_|d=|r_d}g}d}|D]} | t | |rt|d 7}|t| 7}qed ||} |t|7}|j|7_| |fS)Nr rrrrr"asciizInvalid ASCII in A-label invalid_asciirr@rAr#r>.r) r isinstancerUnicodeDecodeErrorrRrrrrBrCr rD) rrrr<errrHrIrJrGrK result_strrrr_buffer_decodesD    z!IncrementalDecoder._buffer_decoder$rNrO)rr)r8rrr*)rrrrr<r=rr!) r%r&r'r(r-r3r7r;r[rPrrr1rrQvs  rQc@ eZdZdS) StreamWriterNr%r&r'rrrrr]r]c@r\) StreamReaderNr^rrrrr`r_r`namerrcodecs.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)rar r incrementalencoderincrementaldecoder streamwriter streamreader) codecs CodecInfor r r r)rQr]r`)rarrrsearch_functions rj)rarrrb) __future__rrhtypingrcorerrrrr r r r BufferedIncrementalEncoderr)BufferedIncrementalDecoderrQr]r`rjregisterrrrrs  $OK