o M,h/.@s:ddlmZGdddZGdddZGdddZdS) )command_builderc@s$eZdZdZddZeddZdS) NetworkEventzRepresents a network event.cKs||_||_dSN) event_classparams)selfrkwargsr r/var/www/www-root/data/www/bot.pdev.uz/venv/lib/python3.10/site-packages/selenium/webdriver/common/bidi/network.py__init__s zNetworkEvent.__init__cCs|dd|di|S)Nrr )get)clsjsonr r r from_jsonszNetworkEvent.from_jsonN)__name__ __module__ __qualname____doc__r classmethodrr r r r rs rc@seZdZddddddddZd d d d Zd dZgddfddZd ddZddZd!ddZ ddZ ddZ ddZ ddZ dS)"Networkznetwork.beforeRequestSentznetwork.responseStartedznetwork.responseCompletedznetwork.authRequiredznetwork.fetchErrornetwork.continueRequestnetwork.continueWithAuth)before_requestresponse_startedresponse_completed auth_required fetch_errorcontinue_request continue_authbeforeRequestSentresponseStarted authRequired)rrrcCs||_g|_i|_i|_dSr)conn intercepts callbacks subscriptions)rr"r r r r 2s zNetwork.__init__NcCsni}|dur ||d<|dur||d<t|dkr||d<ndg|d<td|}|j|}|j|d|S) aAdd an intercept to the network. Parameters: ---------- phases (list, optional): A list of phases to intercept. Default is empty list. contexts (list, optional): A list of contexts to intercept. Default is None. url_patterns (list, optional): A list of URL patterns to intercept. Default is None. Returns: ------- str : intercept id Ncontexts urlPatternsrphasesrznetwork.addIntercept intercept)lenrr"executer#append)rr(r& url_patternsrcmdresultr r r _add_intercept8s     zNetwork._add_interceptc Cs|dur!|j}|D]}|jtdd|i|j|q dSz|jtdd|i|j|WdStyH}ztd|d}~ww)asRemove a specific intercept, or all intercepts. Parameters: ---------- intercept (str, optional): The intercept to remove. Default is None. Raises: ------ Exception: If intercept is not found. Notes: ----- If intercept is None, all intercepts will be removed. Nznetwork.removeInterceptr)z Exception: )r#copyr"r+rremove Exception)rr)intercepts_to_remove intercept_ider r r _remove_interceptWs zNetwork._remove_interceptcsRt|}fdd}j||}|jvr!j|||S|gj|<|S)a`Set a callback function to subscribe to a network event. Parameters: ---------- event_name (str): The event to subscribe to. callback (function): The callback function to execute on event. Takes Request object as argument. Returns: ------- int : callback id c st|jddd|jddd|jddd|jddd|jddd|jddd|jddd|jdddd }|dS) NrequestbodySizecookieszgoog:resourceTypeheaders headersSizetimingsurl) network request_id body_sizer: resource_typer; headers_sizer=r>)Requestrr ) event_datar8callbackrr r _callbacks z&Network._on_request.._callback)rr" add_callbackr$r,)r event_namerGeventrH callback_idr rFr _on_requestss  zNetwork._on_requestc Csz |j|}|j|}Wntytd|dw|j|g||d}|||}||jvr9|j||ni} |g| d<|j t d| |g|j|<|d|j |<|S)a0Add a request handler to the network. Parameters: ---------- event (str): The event to subscribe to. url_patterns (list, optional): A list of URL patterns to intercept. Default is None. contexts (list, optional): A list of contexts to intercept. Default is None. callback (function): The callback function to execute on request interception Takes Request object as argument. Returns: ------- int : callback id Event not found)r(r-r&eventszsession.subscriber)) EVENTSPHASESKeyErrorr3r0rMr%r,r"r+rr$) rrKrGr-r&rJ phase_namer/rLrr r r add_request_handlers       zNetwork.add_request_handlercCsz|j|}Wntytd|dwt|}|j||||j||j|=|j| |t |j|dkrUi}|g|d<|j t d||j|=dSdS)zRemove a request handler from the network. Parameters: ---------- event_name (str): The event to unsubscribe from. callback_id (int): The callback id to remove. rNrOrrPsession.unsubscribeN) rQrSr3rr"remove_callbackr7r$r%r2r*r+r)rrKrLrJ net_eventrr r r remove_request_handlers    zNetwork.remove_request_handlercCst|jD]1}t|}|j|D]}|j||||j||j|=qi}|g|d<|jtd|qi|_dS)z,Clear all request handlers from the network.rPrVN)r%rr"rWr7r$r+r)rrJrXrLrr r r clear_request_handlerss    zNetwork.clear_request_handlerscsd}fdd}|||S)aAdd an authentication handler to the network. Parameters: ---------- username (str): The username to authenticate with. password (str): The password to authenticate with. Returns: ------- int : callback id rcs|dSr)_continue_with_auth)r8passwordusernamer r rHsz+Network.add_auth_handler.._callback)rU)rr^r]rKrHr r\r add_auth_handlers  zNetwork.add_auth_handlercCsd}|||dS)zRemove an authentication handler from the network. Parameters: ---------- callback_id (int): The callback id to remove. rN)rY)rrLrKr r r remove_auth_handlerszNetwork.remove_auth_handlerrNN)rrrrQrRr r0r7rMrUrYrZr_r`r r r r r!s,   '& rc@sLeZdZdZ        d defddZddZd dd Zdd d ZdS)rDz*Represents an intercepted network request.Nr?c Cs@||_||_||_||_||_||_||_||_| |_| |_ dSr) r?r@rAr:rBr;rCmethodr=r>) rr?r@rAr:rBr;rCrbr=r>r r r r s  zRequest.__init__cCs0|jstdd|ji}|jjtd|dS)zFail this request.Request not found.r8znetwork.failRequestNr@ ValueErrorr?r"r+r)rrr r r fail_request s zRequest.fail_requestcCs|jstdd|ji}|dur||d<|dur||d<|dur$||d<|dur,||d<|dur4||d<|jjtd |dS) z)Continue after intercepting this request.rcr8Nbodyrbr;r:r>rrd)rrgrbr;r:r>rr r r r)s zRequest.continue_requestcCsPi}|j|d<|r |sd|d<n d|d<d||d|d<|jjtd|d S) ayContinue with authentication. Parameters: ---------- request (Request): The request to continue with. username (str): The username to authenticate with. password (str): The password to authenticate with. Notes: ----- If username or password is None, it attempts auth with no credentials r8defaultactionprovideCredentialsr])typer^r] credentialsrN)r@r?r"r+r)rr^r]rr r r r[=s  zRequest._continue_with_auth)NNNNNNNN)NNNNNra) rrrrrr rfrr[r r r r rDs   rDN)%selenium.webdriver.common.bidi.commonrrrrDr r r r s  e