o .j^:@s0UddlmZddlmZddlmZddlmZmZm Z ddl m Z m Z hdZ ded<dLd d ZdMddZdNddZdOddZdPddZdQddZdRd d!ZdNd"d#ZdSd'd(ZdLd)d*ZdLd+d,ZdTd/d0ZdUd2d3ZdVd5d6ZdWd7d8ZdXd9d:ZdYd>d?ZdXd@dAZ dXdBdCZ!dZdDdEZ"d[dIdJZ#dKS)\) annotations) decode_header)dumps)findallsearchsub)AnyIterable>as_in_is_and_def_for_not_elif_from_pass_break_while_yield_assert_except_return_finally_zset[str]RESERVED_KEYWORDstringstrreturncCs|ddS)z Normalize a string by applying on it lowercase and replacing '-' to '_'. >>> normalize_str("Content-Type") 'content_type' >>> normalize_str("X-content-type") 'x_content_type' -_)lowerreplacerr$X/home/thesage/.local/lib/python3.10/site-packages/niquests/_vendor/kiss_headers/utils.py normalize_strr&strings list[str]cCsttt|S)zJNormalize a list of string by applying fn normalize_str over each element.)listmapr&)r(r$r$r%normalize_list(sr,namecCsLt|dkr|S|ddkr|dr|dd}|tvr$|dd}|S)a; By choice, this project aims to allow developers to access header or attribute in header by using the property notation. Some keywords are protected by the language itself. So : When starting by a number, prepend an underscore to it. When using a protected keyword, append an underscore to it. >>> unpack_protected_keyword("_3to1") '3to1' >>> unpack_protected_keyword("from_") 'from' >>> unpack_protected_keyword("_from") '_from' >>> unpack_protected_keyword("3") '3' >>> unpack_protected_keyword("FroM_") 'FroM_' rr N)lenisdigitrr-r$r$r%unpack_protected_keyword-s   r4type_type str | NonecCstdt|}|r |dSdS)z5 Typically extract a class name from a Type. zrN)rr)r5rr$r$r%extract_class_nameIsr9 delimiterc CsNt|dks |dvrtdd}d}d}d}dg}t|tdt|D]u\}}|dkr5| }|r4|s4d}n |dkr>|s>d }n|d krG|rGd}n|d koT||d |d v}|sr|s`|d kr`d }n|dkrh|rhd}|rr||krr|srd}||kr|p}|p}|p}|dur|d|d<|dq#|d|7<q#|r|d|d<|S)ak Take a string and split it according to the passed delimiter. It will ignore delimiter if inside between double quote, inside a value, or in parenthesis. The input string is considered perfectly formed. This function does not split coma on a day when attached, see "RFC 7231, section 7.1.1.2: Date". >>> header_content_split("Wed, 15-Apr-2020 21:27:31 GMT, Fri, 01-Jan-2038 00:00:00 GMT", ",") ['Wed, 15-Apr-2020 21:27:31 GMT', 'Fri, 01-Jan-2038 00:00:00 GMT'] >>> header_content_split('quic=":443"; ma=2592000; v="46,43", h3-Q050=":443"; ma=2592000, h3-Q049=":443"; ma=2592000', ",") ['quic=":443"; ma=2592000; v="46,43"', 'h3-Q050=":443"; ma=2592000', 'h3-Q049=":443"; ma=2592000'] >>> header_content_split("Mozilla/5.0 (Macintosh; Intel Mac OS X 10.9; rv:50.0) Gecko/20100101 Firefox/50.0", ";") ['Mozilla/5.0 (Macintosh; Intel Mac OS X 10.9; rv:50.0) Gecko/20100101 Firefox/50.0'] >>> header_content_split("text/html; charset=UTF-8", ";") ['text/html', 'charset=UTF-8'] >>> header_content_split('text/html; charset="UTF-\"8"', ";") ['text/html', 'charset="UTF-"8"'] r/> ,;z9Delimiter should be either semi-colon, a coma or a space.Fr"(T)>FriMonSatSunThuTueWed=r=r0)r1 ValueErrorziprangelstriprstripappend) rr:in_double_quotein_parenthesisin_value is_on_a_dayresultletterindexr$r$r%header_content_splitQsD     rXcCst|dr |jdur |jSt|dddd}|dr&|dd}|dr1|dd}d }|D]}|rF|d krF|d |7}q5||7}q5|S) a_ Take a type and infer its header name. >>> from kiss_headers.builder import ContentType, XContentTypeOptions, BasicAuthorization >>> class_to_header_name(ContentType) 'Content-Type' >>> class_to_header_name(XContentTypeOptions) 'X-Content-Type-Options' >>> class_to_header_name(BasicAuthorization) 'Authorization' __override__N'.r0r r/r>r)hasattrrYrsplitendswith startswithisupper)r5class_raw_name header_namerVr$r$r%class_to_header_names       rd root_typec Cst|dd}|D]:}t|}|durq t|dr!|jdus0t|dd|kr0|S|rFzt||WStyEYq wq td|d) a The opposite of class_to_header_name function. Will raise TypeError if no corresponding entry is found. Do it recursively from the root type. >>> from kiss_headers.builder import CustomHeader, ContentType, XContentTypeOptions, LastModified, Date >>> header_name_to_class("Content-Type", CustomHeader) >>> header_name_to_class("Last-Modified", CustomHeader) r r>NrYr\r0z+Cannot find a class matching header named 'z'.) r&r"__subclasses__r9r]rYr^header_name_to_class TypeError)r-renormalized_namesubclass class_namer$r$r%rgs$   rgcCs"ddd|dddDS)a# Take a header name and prettify it. >>> prettify_header_name("x-hEllo-wORLD") 'X-Hello-World' >>> prettify_header_name("server") 'Server' >>> prettify_header_name("contEnt-TYPE") 'Content-Type' >>> prettify_header_name("content_type") 'Content-Type' rcSsg|]}|qSr$) capitalize).0elr$r$r% sz(prettify_header_name..r )joinr"r^r3r$r$r%prettify_header_names" rqitemsIterable[tuple[str, Any]]list[tuple[str, str]]cCsvt}|D]3\}}d}t|D]!\}}t|tr||7}t|tr0||j|dur*|nddd7}q|||fq|S)u This function takes a list of tuples, representing headers by key, value. Where value is bytes or string containing (RFC 2047 encoded) partials fragments like the following : >>> decode_partials([("Subject", "=?iso-8859-1?q?p=F6stal?=")]) [('Subject', 'pöstal')] r>Nutf-8ignore)errors)r*r isinstancerbytesdecoderP)rr revised_itemsheadcontentrevised_contentpartialpartial_encodingr$r$r%decode_partialss   rcCsDt|dkr|dr|ds|dr |dr |ddS|S)z Remove simple quote or double quote around a string if any. >>> unquote('"hello"') 'hello' >>> unquote('"hello') '"hello' >>> unquote('"a"') 'a' >>> unquote('""') '' r.r?rZr/r0)r1r`r_r#r$r$r%unquote s  rcCsdt|dS)z Surround string by a double quote char. >>> quote("hello") '"hello"' >>> quote('"hello') '""hello"' >>> quote('"hello"') '"hello"' r?)rr#r$r$r%quotes rr}intcCs"|drt|dddSdS)a A recursive function that counts trailing white space at the end of the given string. >>> count_leftover_space("hello ") 3 >>> count_leftover_space("byebye ") 1 >>> count_leftover_space(" hello ") 1 >>> count_leftover_space(" hello ") 4 r;Nr0r/r)r_count_leftover_spacer}r$r$r%r,s relemcCsd}z||}Wn ty|YSw|t|}dt|d||}z |||dd}Wn ty=Ynw|||durM|||dndddd}|drc|dd}|drn|dd}|S)aJ Remove a member for a given header content and take care of the unneeded leftover semi-colon. >>> header_strip("text/html; charset=UTF-8; format=flowed", "charset=UTF-8") 'text/html; format=flowed' >>> header_strip("text/html; charset=UTF-8; format=flowed", "charset=UTF-8") 'text/html; format=flowed' Nr;r=r>r/r0) rWrKr1rr"rOrNr`r_)r}rnext_semi_colon_index elem_indexelem_end_indexr$r$r% header_strip=s:        rboolcCs|dko td|duS)a Verify if a provided header name is valid. >>> is_legal_header_name(":hello") False >>> is_legal_header_name("hello") True >>> is_legal_header_name("Content-Type") True >>> is_legal_header_name("Hello;") False >>> is_legal_header_name("Hello\rWorld") False >>> is_legal_header_name("Hello \tWorld") False >>> is_legal_header_name('Hello World"') False >>> is_legal_header_name("Hello-World/") True >>> is_legal_header_name("") False r>z+[^\x21-\x7F]|[:;(),<>=@?\[\]\r\n\t &{}\"\\]N)rr3r$r$r%is_legal_header_namems rcCs td|S)a Extract parts of content that are considered as comments. Between parenthesis. >>> extract_comments("Mozilla/5.0 (Macintosh; Intel Mac OS X 10.9; rv:50.0) Gecko/20100101 Firefox/50.0 (hello) llll (abc)") ['Macintosh; Intel Mac OS X 10.9; rv:50.0', 'hello', 'abc'] z \(([^)]+)\))rrr$r$r%extract_commentss rcCs tdd|S)aLSome header content may have folded content (CRLF + n spaces) in it, making your job at reading them a little more difficult. This function undoes the folding in the given content. >>> unfold("___utmvbtouVBFmB=gZg\r\n XbNOjalT: Lte; path=/; Max-Age=900") '___utmvbtouVBFmB=gZg XbNOjalT: Lte; path=/; Max-Age=900' z\r\n[ ]+r;)rrr$r$r%unfolds rpayloadrytuple[str, bytes]c Csd}|}d}t|tdt|D],\}}|dkr)|d||ddfSz ||dd7}Wqty>Ynw|d||ddfS) uThis function's purpose is to extract lines that can be decoded using the UTF-8 decoder. >>> extract_encoded_headers("Host: developer.mozilla.org\r\nX-Hello-World: 死の漢字\r\n\r\n".encode("utf-8")) ('Host: developer.mozilla.org\r\nX-Hello-World: 死の漢字\r\n', b'') >>> extract_encoded_headers("Host: developer.mozilla.org\r\nX-Hello-World: 死の漢字\r\n\r\nThat IS totally random.".encode("utf-8")) ('Host: developer.mozilla.org\r\nX-Hello-World: 死の漢字\r\n', b'That IS totally random.') r>rs r/Nruz ) splitlinesrLrMr1rprzUnicodeDecodeError)rrUlinesrWliner$r$r%extract_encoded_headerss rcCs |ddS)z Replace escaped double quote in content by removing the backslash. >>> unescape_double_quote(r'UTF"-8') 'UTF"-8' >>> unescape_double_quote(r'UTF"-8') 'UTF"-8' \"r?)r"rr$r$r%unescape_double_quotes rcCst|ddS)z Replace not escaped double quote in content by adding a backslash beforehand. >>> escape_double_quote(r'UTF\"-8') 'UTF\\"-8' >>> escape_double_quote(r'UTF"-8') 'UTF\\"-8' r?r)rr"rr$r$r%escape_double_quoter'rcCs0|}|dr|dp|do|dS)zj Sometime, you may receive a header that hold a JSON list or object. This function detect it. {}[])stripr`r_rr$r$r%is_content_json_objectsrheaders)Iterable[tuple[str | bytes, str | bytes]]Iterable[tuple[str, str]]cCsg}|D]<\}}|dur qt|tr|d}t|tr"|d}nt|tdur9t|ttfr5t|}nt|}|||fq|S)Nutf_8F)rxryrzrdictr*rrP)rdecodedkvr$r$r%transform_possible_encodeds      rN)rrrr)r(r)rr))r-rrr)r5r6rr7)rrr:rrr))r5r6rr)r-rrer6rr6)rrrsrrt)r}rrr)r}rrrrr)r-rrr)r}rrr))r}rrr)rryrr)r}rrr)rrrr)$ __future__r email.headerrjsonrrerrrtypingrr r__annotations__r&r,r4r9rXrdrgrqrrrrrrrrrrrrrr$r$r$r%s6       J ! $     0