3 gYl@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddl mZddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*dcZ+ej,Z-ej.dkr&ddZ/ddZddZ0ddZ1ddddZ2ddZ3ddZ4ddZ5d d!Z6d"d#Z7ded$d%Z8d&d'Z9d(d)Z:d*d+Z;d,d-Zd2d3Z?e@d4d5ZAd6d7ZBd8d9ZCd:d;ZDdd?ZFd@dAZGejHdBdCZIdDdEZJdfdFdGZKdHdIZLdgdKdLZMdMdNZNdOdPZOdQjPdRZQeQdSZReQdTZSdUdVZTdWdXZUdYdZZVejWd[ZXejWd\ZYd]d^ZZd_d`Z[dadbZ\dS)hz requests.utils ~~~~~~~~~~~~~~ This module provides utility functions that are used within Requests that are also useful for external consumption. N) __version__)certs)to_native_string)parse_http_list)quoteurlparsebytesstr OrderedDictunquote getproxies proxy_bypass urlunparse basestring integer_typesis_py3proxy_bypass_environmentgetproxies_environment)cookiejar_from_dict)CaseInsensitiveDict) InvalidURL InvalidHeaderFileModeWarningUnrewindableBodyError.netrc_netrcWindowsc Cstrddl}nddl}y2|j|jd}|j|dd}|j|dd}Wntk r\dSX| sj| rndS|jd}xX|D]P}|dkrd|krd S|jdd }|jd d }|jd d}t j ||t j r~d Sq~WdS)Nrz;Software\Microsoft\Windows\CurrentVersion\Internet SettingsZ ProxyEnableZ ProxyOverrideF;z.Tz\.*z.*?) rwinreg_winregOpenKeyHKEY_CURRENT_USER QueryValueExOSErrorsplitreplacerematchI)hostr"ZinternetSettingsZ proxyEnableZ proxyOverridetestr/?/Users/olari/OneDrive/sandbox/awsBlog2/lambda/requests/utils.pyproxy_bypass_registry-s2        r1cCstrt|St|SdS)zReturn True, if the host should be bypassed. Checks proxy settings gathered from the environment, if specified, or the registry. N)rrr1)r-r/r/r0rNsrcCst|dr|j}|S)z/Returns an internal sequence dictionary update.items)hasattrr2)dr/r/r0dict_to_sequenceZs r5cCs2d}d}t|drt|}nbt|dr.|j}nPt|dr~y |j}Wntjk rZYn$Xtj|j}d|jkr~t j dt t|dry |j }Wn$t tfk r|dk r|}Yn\Xt|do|dkry&|jdd |j }|j|pdWnt tfk rd}YnX|dkr$d}td||S) Nr__len__lenfilenoba%Requests has determined the content-length for this request using the binary size of the file: however, the file has been opened in text mode (i.e. without the 'b' flag in the mode). This may lead to an incorrect content-length. In Requests 3.0, support will be removed for files in text mode.tellseek)r3r7r8ioUnsupportedOperationosfstatst_sizemodewarningswarnrr:r'IOErrorr;max)oZ total_lengthZcurrent_positionr8r/r/r0 super_lencs@             rHFc Csyddlm}m}d}xJtD]B}ytjjdj|}Wntk rJdSXtjj|r|}PqW|dkrndSt |}d}t |t r|j d}|j j|d} y6||j| } | r| drdnd} | | | dfSWn|tfk r|rYnXWnttfk rYnXdS) z;Returns the Requests tuple auth for a given url from netrc.r)netrcNetrcParseErrorNz~/{0}:asciirr<)rIrJ NETRC_FILESr?path expanduserformatKeyErrorexistsr isinstancer decodenetlocr(authenticatorsrE ImportErrorAttributeError) urlZ raise_errorsrIrJ netrc_pathflocriZsplitstrr-rlogin_ir/r/r0get_netrc_auths8     r_cCsBt|dd}|r>t|tr>|ddkr>|ddkr>tjj|SdS)z0Tries to guess the filename of the given object.nameNr)getattrrSrr?rNbasename)objr`r/r/r0guess_filenames  rgcCs.|dkr dSt|ttttfr&tdt|S)aTake an object and test to see if it can be represented as a dictionary. Unless it can not be represented as such, return an OrderedDict, e.g., :: >>> from_key_val_list([('key', 'val')]) OrderedDict([('key', 'val')]) >>> from_key_val_list('string') ValueError: need more than 1 value to unpack >>> from_key_val_list({'key': 'val'}) OrderedDict([('key', 'val')]) :rtype: OrderedDict Nz+cannot encode objects that are not 2-tuples)rSr r boolint ValueErrorr )valuer/r/r0from_key_val_lists rlcCsB|dkr dSt|ttttfr&tdt|tjr:|j}t |S)aTake an object and test to see if it can be represented as a dictionary. If it can be, return a list of tuples, e.g., :: >>> to_key_val_list([('key', 'val')]) [('key', 'val')] >>> to_key_val_list({'key': 'val'}) [('key', 'val')] >>> to_key_val_list('string') ValueError: cannot encode objects that are not 2-tuples. :rtype: list Nz+cannot encode objects that are not 2-tuples) rSr r rhrirj collectionsMappingr2list)rkr/r/r0to_key_val_lists rpcCs\g}xRt|D]F}|dd|ddko4dknrJt|dd}|j|qW|S)aParse lists as described by RFC 2068 Section 2. In particular, parse comma-separated lists where the elements of the list may include quoted-strings. A quoted-string could contain a comma. A non-quoted string could have quotes in the middle. Quotes are removed automatically after parsing. It basically works like :func:`parse_set_header` just that items may appear multiple times and case sensitivity is preserved. The return value is a standard :class:`list`: >>> parse_list_header('token, "quoted value"') ['token', 'quoted value'] To create a header from the :class:`list` again, use the :func:`dump_header` function. :param value: a string with a list header. :return: :class:`list` :rtype: list Nr"rcrc)_parse_list_headerunquote_header_valueappend)rkresultitemr/r/r0parse_list_headers (rwcCs|i}xrt|D]f}d|kr$d||<q|jdd\}}|dd|ddkoVdknrlt|dd}|||<qW|S)a^Parse lists of key, value pairs as described by RFC 2068 Section 2 and convert them into a python dict: >>> d = parse_dict_header('foo="is a fish", bar="as well"') >>> type(d) is dict True >>> sorted(d.items()) [('bar', 'as well'), ('foo', 'is a fish')] If there is no value for a key it will be `None`: >>> parse_dict_header('key_without_value') {'key_without_value': None} To create a header from the :class:`dict` again, use the :func:`dump_header` function. :param value: a string with a dict header. :return: :class:`dict` :rtype: dict =Nrrqrcrc)rrr(rs)rkrurvr`r/r/r0parse_dict_header0s( rycCs^|rZ|d|d kodknrZ|dd }| sF|dddkrZ|jddjddS|S) zUnquotes a header value. (Reversal of :func:`quote_header_value`). This does not use the real unquoting but what browsers are actually using for quoting. :param value: the header value to unquote. :rtype: str rrrqNr<z\\\z\"rcrc)r))rk is_filenamer/r/r0rsSs $ rscCs"i}x|D]}|j||j<q W|S)zReturns a key/value dictionary from a CookieJar. :param cj: CookieJar object to extract cookies from. :rtype: dict )rkr`)cj cookie_dictcookier/r/r0dict_from_cookiejarls rcCs t||S)zReturns a CookieJar from a key/value dictionary. :param cj: CookieJar to insert cookies into. :param cookie_dict: Dict of key/values to insert into CookieJar. :rtype: CookieJar )r)r|r}r/r/r0add_dict_to_cookiejar{srcCsTtjdttjdtjd}tjdtjd}tjd}|j||j||j|S)zlReturns encodings from given content string. :param content: bytestring to extract encodings from. zIn requests 3.0, get_encodings_from_content will be removed. For more information, please see the discussion on issue #2266. (This warning should only appear once.)z!])flagsz+]z$^<\?xml.*?encoding=["\']*(.+?)["\'>])rCrDDeprecationWarningr*compiler,findall)content charset_re pragma_rexml_rer/r/r0get_encodings_from_contents rcCsF|jd}|sdStj|\}}d|kr6|djdSd|krBdSdS)z}Returns encodings from given HTTP Header Dict. :param headers: dictionary to extract encoding from. :rtype: str z content-typeNcharsetz'"textz ISO-8859-1)getcgi parse_headerstrip)headers content_typeparamsr/r/r0get_encoding_from_headerss rccsr|jdkr"x|D] }|VqWdStj|jdd}x |D]}|j|}|r:|Vq:W|jddd}|rn|VdS)zStream decodes a iterator.Nr))errorsT)final)encodingcodecsgetincrementaldecoderrT)iteratorrrvdecoderchunkrvr/r/r0stream_decode_response_unicodes      rccsLd}|dks|dkrt|}x*|t|krF||||V||7}qWdS)z Iterate over slices of a string.rN)r7)string slice_lengthposr/r/r0 iter_slicess rcCsvtjdtg}t|j}|rJy t|j|Stk rH|j|YnXyt|j|ddSt k rp|jSXdS)zReturns the requested content back in unicode. :param r: Response object to get unicode content from. Tried: 1. charset from content-type 2. fall back and replace all unicode characters :rtype: str zIn requests 3.0, get_unicode_from_response will be removed. For more information, please see the discussion on issue #2266. (This warning should only appear once.)r))rN) rCrDrrrr r UnicodeErrorrt TypeError)rtried_encodingsrr/r/r0get_unicode_from_responses   rZ4ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzz0123456789-._~c Cs|jd}xtdt|D]}||dd}t|dkr|jrytt|d}Wn tk rttd|YnX|tkr|||dd||<qd||||<qd||||<qWdj |S) zUn-escape any percent-escape sequences in a URI that are unreserved characters. This leaves all reserved, illegal and non-ASCII bytes encoded. :rtype: str %rrr<z%Invalid percent-escape sequence: '%s'N) r(ranger7isalnumchrrirjrUNRESERVED_SETjoin)uripartsihcr/r/r0unquote_unreserveds rc Cs:d}d}ytt||dStk r4t||dSXdS)zRe-quote the given URI. This function passes the given URI through an unquote/quote cycle to ensure that it is fully and consistently quoted. :rtype: str z!#$%&'()*+,/:;=?@[]~z!#$&'()*+,/:;=?@[]~)safeN)rrr)rsafe_with_percentsafe_without_percentr/r/r0 requote_uri s rcCsltjdtj|d}|jd\}}tjdtjtt|d}tjdtj|d|@}||@||@kS)zThis function allows you to check if an IP belongs to a network subnet Example: returns True if ip = 192.168.1.1 and net = 192.168.1.0/24 returns False if ip = 192.168.1.1 and net = 192.168.100.0/24 :rtype: bool z=Lr/)structunpacksocket inet_atonr(dotted_netmaskri)ipnetipaddrnetaddrbitsnetmasknetworkr/r/r0address_in_network"s rcCs&ddd|>dA}tjtjd|S)zConverts mask from /xx format to xxx.xxx.xxx.xxx Example: if mask is 24 function returns 255.255.255.0 :rtype: str lr z>I)r inet_ntoarpack)maskrr/r/r0r1src Cs*ytj|Wntjk r$dSXdS)z :rtype: bool FT)rrerror) string_ipr/r/r0is_ipv4_address<s rcCs|jddkryt|jdd}Wntk r8dSX|dksJ|dkrNdSytj|jddWqtjk r|dSXndSdS)zV Very simple check of the cidr format in no_proxy variable. :rtype: bool rrFrrT)countrir(rjrrr)string_networkrr/r/r0 is_valid_cidrGsrc csT|dk }|r"tjj|}|tj|<z dVWd|rN|dkrDtj|=n |tj|<XdS)zSet the environment variable 'env_name' to 'value' Save previous value, yield, and then restore the previous value stored in the environment variable 'env_name'. If 'value' is None, do nothingN)r?environr)Zenv_namerkZ value_changed old_valuer/r/r0 set_environ_s    rc Csdd}|}|dkr|d}t|j}|rdd|jddjd D}|jd d }t|rxb|D](}t|r~t||rd Sqb||krbd SqbWn0x.|D]&}|j|s|jd d j|rd SqWtd|2y t |}Wnt t j fk rd }YnXWdQRX|r d Sd S)zL Returns whether we should bypass proxies or not. :rtype: bool cSstjj|ptjj|jS)N)r?rrupper)kr/r/r0{sz'should_bypass_proxies..Nno_proxycss|]}|r|VqdS)Nr/).0r-r/r/r0 sz(should_bypass_proxies.. r,:rTF) rrUr)r(rrrendswithrrrrgaierror) rYr get_proxyZ no_proxy_argrUrproxy_ipr-bypassr/r/r0should_bypass_proxiesus4       rcCst||driStSdS)zA Return a dict of environment proxies. :rtype: dict )rN)rr )rYrr/r/r0get_environ_proxiess rcCsv|pi}t|}|jdkr.|j|j|jdS|jd|j|jd|jdg}d}x|D]}||krX||}PqXW|S)zSelect a proxy for the url, if applicable. :param url: The url being for the request :param proxies: A dictionary of schemes or schemes and hosts to proxy URLs Nallz://zall://)rhostnamerscheme)rYproxiesurlpartsZ proxy_keysproxyZ proxy_keyr/r/r0 select_proxys  rpython-requestscCs d|tfS)zO Return a string representing the default user agent. :rtype: str z%s/%s)r)r`r/r/r0default_user_agentsrcCsttdjddddS)z9 :rtype: requests.structures.CaseInsensitiveDict z, gzipdeflatez*/*z keep-alive)z User-AgentzAccept-EncodingAccept Connection)rr)rrrr/r/r/r0default_headerss rc Csg}d}xtjd|D]}y|jdd\}}Wntk rL|d}}YnXd|jdi}xP|jdD]B}y|jd\}}Wntk rPYnX|j|||j|<qhW|j|qW|S) zReturn a dict of parsed link headers proxies. i.e. Link: ; rel=front; type="image/jpeg",; rel=back;type="image/jpeg" :rtype: list z '"z, * '"rx)r*r(rjrrt) rklinks replace_charsvalrYrlinkparamkeyr/r/r0parse_header_linkss rrLr<cCs|dd}|tjtjfkr dS|ddtjkr6dS|ddtjtjfkrRdS|jt}|dkrhd S|dkr|dddtkrd S|d ddtkrd S|dkr|ddt krd S|d dt krdSdS)z :rtype: str Nzutf-32rz utf-8-sigr<zutf-16rzutf-8z utf-16-berz utf-16-lez utf-32-bez utf-32-le) r BOM_UTF32_LE BOM_UTF32_BEBOM_UTF8 BOM_UTF16_LE BOM_UTF16_BEr_null_null2_null3)datasample nullcountr/r/r0guess_json_utfs*  rcCs8t||\}}}}}}|s$||}}t||||||fS)zGiven a URL that may or may not have a scheme, prepend the given scheme. Does not replace a present scheme with the one provided as an argument. :rtype: str )rr)rY new_schemerrUrNrqueryfragmentr/r/r0prepend_scheme_if_needed0s r c CsBt|}yt|jt|jf}Wnttfk r<d}YnX|S)z{Given a url with authentication components, extract them into a tuple of username,password. :rtype: (str,str) r)rr)rr usernamepasswordrXr)rYparsedauthr/r/r0get_auth_from_urlAs  rs^\S[^\r\n]*$|^$z^\S[^\r\n]*$|^$c Csf|\}}t|trt}nt}y|j|s4td|Wn*tk r`td||t|fYnXdS)zVerifies that header value is a string which doesn't contain leading whitespace or return characters. This prevents unintended header injection. :param header: tuple, in the format (name, value). z7Invalid return character or leading space in header: %sz>Value for header {%s: %s} must be of type str or bytes, not %sN)rSr _CLEAN_HEADER_REGEX_BYTE_CLEAN_HEADER_REGEX_STRr+rrtype)headerr`rkpatr/r/r0check_header_validityVs  rcCsFt|\}}}}}}|s"||}}|jddd}t|||||dfS)zW Given a url remove the fragment and the authentication part. :rtype: str @rrrc)rrsplitr)rYrrUrNrr r r/r/r0 urldefragauthks  rc Cs`t|jdd}|dk rTt|jtrTy||jWq\ttfk rPtdYq\XntddS)zfMove file pointer back to its recorded starting position so it can be read again on redirect. r;Nz;An error occurred when rewinding request body for redirect.z+Unable to rewind request body for redirect.)rdbodyrSZ_body_positionrrEr'r)prepared_request body_seekr/r/r0 rewind_body|sr)rr)F)F)N)r)]__doc__rrrm contextlibr=r?platformr*rrrCrrrZ_internal_utilsrcompatrrrrrr r r r r rrrrrrrcookiesr structuresr exceptionsrrrrrMwhereDEFAULT_CA_BUNDLE_PATHsystemr1r5rHr_rgrlrprwryrsrrrrrrr frozensetrrrrrrrcontextmanagerrrrrrrrencoderrrrr rrrrrrrr/r/r/r0 s    @  !  = 3 #   %   9   "