o NK&hI @sddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z ddl mZeeeefZeeeefZeeeeeffZeeeefZd\ZZed ed ed ed ejZd ZeeZededZGdddZ GdddZ!Gddde"Z#dDddZ$dEddZ%ed ed ed!ed"ejZ&dFd$d%Z'dFd&d'Z(dGd+d,Z)dHd.d/Z*dId2d3Z+d4d5e,d6DZ-dJdH"z%MediaType.__repr__..)rjoinritemsrrrr__repr__Gs zMediaType.__repr__cCs>t|trd|vr td|j|kSt|tr|j|jkStS)zjCheck for mime (str or MediaType) identical type/subtype. Parameters such as q are not considered.r z&Use match() to compare with parameters) isinstancer ValueErrorrr NotImplemented)rotherrrr__eq__Js    zMediaType.__eq__mime_with_paramsUnion[str, MediaType]returnOptional[MediaType]cst|tr t|n|}|r>tfdd|jDr>j|jks,jdks,|jdkr>j|jks<jdks<|jdkr>SdS)aMatch this media type against another media type. Check if this media type matches the given mime type/subtype. Wildcards are supported both ways on both type and subtype. If mime contains a semicolon, optionally followed by parameters, the parameters of the two media types must match exactly. .. note:: Use the `==` operator instead to check for literal matches without expanding wildcards. Args: media_type (str): A type/subtype string to match. Returns: MediaType: Returns `self` if the media types are compatible. None: Returns `None` if the media types are not compatible. c3s$|] \}}j||kVqdSN)rrr"r*rrr&xr'z"MediaType.match..rN) r,rr _parseallrr)rr)rr1mtrr*rmatchWs      zMediaType.matchboolcCstdd|j|jfDS)zReturn True if this media type has a wildcard in it. Returns: bool: True if this media type has a wildcard in it. css|]}|dkVqdS)rNr)r#partrrrr&z)MediaType.has_wildcard..)anyrrr*rrr has_wildcardszMediaType.has_wildcardcCsz|}d|vr dS|d^}}|dd\}}|r|s$td|dddd|DD}|||fi|S) Nrr zInvalid media type: cSsi|] \}}||qSr)strip)r#rvaluerrr s z$MediaType._parse..css|] }|ddVqdS)r!r?N)split)r#paramrrrr&sz#MediaType._parse..)r@rCr-lstriprstrip)clsr1mtyper raw_paramsr rrrrrr6s zMediaType._parseN)r rrrrr)r1r2r3r4)r3r:)r1rr3r4) __name__ __module__ __qualname____doc__rr+r0r9propertyr> classmethodr6rrrrr $s  0 r c@s^eZdZdZdddZdd Zd d Zd d Zd ddZd!ddZ d"ddZ e d#ddZ dS)$MatchedaA matching result of a MIME string against a header. This class is a representation of a matching result of a MIME string against a header. It encapsulates the MIME string, the header, and provides methods for matching against other MIME strings. Args: mime (str): The MIME string to match. header (MediaType): The header to match against, if any. rrheaderr4cCs||_||_dSr5)rrQ)rrrQrrrrs zMatched.__init__cCs|r d|d|jdSdS)Nz rQr*rrrr+szMatched.__repr__cC|jSr5)rr*rrr__str__szMatched.__str__cCs |jduSr5rTr*rrr__bool__s zMatched.__bool__r/rr3r:cCsLz ||\}}Wn tyYdSwt|o$|jr|jp$|j o$|j S)NF)_compare TypeErrorr:rQ)rr/comp other_acceptrrrr0s  zMatched.__eq__tuple[bool, Matched]cCs\t|trt|}|j|krd|fS|}t|tr"|j|jk|fStd|jd|d)NTz8Comparison not supported between unequal mime types of 'z' and '')r,rrPparserrQrY)rr/parsedrrrrXs    zMatched._compareUnion[str, Matched]Optional[Matched]cCs>t|tr t|n|}|jr|jsdS|j|jr|SdS)aMatch this MIME string against another MIME string. Check if this MIME string matches the given MIME string. Wildcards are supported both ways on both type and subtype. Args: other (str): A MIME string to match. Returns: Matched: Returns `self` if the MIME strings are compatible. None: Returns `None` if the MIME strings are not compatible. N)r,rrPr^rQr9)rr/acceptrrrr9s  z Matched.matchrawcCst|}|||Sr5r r6)rGrc media_typerrrr^s  z Matched.parseN)rrrQr4)r/rr3r:)r3r\)r/r`r3ra)rcrr3rP) rJrKrLrMrr+rVrWr0rXr9rOr^rrrrrPs    rPc@s(eZdZdZddd dd Zd d Zd S) AcceptLista'A list of media types, as used in the Accept header. The Accept header entries are listed in order of preference, starting with the most preferred. This class is a list of `MediaType` objects, that encapsulate also the q value or any other parameters. Two separate methods are provided for searching the list: - 'match' for finding the most preferred match (wildcards supported) - operator 'in' for checking explicit matches (wildcards as literals) Args: *args (MediaType): Any number of MediaType objects. T)accept_wildcardsmimesrr3rPcs:tfddt|D}t|r|dddSdS)aFind a media type accepted by the client. This method can be used to find which of the media types requested by the client is most preferred against the ones given as arguments. The ordering of preference is set by: 1. The order set by RFC 7231, s. 5.3.2, giving a higher priority to q values and more specific type definitions, 2. The order of the arguments (first is most preferred), and 3. The first matching entry on the Accept header. Wildcards are matched both ways. A match is usually found, as the Accept headers typically include `*/*`, in particular if the header is missing, is not manually set, or if the client is a browser. Note: the returned object behaves as a string of the mime argument that matched, and is empty/falsy if no match was found. The matched header entry `MediaType` or `None` is available as the `m` attribute. Args: mimes (List[str]): Any MIME types to search for in order of preference. accept_wildcards (bool): Match Accept entries with wildcards in them. Returns: Match: A match object with the mime string and the MediaType object. c3sL|]!\}}s |jstD]\}}||r|j ||||fVqqdSr5)r> enumerater9r)r#jaccirrgrhrrr&s z#AcceptList.match..rN)rN)sortedrirP)rrgrharrmrr9szAcceptList.matchcCsddd|DS)z5Format as Accept header value (parsed, not original).z, css|]}t|VqdSr5)rr#mrrrr&'r<z%AcceptList.__str__..)r(r*rrrrV%szAcceptList.__str__N)rhrr3rP)rJrKrLrMr9rVrrrrrfs $rfrb Optional[str]r3cCsp|s |dkr tSd}zdddd|dDD}|sttt|ddd WSty7td |w) a~Parse an Accept header and order the acceptable media types according to RFC 7231, s. 5.3.2 https://datatracker.ietf.org/doc/html/rfc7231#section-5.3.2 Args: accept (str): The Accept header value to parse. Returns: AcceptList: A list of MediaType objects, ordered by preference. Raises: InvalidHeader: If the header value is invalid. rz*/*cSg|]}|r|qSrr)r#r8rrr =s z parse_accept..cSsg|]}t|qSrrd)r#rHrrrru?s,cSrUr5r)xrrrDszparse_accept..rwz Invalid header value in Accept: )rfrCr-ror)rbrprrr parse_accept*s rzrArtuple[str, Options]cCsP|d}|dkr i}nddt||dD}|d|}||fS)a Parse content-type and content-disposition header values. E.g. `form-data; name=upload; filename="file.txt"` to ('form-data', {'name': 'upload', 'filename': 'file.txt'}) Mostly identical to cgi.parse_header and werkzeug.parse_options_header but runs faster and handles special characters better. Unescapes %22 to `"` and %0D%0A to ` ` in field values. Args: value (str): The header value to parse. Returns: Tuple[str, Options]: The header value and a dict of options. r rcSs<i|]}|d|dp|dddddqS)r?z%22"z%0D%0A )grouplowerreplacerqrrrrB^s z(parse_content_header..N)find_paramfinditerr@r)rAposoptionsrrrparse_content_headerIs  rz(?:z)=z \s*($|[;,])Optional[Options]c Cs|dd}|j}|dus|sdSd|}||vrdSd}}g}d}t|dddD]P}||ks:|dkrC|r>nA|dd=|}|\} } } }| ddd} | p_| ddddd} | | | f| d vrv| |krvd }|r~|dkr~nq.|rt t |SdS) zParse RFC 7239 Forwarded headers. The value of `by` or `secret` must match `config.FORWARDED_SECRET` :return: dict with keys and values, or None if nothing matched forwardedNrvFrr z"\r~)secretbyT) getallFORWARDED_SECRETr(_rparamrstartendgroupsrrappend fwd_normalizereversed) headersconfigrQrseprrfoundrr val_token val_quotedrvalrrrparse_forwardedps4     rc s|j}|j}|o |ds=|r=|dksJz|j}dddd|DD}|| Wn ttfy<YnwsAdSfdd}t|S) z Parse traditional proxy headers.NrcSrtrr)r#prrrrus z$parse_xforwarded..css(|]}|dD]}|Vq qdS)rvN)rCr@)r#hrrrrr&s   z#parse_xforwarded..c3s0dfVdD] \}}||dfVqdS)Nfor))protozx-scheme)rzx-forwarded-proto)hostzx-forwarded-host)portzx-forwarded-port)pathzx-forwarded-path)getone)rrQaddrrrrrs   z!parse_xforwarded..options)REAL_IP_HEADER PROXIES_COUNTrrFORWARDED_FOR_HEADERKeyError IndexErrorr)rrreal_ip_header proxies_count forwarded_forproxiesrrrrparse_xforwardeds(   rfwdOptionsIterableOptionsc Csi}|D]D\}}|durHz2|dvrt|||<n%|dvr#|||<n|dkr.t|||<n|dkr9t|||<n|||<WqtyGYqwq|S)zNormalize and convert values extracted from forwarded headers. Args: fwd (OptionsIterable): An iterable of key-value pairs. Returns: Options: A dict of normalized key-value pairs. N)rr)rrrr)fwd_normalize_addressrintrr-)rretrrrrrrs&  rrcCs:|dkrt|dr|St|rd|d}|S)zNormalize address fields of proxy headers. Args: addr (str): An address string. Returns: str: A normalized address string. unknown_[])r- startswith_ipv6_re fullmatchr)rrrrrs    rr#tuple[Optional[str], Optional[int]]cCs<t|}|s dS|\}}||durt|fSdfS)zSplit host:port into hostname and port. Args: host (str): A host string. Returns: Tuple[Optional[str], Optional[int]]: A tuple of hostname and port. )NNN)_host_rerrrr)rrrrrrr parse_hosts  rcCs g|] }d|t|dfqS)sHTTP/1.1 %d %b sUNKNOWN)r r)r#statusrrrrusruirrrHeaderBytesIterablebytescCs*t|}|D]}|d|7}q|d7}|S)zFormat a HTTP/1.1 response header. Args: status (int): The HTTP status code. headers (HeaderBytesIterable): An iterable of header tuples. Returns: bytes: The formatted response header. s%b: %b s )_HTTP1_STATUSLINES)rrrrrrrformat_http1_responses  rrQprefixes!Optional[Union[list, tuple, set]]#tuple[Optional[str], Optional[str]]cCsT|r t|tttfs d}|dur&|D]}||vr%|||dfSqd|fS)aOParses any header with the aim to retrieve any credentials from it. Args: header (Optional[str]): The header to parse. prefixes (Optional[Union[List, Tuple, Set]], optional): The prefixes to look for. Defaults to None. Returns: Tuple[Optional[str], Optional[str]]: The prefix and the credentials. )BasicBearerTokenNr)r,listtupleset partitionr@)rQrprefixrrrparse_credentialss r)rbrsr3rf)rArr3r{)r3r)rrr3r)rrr3r)rrr3r)rrrrr3rr5)rQrsrrr3r)0 __future__rrecollections.abcrtypingrrr urllib.parsersanic.exceptionsr sanic.helpersr rrHeaderIterablerrdictrrr_token_quotedcompileASCIIr_ipv6rrr rPrrfrzrrrrrrrrangerrrrrrrsF     "   O 8 "$  & &