DgddlZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZej eZdZdZd Zd Zd Zd Zd ZdZde ee fde eeffdZddZddZdZdZ de!dede!fdZ"y)N) namedtuple)Path)Dict NamedTuple) _parse_proxy)encodingca.crtzca.keyzseleniumwire-ca.pemmode upstream_authupstream_custom_authno_proxycZ|xsijdi}tjjd}tjjd}tjjd}i}|r||d<|r||d<|r||d<|j ||jd}t |t r0|jdDcgc]}|jc}|d<td d }d D](}|j||t||||<*|Scc}w) aGet the upstream proxy configuration from the options dictionary. This will be overridden with any configuration found in the environment variables HTTP_PROXY, HTTPS_PROXY, NO_PROXY The configuration will be returned as a dictionary with keys 'http', 'https' and 'no_proxy'. The value of the 'http' and 'https' keys will be a named tuple with the attributes: scheme, username, password, hostport The value of 'no_proxy' will be a list. Note that the keys will only be present in the dictionary when relevant proxy configuration exists. Args: options: The selenium wire options. Returns: A dictionary. proxy HTTP_PROXY HTTPS_PROXYNO_PROXYhttphttpsr , ProxyConfz!scheme username password hostport)rr) poposenvirongetupdate isinstancestrsplitstriprr) options proxy_options http_proxy https_proxyr mergedhconf proxy_types R/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/seleniumwire/utils.pyget_upstream_proxyr)s$]''4M -J**../Kzz~~j)H F#v%w%z MM- zz*%H(C 191DEAaggiEz k#F GD'I  ::j ! -!%|F:4F'G!HF:  I MFsD( proxy_configreturnc|jd}|jd}d}|r)|r'|j|jk7r td|}n |r|}n|r|}i}|rs|\}}}}dj|||t<|rdj|||t <|jd} | r | |t <|jd} | r | |t<|S) zBuild the arguments needed to pass an upstream proxy to mitmproxy. Args: proxy_config: The proxy config parsed out of the Selenium Wire options. Returns: A dictionary of arguments suitable for passing to mitmproxy. rrNzADifferent settings for http and https proxy servers not supportedzupstream:{}://{}z{}:{}custom_authorizationr )rhostport ValueErrorformat MITM_MODEMITM_UPSTREAM_AUTHMITM_UPSTREAM_CUSTOM_AUTH MITM_NO_PROXY) r*r"r#r&argsschemeusernamepasswordr. custom_authr s r(build_proxy_argsr:Js!!&)J""7+K Dk   +"6"6 6`a a   D /3,(H,33FHEY '.~~h'ID# $"&&'=> .9D* +##J/ "*D  Kc tjt|}tt t j |d5}|j|dddtjdj|y#1swY.xYw#t$r'tjdj|YywxYw)z?Extracts the root certificate to the current working directory.wbNz5{} extracted. You can now import this into a browser.zInvalid certificate '{}') pkgutilget_data __package__openrrgetcwdwriteloginfor0FileNotFoundErrorerror) cert_namecertouts r( extract_certrKws\ Y7$ryy{I. 5  IIdO  HOOPYZ[   @ ,33I>?@sBBB -B?>B?ctj|dt|t}|r|j ry|5|3t|j }t|j }nK|| t dtjtt}tjtt}t|d5}|j|dz|zdddy#1swYyxYw)aExtracts the root certificate and key and combines them into a single file called seleniumwire-ca.pem in the specified destination folder. Args: dest_folder: The destination folder that the combined certificate and key will be written to. cert_path: Optional path to the root certificate. When not supplied selenium wire's own root certificate will be used. key_path: Optional path to the private key. When not supplied selenium wire's own private key will be used. Note that the key must always be supplied when a certificate is supplied. check_exists: If True the combined file will not be overwritten if it already exists in the destination folder. T)exist_okNz+A certificate and key must both be suppliedr= )rmakedirsr COMBINED_CERTexists read_bytesr/r>r?r@ ROOT_CERTROOT_KEYrArC) dest_folder cert_pathkey_path check_exists combined_path root_certroot_keyf_outs r(extract_cert_and_keyr]s KK d+m4M ,,.!5O..0 >,,.  ("6FGG$$[)< ##K: mT "2e I%01222s C**C3cpt|tjjxrt|t S)N)r collectionsabcSequencer) containers r( is_list_alikercs) i!9!9 : ]:iY\C]?]]r;c$|^}}}|rd|d}||fS)zMake an address safe to use in a URL. Args: address: A tuple of address information. Returns: A 2-tuple of url-safe (address, port) [])addressaddrportrests r(urlsafe_addressrls* D$ 4&{ :r;datarc.tj||S)aAttempt to decode data based on the supplied encoding. If decoding fails a ValueError is raised. Args: data: The encoded data. encoding: The encoding type. Returns: The decoded data. Raises: ValueError if the data could not be decoded. )decoderdecode)rmrs r(rprps >>$ ))r;)r )NNT)#collections.abcr_loggingrr>rpathlibrtypingrrurllib.requestr*seleniumwire.thirdparty.mitmproxy.net.httprro getLogger__name__rDrSrTrPr1r2r3r4r)rr:rKr]rcrlbytesrprgr;r(rzs "#'Jg!  %  $2 /d*4Z#8*T#s(^*Z \2D^" * *# *% *r;