\h-4bddlZ ddlmZmZddlZGddeZy#e$r ddlmZmZYwxYw)N) urlencodeunquoteceZdZd#dZdZdZdZdZdZe dZ e d Z d$d Z d#d Z d Zd ZdZdZdZdZddddZdZdddddZdZd$dZdZdZd Zd!Zd"Zy)%ClientNc ||ni}||ni}d|z|_|rdtt|z}nd}d|vr |d|_n`t j d|jz|z}|j jd} tj|}|d |_|jjd } | d d dd zt|jz|_ y#t$r}td|dd zd}~wwxYw)aY Initialises a new Client object :param url: This is where the BrowserMob Proxy lives :param params: URL query (for example httpProxy and httpsProxy vars) :param options: Dictionary that can contain the port of an existing proxy to use (for example 'existing_proxy_port_to_use') Nzhttp://?existing_proxy_port_to_use%s/proxyzutf-8zOCould not read Browsermob-Proxy json Another server running on this port? %s...iport:)hostrrr requestspostcontentdecodejsonloads Exceptionsplitstrproxy) selfurlparamsoptions urlparamsresprjcontente url_partss R/var/www/html/trade_iq/venv/lib/python3.12/site-packages/browsermobproxy/client.py__init__zClient.__init__ s"-2$0'bO gi&788II '7 2 <=DI==dii!7)!CDDll))'2G _::g.!(DIIIOOC( q\!"%+c$))n<  _!NPWX\Y\P]!^__ _sC$$ D-C>>Dcvtj|jd|j}|jS)z: shuts down the proxy and closes the port /proxy/rdeleterr status_coderrs r$closez Client.close*s) OOTYY B C}}c^ddlm}|j|j|jdS)Y Returns a Selenium WebDriver Proxy class with details of the HTTP Proxy r) webdriver) httpProxysslProxy)seleniumr1Proxyr)rr1s r$selenium_proxyzClient.selenium_proxy3s+ '    r.c"|jS)r0)r6)rs r$webdriver_proxyzClient.webdriver_proxy=s""$$r.c>d|j|jd|d<y)z Adds an 'proxy' entry to a desired capabilities dictionary with the BrowserMob proxy information :param capabilities: The Desired capabilities object from Selenium WebDriver MANUAL) proxyTyper2r3rN)rr capabilitiess r$add_to_capabilitieszClient.add_to_capabilitiesDs!" ! Wr.c&|j|y)N)r>r<s r$add_to_webdriver_capabilitiesz$Client.add_to_webdriver_capabilitiesRs   .r.ctjd|jzj}|dDcgc]}|d }}|Scc}w)z8 Return a list of proxy ports available r proxyListr )rgetrr)rr,r portss r$ proxy_portszClient.proxy_portsVsI LLdii/ 0 5 5 7*+K.9$f99 :s Actj|jd|jd}|j S)z5 Gets the HAR that has been recorded r'/har)rrCrr rr+s r$harz Client.haras* LLdiiC Dvvxr.c@||ni}|d|ini}||jd|i|r|j|tj|jd|jd|}|j dk(r|j |j fS|j dfS)a  This sets a new HAR to be recorded :param str ref: A reference for the HAR. Defaults to None :param dict options: A dictionary that will be passed to BrowserMob Proxy with specific keywords. Keywords are: - captureHeaders: Boolean, capture headers - captureContent: Boolean, capture content bodies - captureBinaryContent: Boolean, capture binary content :param str title: the title of first har page. Defaults to ref. NinitialPageRefinitialPageTitler'rG)updaterputrr r*r)rrefrtitlepayloadr,s r$new_harzClient.new_harjs%0'b-0_#S)"   NN.6 7  NN7 # LLdiiCW M ==C MM1668, ,MM4( (r.c|d|ini}||jd|itj|jd|jd|}|j S)z This sets a new page to be recorded :param str ref: A reference for the new page. Defaults to None :param str title: the title of new har page. Defaults to ref. pageRef pageTitler'z /har/pageRef)rMrrNrr r*)rrOrPrQr,s r$new_pagezClient.new_pagesU'*o9c"2   NNK/ 0 LLdiiK  "}}r.ctj|jd|jd||d}|jS)z Sets a list of URL patterns to blacklist :param str regex: a comma separated list of regular expressions :param int status_code: the HTTP status code to return for URLs that do not match the blacklist r'z /blacklistregexstatusrrNrr r*rregexpr*r,s r$ blacklistzClient.blacklist4 LLDIItyyI#)[A C}}r.ctj|jd|jd||d}|jS)z Sets a list of URL patterns to whitelist :param str regex: a comma separated list of regular expressions :param int status_code: the HTTP status code to return for URLs that do not match the whitelist r'z /whitelistrXr[r\s r$ whitelistzClient.whitelistr_r.ctj|jd|jd|t j ||dddi}|j S)a This add automatic basic authentication :param str domain: domain to set authentication credentials for :param str username: valid username to use when authenticating :param str password: valid password to use when authenticating r'z /auth/basic/)usernamepassword content-typeapplication/jsonrdataheadersrrrr rdumpsr*)rdomainrcrdr,s r$basic_authenticationzClient.basic_authenticationsK MMTYY SYZ#zzxX*VW#13E"F H}}r.ct|ts tdtj|j d|j dtj|ddi}|jS)z This sets the headers that will set by the proxy on all requests :param dict headers: this is a dictionary of the headers to be set zheaders needs to be dictionaryr'z/headersrerfrg) isinstancedict TypeErrorrrrr rrkr*)rrir,s r$rizClient.headerssY '4(<= = MMtyy$))L#zz'2#13E"F H}}r.ctj|jd|jd|ddi}|jS) Executes the java/js code against each response `HttpRequest request `_, `HttpMessageContents contents `_, `HttpMessageInfo messageInfo `_ are available objects to interact with. :param str js: the js/java code to execute r'z/filter/responsere text/plainrgrrrr r*rjsr,s r$response_interceptorzClient.response_interceptors9 MMtyy$))T!#1<"@ B}}r.ctj|jd|jd|ddi}|jS)rsr'z/filter/requestrertrgrurvs r$request_interceptorzClient.request_interceptors9 MMdiiS!#1<"@ B}}r. upstreamKbpsdownstreamKbpslatency) upstream_kbpsdownstream_kbpsr}ci}t|jD]<\}}||jvrtd|zt |||j|<>t t|jdk(r tdt j|jd|jd|}|jS)a4 Limit the bandwidth through the proxy. :param dict options: A dictionary with all the details you want to set. downstream_kbps - Sets the downstream kbps upstream_kbps - Sets the upstream kbps latency - Add the given latency to each HTTP request invalid key: %sr)You need to specify one of the valid Keysr'z/limit) listitemsLIMITSKeyErrorintlenrrNrr r*rrrkvr,s r$limitsz Client.limitss7==?+ ,FQ #01455%(VF4;;q> "  , tFLLN# $ )FG G LL 499E !}}r.requestTimeout readTimeoutconnectionTimeoutdnsCacheTimeout)requestread connectiondnsci}t|jD]<\}}||jvrtd|zt |||j|<>t t|jdk(r tdt j|jd|jd|}|jS)aU Configure various timeouts in the proxy :param dict options: A dictionary with all the details you want to set. request - request timeout (in seconds) read - read timeout (in seconds) connection - connection timeout (in seconds) dns - dns lookup timeout (in seconds) rrrr'z/timeout) rrTIMEOUTSrrrrrNrr r*rs r$timeoutszClient.timeoutss7==?+ .FQ %01455'*1vF4==# $  . tFLLN# $ )FG G LL$))TYYG !}}r.c||ni}|||||<tj|jd|jdt j |ddi}|j S)a  Remap the hosts for a specific URL :param str address: url that you wish to remap :param str ip_address: IP Address that will handle all traffic for the address passed in :param **hostmap: Other hosts to be added as keyword arguments r'z/hostsrerf)rirj)raddress ip_addresshostmapr,s r$ remap_hostszClient.remap_hosts!sd%0'b  J$:)GG  MMDIIF**W-#13E"F H}}r.ctj|jd|jd||d}|jS)z Waits for the network to be quiet :param int quiet_period: number of milliseconds the network needs to be quiet for :param int timeout: max number of milliseconds to wait r'z/wait)quietPeriodInMs timeoutInMsr[)r quiet_periodtimeoutr,s r$wait_for_traffic_to_stopzClient.wait_for_traffic_to_stop3s4 LLtyy$))D-9'R T}}r.cxtj|jd|jd}|jS)zI Clears the DNS cache associated with the proxy instance r'z /dns/cacher(r+s r$clear_dns_cachezClient.clear_dns_cache?s) OOtyy$))L M}}r.c||d}tj|jd|jd|}|jS)z Rewrites the requested url. :param match: a regex to match requests with :param replace: unicode a string to replace the matches with ) matchRegexreplacer'/rewriter[)rmatchrrr,s r$ rewrite_urlzClient.rewrite_urlFs=   LL$))TYYG !}}r.cxtj|jd|jd}|jS)zK Clears all URL rewrite rules :return: status code r'rr(r+s r$clear_all_rewrite_url_rulesz"Client.clear_all_rewrite_url_rulesVs) OOTYY J K}}r.c~tj|jd|jdd|i}|jS)z Retries. No idea what its used for, but its in the API... :param int retry_count: the number of retries r'z/retry retrycountr[)r retry_countr,s r$retryz Client.retry_s3 LL 499E& 4 6}}r.)NN)NNN)__name__ __module__ __qualname__r%r-r6r8r>r@propertyrErHrRrVr^rarmrirxrzrrrrrrrrrrr.r$rr s=> %  /)8       (+F 2$) H2$  r.r) r urllib.parserr ImportErrorurllibrobjectrrr.r$rs:*/ ]V] *))*s  ..