i0|ddlmZddlmZddlmZddlmZddlm Z GddZ Gdd Z Gd d Z y ) ) annotations)Callable)Any)command_builder)WebSocketConnectionc*eZdZdZddZeddZy) NetworkEventzRepresents a network event.c  ||_||_yN) event_classparams)selfr kwargss b/var/www/html/BatchJob/venv/lib/python3.12/site-packages/selenium/webdriver/common/bidi/network.py__init__zNetworkEvent.__init__s& c6|dd|jddi|S)Nr )get)clsjsons r from_jsonzNetworkEvent.from_json"s Ctxx r:CdCCrN)r strrrreturnNone)rdict[str, Any]rr )__name__ __module__ __qualname____doc__r classmethodrrrrr r s %DDrr ceZdZddddddddZd d d d Zdd Z d ddZdddZddZ d ddZ ddZ d dZ d!dZ d"dZ y)#Networkznetwork.beforeRequestSentznetwork.responseStartedznetwork.responseCompletedznetwork.authRequiredznetwork.fetchErrornetwork.continueRequestnetwork.continueWithAuth)before_requestresponse_startedresponse_completed auth_required fetch_errorcontinue_request continue_authbeforeRequestSentresponseStarted authRequired)r'r(r*c<||_g|_i|_i|_yr )conn intercepts callbacks subscriptions)rr2s rrzNetwork.__init__8s %'/135rNc|g}i}|||d<|||d<t|dkDr||d<ndg|d<td|}|jj|}|jj |d|S)aQAdd an intercept to the network. Args: phases: A list of phases to intercept. Default is None (empty list). contexts: A list of contexts to intercept. Default is None. url_patterns: A list of URL patterns to intercept. Default is None. Returns: str: intercept id contexts urlPatternsrphasesr.znetwork.addIntercept intercept)lenrr2executer3append)rr9r7 url_patternsr cmdresults r_add_interceptzNetwork._add_intercept>s >F  !)F:   #$0F= ! v;?%F8  34F8 4f=!%!2!23!7 vk23 rc|d|jj}|D]D}|jjt dd|i|jj |Fy |jjt dd|i|jj |y#t $r}t d|d}~wwxYw)aRemove a specific intercept, or all intercepts. Args: intercept: The intercept to remove. Default is None. Raises: ValueError: If intercept is not found. Note: If intercept is None, all intercepts will be removed. Nznetwork.removeInterceptr:z Exception: )r3copyr2r<rremove Exception)rr:intercepts_to_remove intercept_ides r_remove_interceptzNetwork._remove_intercept_s  #'??#7#7#9 4 5  !!/2Kk[gMh"ij&&|4 5 3 !!/2Kk[dMe"fg&&y1 3+aS 122 3s(AB++ C4CCct|}dfd }jj||}|jvr j|j ||S|gj|<|S)a%Set a callback function to subscribe to a network event. Args: event_name: The event to subscribe to. callback: The callback function to execute on event. Takes Request object as argument. Returns: int: callback id ct|jdjdd|jdjdd|jdjdd|jdjdd|jdjdd|jdjdd|jdjdd|jdjdd }|y) NrequestbodySizecookieszgoog:resourceTypeheaders headersSizetimingsurl) network request_id body_sizerN resource_typerO headers_sizerQrR)Requestr r) event_datarLcallbackrs r _callbackz&Network._on_request.._callbacks%,,Y7;;ItL$++I6:::tL")))488DI(// :>>?RTXY")))488DI'..y9==mTR")))488DI%%i044UDA G W r)rYr rr)r r2 add_callbackr4r=)r event_namerZeventr[ callback_ids` ` r _on_requestzNetwork._on_requestwsoZ(  9911%C  ' NN: & - -k :+6DNN: &rc |j|}|j|}|j |g||}|j ||}||j vr|j |j|n=i} |g| d<|jjtd| |g|j |<|d|j|<|S#t$rtd|dwxYw)aAdd a request handler to the network. Args: event: The event to subscribe to. callback: The callback function to execute on request interception. Takes Request object as argument. url_patterns: A list of URL patterns to intercept. Default is None. contexts: A list of contexts to intercept. Default is None. Returns: int: callback id Event not found)r9r>r7eventszsession.subscriber:) EVENTSPHASESKeyErrorrErAr`r5r=r2r<rr4) rr^rZr>r7r] phase_namer@r_r s radd_request_handlerzNetwork.add_request_handlers& 8U+JU+J$$ZL|^f$g&&z8< ++ +   z * 1 1+ >%'F *|F8  II  o.A6J K.9]D  z *&,[&9{# 8fUG:67 7 8s CCc |j|}t|}|jj |||j |j||j|=|j|j|t|j|dk(r;i}|g|d<|jjtd||j|=yy#t$rtd|dwxYw)zRemove a request handler from the network. Args: event: The event to unsubscribe from. callback_id: The callback id to remove. rbrcrrdsession.unsubscribeN) rergrEr r2remove_callbackrIr4r5rDr;r<r)rr^r_r] net_eventr s rremove_request_handlerzNetwork.remove_request_handlers 8U+J!,  !!)[9 t~~k:; NN; ' :&--k: t!!*- .! 3%'F *|F8  II  o.CVL M"":. 4 8fUG:67 7 8s CC1cZ|jD]}t|}|j|D]I}|jj|||j |j ||j |=Ki}|g|d<|jj td|i|_y)z,Clear all request handlers from the network.rdrkN)r5r r2rlrIr4r<r)rr]rmr_r s rclear_request_handlerszNetwork.clear_request_handlerss,, NJ$Z0I#11*= 0  )))[A&&t~~k'BCNN;/ 0&(F *|F8  II  o.CVL M N rc<d}dfd }|j||S)zAdd an authentication handler to the network. Args: username: The username to authenticate with. password: The password to authenticate with. Returns: int: callback id r*c*|jyr )_continue_with_auth)rLpasswordusernames rr[z+Network.add_auth_handler.._callbacks  ' '( ;r)rLrXrr)ri)rrurtr^r[s `` radd_auth_handlerzNetwork.add_auth_handlers$  <''y99rc,d}|j||y)z~Remove an authentication handler from the network. Args: callback_id: The callback id to remove. r*N)rn)rr_r^s rremove_auth_handlerzNetwork.remove_auth_handlers   ##E;7r)r2rrr)NNN)r9list[str] | Noner7ryr>list[Any] | Nonerrr )r: str | Nonerr)r]rrZCallable[[Request], Any]rintNN) r^rrZr|r>rzr7ryrr})r^rr_r}rrrr)rurrtrrr})r_r}rr)rrr rerfrrArIr`rirnrprvrxrrrr$r$'s559/+53F.-'F 6$(%))-  #'   B30"P*.%) %%+%' % # % %N/0  :"8rr$ceZdZdZ d ddZd dZ d d dZd d dZy)rXz*Represents an intercepted network request.Nc ||_||_||_||_||_||_||_||_| |_| |_ yr ) rSrTrUrNrVrOrWmethodrQrR) rrSrTrUrNrVrOrWrrQrRs rrzRequest.__init__sL $" * (  rc|js tdd|ji}|jjj t d|y)zFail this request.Request not found.rLznetwork.failRequestNrT ValueErrorrSr2r<r)rr s r fail_requestzRequest.fail_requestsC12 2"+T__!= !!/2G"PQrc|js tdd|ji}|||d<|||d<|||d<|||d<|||d<|jjj t d |y) z)Continue after intercepting this request.rrLNbodyrrOrNrRr%r)rrrrOrNrRr s rr,zRequest.continue_request%s12 2"+T__!=  !F6N  %F8    'F9    'F9  ?F5M !!/2KV"TUrci}|j|d<|r|sd|d<nd|d<d||d|d<|jjjt d|y ) a Continue with authentication. Args: username: The username to authenticate with. password: The password to authenticate with. Note: If username or password is None, it attempts auth with no credentials. rLdefaultactionprovideCredentialsrt)typerurt credentialsr&N)rTrSr2r<r)rrurtr s rrszRequest._continue_with_auth?sb"$ OOyx(F8 3F8 -7X[c$dF= ! !!/2Lf"UVr)NNNNNNNN)rSr$rTrrU int | NonerNrrVr{rOrrWrrr{rQrrRr{rrr)NNNNN) rrrr{rOrrNrrRr{rrr~)rur{rtr{rr)rrr r!rrr,rsrrrrXrXs4 !%$(#'!    " ! 0R! VVV V  V  V V4WrrXN) __future__rcollections.abcrtypingr%selenium.webdriver.common.bidi.commonr.selenium.webdriver.remote.websocket_connectionrr r$rXrrrrs=$#$AN D DX8X8vPWPWr