o 衡i0@sjddlmZddlmZddlmZddlmZddlm Z GdddZ Gdd d Z Gd d d Z d S) ) annotations)Callable)Any)command_builder)WebSocketConnectionc@s(eZdZdZddd Zedd d ZdS) NetworkEventzRepresents a network event. event_classstrkwargsrreturnNonecKs||_||_dSN)rparams)selfrr rw/var/www/html/asbeauty/laura_geller_scraping/env/lib/python3.10/site-packages/selenium/webdriver/common/bidi/network.py__init__s zNetworkEvent.__init__jsondict[str, Any]cCs|dd|ddi|S)Nrr)get)clsrrrr from_json"szNetworkEvent.from_jsonN)rr r rr r )rrr r)__name__ __module__ __qualname____doc__r classmethodrrrrrrs  rc@seZdZddddddddZd d d d Zd5ddZ   d6d7ddZd8d9ddZd:d%d&Z  d;dd-d.Z d?d1d2Z d@d3d4Z dS)ANetworkznetwork.beforeRequestSentznetwork.responseStartedznetwork.responseCompletedznetwork.authRequiredznetwork.fetchErrornetwork.continueRequestnetwork.continueWithAuth)before_requestresponse_startedresponse_completed auth_required fetch_errorcontinue_request continue_authbeforeRequestSentresponseStarted authRequired)r!r"r$connrr r cCs||_g|_i|_i|_dSr )r+ intercepts callbacks subscriptions)rr+rrrr8s zNetwork.__init__Nphaseslist[str] | Nonecontexts url_patternslist[Any] | NonercCsz|durg}i}|dur||d<|dur||d<t|dkr#||d<ndg|d<td|}|j|}|j|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 Nr1 urlPatternsrr/r(znetwork.addIntercept intercept)lenrr+executer,append)rr/r1r2rcmdresultrrr_add_intercept>s     zNetwork._add_interceptr5 str | Nonec Cs|dur!|j}|D]}|jtdd|i|j|q dSz|jtdd|i|j|WdStyH}ztd|d}~ww)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.removeInterceptr5z Exception: )r,copyr+r7rremove Exception)rr5intercepts_to_remove intercept_iderrr_remove_intercept_s zNetwork._remove_intercept event_namer callbackCallable[[Request], Any]intcsTt|}dfdd }j||}|jvr"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 event_datarr r c st|jddd|jddd|jddd|jddd|jddd|jddd|jddd|jdddd }|dS) NrequestbodySizecookieszgoog:resourceTypeheaders headersSizetimingsurl) network request_id body_sizerK resource_typerL headers_sizerNrO)Requestrr)rHrIrErrr _callbacks z&Network._on_request.._callbackN)rHrr r )rr+ add_callbackr-r8)rrDrEeventrW callback_idrrVr _on_requestws   zNetwork._on_requestrYc Csz |j|}|j|}Wntytd|dw|j|g||d}|||}||jvr9|j||ni} |g| d<|j t d| |g|j|<|d|j |<|S)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)r/r2r1eventszsession.subscriber5) EVENTSPHASESKeyErrorr?r;r[r.r8r+r7rr-) rrYrEr2r1rD phase_namer:rZrrrradd_request_handlers       zNetwork.add_request_handlerrZcCsz|j|}Wntytd|dwt|}|j||||j||j|=|j| |t |j|dkrUi}|g|d<|j t d||j|=dSdS)zRemove a request handler from the network. Args: event: The event to unsubscribe from. callback_id: The callback id to remove. r\r]rr^session.unsubscribeN) r_rar?rr+remove_callbackrCr-r.r>r6r7r)rrYrZrD net_eventrrrrremove_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.r^rdN)r.rr+rerCr-r7r)rrDrfrZrrrrclear_request_handlerss    zNetwork.clear_request_handlersusernamepasswordcs d}d fdd }|||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$rIrUr r cs|dSr )_continue_with_auth)rIrjrirrrWsz+Network.add_auth_handler.._callbackN)rIrUr r )rc)rrirjrYrWrrlradd_auth_handlers  zNetwork.add_auth_handlercCsd}|||dS)z~Remove an authentication handler from the network. Args: callback_id: The callback id to remove. r$N)rg)rrZrYrrrremove_auth_handlerszNetwork.remove_auth_handler)r+rr r )NNN)r/r0r1r0r2r3r rr )r5r<r r )rDr rErFr rGNN) rYr rErFr2r3r1r0r rG)rYr rZrGr r r r )rir rjr r rG)rZrGr r )rrrr_r`rr;rCr[rcrgrhrmrnrrrrr's6   ! ( '  rc@sXeZdZdZ        ddddZd ddZ     d!d"ddZd#d$ddZdS)%rUz*Represents an intercepted network request.NrPrrQrrR int | NonerKrSr<rLrTmethodrNrOr r c Cs@||_||_||_||_||_||_||_||_| |_| |_ dSr ) rPrQrRrKrSrLrTrrrNrO) rrPrQrRrKrSrLrTrrrNrOrrrrs  zRequest.__init__cCs0|jstdd|ji}|jjtd|dS)zFail this request.Request not found.rIznetwork.failRequestNrQ ValueErrorrPr+r7r)rrrrr fail_requests zRequest.fail_requestbodycCs|jstdd|ji}|dur||d<|dur||d<|dur$||d<|dur,||d<|dur4||d<|jjtd |dS) z)Continue after intercepting this request.rsrINrwrrrLrKrOrrt)rrwrrrLrKrOrrrrr&%s  zRequest.continue_requestrirjcCsPi}|j|d<|r |sd|d<n d|d<d||d|d<|jjtd|d S) 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. rIdefaultactionprovideCredentialsrj)typerirj credentialsr N)rQrPr+r7r)rrirjrrrrrk?s  zRequest._continue_with_auth)NNNNNNNN)rPrrQrrRrqrKrrSr<rLrrTrqrrr<rNrrOr<r r rp)NNNNN) rwrrrr<rLrrKrrOr<r r ro)rir<rjr<r r )rrrrrrvr&rkrrrrrUs&   rUN) __future__rcollections.abcrtypingr%selenium.webdriver.common.bidi.commonr.selenium.webdriver.remote.websocket_connectionrrrrUrrrrs      \