o j"@sdZddlmZmZmZmZddlmZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZmZe jdkrJdd lmZmZdd lmZmZn dd lmZmZmZmZdd dZddZddZe dedddZ!dddZ"dS)z Functions to convert unicode IRIs into ASCII byte string URIs and back. Exports the following items: - iri_to_uri() - uri_to_iri() )unicode_literalsdivisionabsolute_importprint_function)idnaN)unwrap)byte_clsstr_cls type_name bytes_to_list int_types))urlsplit urlunsplit)quoteunquote)runquote_to_bytesrrFcCst|tsttdt|d}tjdkrO|dsO|dsOd}t d|}|r8| d}d|t |d}t |}|rN||dd}t |dd }nt |}|dur\t |j}|j}|durh|d }t |jd d }t |jd d }|j} | durt| d } d} |dur| |7} |r| d|7} | d7} |dur| |7} | dur|dko| dk} |dko| dk} |r| s| s| d| 7} t |jdd } t |jdd }t |jdd }|r|dur|dur| dkrd} | durd} t|| | ||f}t|tr|d}|S)z Encodes a unicode IRI into an ASCII byte string URI :param value: A unicode string of an IRI :param normalize: A bool that controls URI normalization :return: A byte string of the ASCII-encoded URI z@ value must be a unicode string, not %s N)zhttp://zhttps://z ^[^:]*://rrrz !$&'()*+,;=safeascii:@shttps80shttpss443z/!$&'()*+,;=@:z/?!$&'()*+,;=@:/latin1) isinstancer TypeErrorrr sys version_info startswithrematchgrouplenr _urlquoteschemehostnameencodeusernamepasswordportpathqueryfragmentr)value normalizer* real_prefix prefix_matchparsedr+r-r.r/netloc default_http default_httpsr0r1r2outputr</root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asn1crypto/_iri.py iri_to_uri%sh          r>c Cs(t|tsttdt|t|}|j}|dur|d}t|j ddgd}t|j ddgd}|j }|r:|d}|j }|rIt|t sI|d}d}|dur_||7}|r[|d|7}|d7}|durg||7}|durs|dt|7}t|jd gd d }t|jd d gd d } t|j} t|||| | fS)z Converts an ASCII URI byte string into a unicode IRI :param value: An ASCII-encoded byte string of the URI :return: A unicode string of the IRI z= value must be a byte string, not %s Nr:@)remaprr/T)rApreserve&=)r r r!rr rr*decode _urlunquoter-r.r+r/r r r0r1r2r) r3r7r*r-r.r+r/r8r0r1r2r<r<r= uri_to_irixs>      rHcCs4t|j|j|j}dd|D}d||jfS)a> Error handler for decoding UTF-8 parts of a URI into an IRI. Leaves byte sequences encoded in %XX format, but as part of a unicode string. :param exc: The UnicodeDecodeError exception :return: A 2-element tuple of (replacement unicode string, integer index to resume at) cSsg|]}d|qS)%%%02xr<).0numr<r<r= sz,_iri_utf8_errors_handler..r)r objectstartendjoin)exc bytes_as_ints replacementsr<r<r=_iri_utf8_errors_handlers rTiriutf8rcs|dus|dkr dSgtd|r,fdd}td||}fdd}td||}t|d d d }t|tsB|d }td krUfd d}td||}|S)a Quotes a unicode string for use in a URL :param string: A unicode string :param safe: A unicode string of character to not encode :return: None (if string is None) or an ASCII byte string of the quoted string Nrz%[0-9a-fA-F]{2}cs@t|d}|dd}tD] }||dt|}q|S)Nrutf-8rUrI)rr'rFlistreplaceord)r& byte_stringunicode_string safe_charrr<r= _try_unescapes   z _urlquote.._try_unescapez(?:%[0-9a-fA-F]{2})+cs|dddS)Nrr)appendr'r,)r&escapesr<r=_extract_escapesz"_urlquote.._extract_escaperVrrrcs dS)Nr)pop)_r`r<r=_return_escapes z!_urlquote.._return_escapes%00)r%searchsuburlquoter,r r r()stringrr]rbr;rer<)rarr=r)s       r)c Cs|dur|S|dkr dS|r.gd}i}|D]}|d}|||<||d|d}qt|}|rI|D]}||ddt|d}q6|dd }|r`|D] \}}|||}qU|S) a Unquotes a URI portion from a byte string into unicode using UTF-8 :param byte_string: A byte string of the data to unquote :param remap: A list of characters (as unicode) that should be re-mapped to a %XX encoding. This is used when characters are not valid in part of a URL. :param preserve: A bool - indicates that the chars to be remapped if they occur in non-hex form, should be preserved. E.g. / for URL path. :return: A unicode string Nrr)rrrIrVrU)rcrXr,rrYrFitems) rZrArCrSpreserve_unmapchar replacementr;originalr<r<r=rGs( " rG)F)r)NN)#__doc__ __future__rrrr encodingsrcodecsr%r"_errorsr_typesr r r r r r#urlparserrurllibrrhrr urllib.parser>rHrTregister_errorr)rGr<r<r<r=s$    S4  4