o M,hK@sddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZGdd d ZGd d d ZeGd d d ZeGdddZeGdddZGdddZGdddZGdddZGdddZdS)N) dataclass)AnyOptional)WebDriverException)command_builder) LogEntryAdded)Sessionc@seZdZdZdZdZdS)ResultOwnershipz/Represents the possible result ownership types.nonerootN)__name__ __module__ __qualname____doc__NONEROOTrrq/var/www/www-root/data/www/bot.pdev.uz/venv/lib/python3.10/site-packages/selenium/webdriver/common/bidi/script.pyr sr c@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) RealmTypez$Represents the possible realm types.windowzdedicated-workerz shared-workerzservice-workerworkerz paint-workletz audio-workletworkletN) r r rrWINDOWDEDICATED_WORKER SHARED_WORKERSERVICE_WORKERWORKER PAINT_WORKLET AUDIO_WORKLETWORKLETrrrrr$src@sheZdZUdZeed<eed<eed<dZeeed<dZeeed<e de ee fd dfd d Z dS) RealmInfoz%Represents information about a realm.realmorigintypeNcontextsandboxjsonreturncCsZd|vrtdd|vrtdd|vrtd||d|d|d|d|dd S) zCreates a RealmInfo instance from a dictionary. Parameters: ----------- json: A dictionary containing the realm information. Returns: ------- RealmInfo: A new instance of RealmInfo. r!z+Missing required field 'realm' in RealmInfor"z,Missing required field 'origin' in RealmInfor#z*Missing required field 'type' in RealmInfor$r%)r!r"r#r$r% ValueErrorgetclsr&rrr from_json;s zRealmInfo.from_json) r r rrstr__annotations__r$rr% classmethoddictrr-rrrrr 1s  r c@sHeZdZUdZeed<dZeeed<ede ee fddfddZ dS) Sourcez*Represents the source of a script message.r!Nr$r&r'cCs&d|vrtd||d|ddS)zCreates a Source instance from a dictionary. Parameters: ----------- json: A dictionary containing the source information. Returns: ------- Source: A new instance of Source. r!z(Missing required field 'realm' in Sourcer$)r!r$r(r+rrrr-^s  zSource.from_json) r r rrr.r/r$rr0r1rr-rrrrr2Ws  r2c@s`eZdZUdZeed<eed<dZeeed<dZ eeed<e deee fddfd d Z dS) EvaluateResultz+Represents the result of script evaluation.r#r!Nresultexception_detailsr&r'cCsDd|vrtdd|vrtd||d|d|d|ddS)aCreates an EvaluateResult instance from a dictionary. Parameters: ----------- json: A dictionary containing the evaluation result. Returns: ------- EvaluateResult: A new instance of EvaluateResult. r!z0Missing required field 'realm' in EvaluateResultr#z/Missing required field 'type' in EvaluateResultr4exceptionDetails)r#r!r4r5r(r+rrrr-|s zEvaluateResult.from_json) r r rrr.r/r4rr1r5r0rr-rrrrr3ss  r3c@sHeZdZdZdZdededefddZe deee fd dfd d Z d S) ScriptMessagez"Represents a script message event.script.messagechanneldatasourcecCs||_||_||_dSNr9r:r;)selfr9r:r;rrr__init__s zScriptMessage.__init__r&r'cCsPd|vrtdd|vrtdd|vrtd||d|dt|ddS)aCreates a ScriptMessage instance from a dictionary. Parameters: ----------- json: A dictionary containing the script message. Returns: ------- ScriptMessage: A new instance of ScriptMessage. r9z1Missing required field 'channel' in ScriptMessager:z.Missing required field 'data' in ScriptMessager;z0Missing required field 'source' in ScriptMessager=)r)r2r-r+rrrr-s  zScriptMessage.from_jsonN) r r rr event_classr.r1r2r?r0rr-rrrrr7s  r7c@s@eZdZdZdZdefddZedee e fddfdd Z d S) RealmCreatedz!Represents a realm created event.script.realmCreated realm_infocC ||_dSr<rC)r>rCrrrr? zRealmCreated.__init__r&r'cCs|t|dS)aCreates a RealmCreated instance from a dictionary. Parameters: ----------- json: A dictionary containing the realm created event. Returns: ------- RealmCreated: A new instance of RealmCreated. rEr r-r+rrrr-s zRealmCreated.from_jsonN) r r rrr@r r?r0r1r.rr-rrrrrA  rAc@s@eZdZdZdZdefddZedeee fddfdd Z d S) RealmDestroyedz#Represents a realm destroyed event.script.realmDestroyedr!cCrDr<r!)r>r!rrrr?rFzRealmDestroyed.__init__r&r'cCsd|vrtd||ddS)a Creates a RealmDestroyed instance from a dictionary. Parameters: ----------- json: A dictionary containing the realm destroyed event. Returns: ------- RealmDestroyed: A new instance of RealmDestroyed. r!z0Missing required field 'realm' in RealmDestroyedrK)r)r+rrrr-s zRealmDestroyed.from_jsonN) r r rrr@r.r?r0r1rr-rrrrrIrHrIc@seZdZdZddddZdde d(ed$e deee d)ee d*ee d+ee d,edefd-d.Z   'd?d/e d$e d(ed)ee d*ee d,edefd0d1Z  d@d2ee d3ee deefd4d5Zd6d7Zd8d9Zd:d;ZdS)AScriptz)BiDi implementation of the script module.r8rBrJ)message realm_createdrealm_destroyedNcCs"||_||_d|_i|_i|_dSNF)conndriverlog_entry_subscribed subscriptions callbacks)r>rQrRrrrr?s  zScript.__init__cC||jt|d|S)Nconsole_subscribe_to_log_entriesrQ add_callbackr_handle_log_entryr>handlerrrradd_console_message_handlerz"Script.add_console_message_handlercCrV)N javascriptrXr\rrradd_javascript_error_handlerr_z#Script.add_javascript_error_handlercCs|jt||dSr<)rQremove_callbackr_unsubscribe_from_log_entries)r>idrrrremove_console_message_handlers z%Script.remove_console_message_handlerscriptr'cCs ||S)zPins a script to the current browsing context. Parameters: ----------- script: The script to pin. Returns: ------- str: The ID of the pinned script. )_add_preload_script)r>rfrrrpin s z Script.pin script_idcCs||dS)zUnpins a script from the current browsing context. Parameters: ----------- script_id: The ID of the pinned script to unpin. N)_remove_preload_script)r>rirrrunpinsz Script.unpinc Gs|jdur td|jj}g}|D] }|||qd|i}|j|d||r)|ndd}|jdkr<|jdur:|jSiSd}|jrcd|jvrT|d |jd7}t|d |jvrc|d |jd 7}t|) aExecutes a script in the current browsing context. Parameters: ----------- script: The script function to execute. *args: Arguments to pass to the script function. Returns: ------- dict: The result value from the script execution. Raises: ------ WebDriverException: If the script execution fails. Nz1Driver reference is required for script executionr$T)function_declaration await_promisetarget argumentssuccesszError while executing scripttextz: rM) rRrcurrent_window_handleappend_Script__convert_to_local_value_call_functionr#r4r5) r>rfargsbrowsing_context_idroargrnr4 error_messagerrrexecute s(    zScript.executecs|durddiSt|trd|dSt|ttfrht|trLt|r(dddSt|r;|dkr6dd dSdd dS|d krLtd |dkrLdd dSd}t|trc||ks\|| krcdt|dSd|dSt|trrd|dSt|t j rd|j dur| ddS| dSt|t j rt j |t jjjt jjd}d| dSt|trdfdd|DdSt|ttfrdfdd|DdSt|trdfdd|DdSdt|dS)zD Converts a Python value to BiDi LocalValue format. Nr#nullboolean)r#valuenumberNaNrInfinityz -Infinitygg?z-0lbigintstringdateZ)tzinfosetcg|]}|qSrrt.0itemr>rr pz3Script.__convert_to_local_value..arraycrrrrrrrrrrobjectcs$g|]\}}||gqSrr)rkvrrrrvs) isinstanceboolintfloatmathisnanisinfcopysignr.datetimer isoformatrcombinetimeminreplacetimezoneutcrlisttupler1items)r>r}JS_MAX_SAFE_INTEGERdtrrr__convert_to_local_valueLsH             (    zScript.__convert_to_local_valuerlrocontexts user_contextsr%cCsz|dur |dur tdd|i}|dur||d<|dur ||d<|dur(||d<|dur0||d<|jtd|}|d S) aMAdds a preload script. Parameters: ----------- function_declaration: The function declaration to preload. arguments: The arguments to pass to the function. contexts: The browsing context IDs to apply the script to. user_contexts: The user context IDs to apply the script to. sandbox: The sandbox name to apply the script to. Returns: ------- str: The preload script ID. Raises: ------ ValueError: If both contexts and user_contexts are provided. Nz.Cannot specify both contexts and user_contextsfunctionDeclarationror userContextsr%zscript.addPreloadScriptrf)r)rQrzr)r>rlrorrr%paramsr4rrrrgszScript._add_preload_scriptcCsd|i}|jtd|dS)zRemoves a preload script. Parameters: ----------- script_id: The preload script ID to remove. rfzscript.removePreloadScriptNrQrzr)r>rirrrrrjszScript._remove_preload_scripthandlesrncCs ||d}|jtd|dS)zDisowns the given handles. Parameters: ----------- handles: The handles to disown. target: The target realm or context. )rrnz script.disownNr)r>rrnrrrr_disowns zScript._disownFrmresult_ownershipserialization_optionsthisuser_activationc Csj||||d} |dur|| d<|dur|| d<|dur|| d<|dur'|| d<|jtd| } t| S)aCalls a provided function with given arguments in a given realm. Parameters: ----------- function_declaration: The function declaration to call. await_promise: Whether to await promise resolution. target: The target realm or context. arguments: The arguments to pass to the function. result_ownership: The result ownership type. serialization_options: The serialization options. this: The 'this' value for the function call. user_activation: Whether to trigger user activation. Returns: ------- EvaluateResult: The result of the function call. )r awaitPromisernuserActivationNroresultOwnershipserializationOptionsrzscript.callFunctionrQrzrr3r-) r>rlrmrnrorrrrrr4rrrrus zScript._call_function expressionc CsJ||||d}|dur||d<|dur||d<|jtd|}t|S)a&Evaluates a provided script in a given realm. Parameters: ----------- expression: The script expression to evaluate. target: The target realm or context. await_promise: Whether to await promise resolution. result_ownership: The result ownership type. serialization_options: The serialization options. user_activation: Whether to trigger user activation. Returns: ------- EvaluateResult: The result of the script evaluation. )rrnrrNrrzscript.evaluater) r>rrnrmrrrrr4rrr _evaluates zScript._evaluater$r#cCsHi}|dur ||d<|dur||d<|jtd|}dd|dDS)a*Returns a list of all realms, optionally filtered. Parameters: ----------- context: The browsing context ID to filter by. type: The realm type to filter by. Returns: ------- List[RealmInfo]: A list of realm information. Nr$r#zscript.getRealmscSsg|]}t|qSrrG)rr!rrrr0rz&Script._get_realms..realmsr)r>r$r#rr4rrr _get_realmsszScript._get_realmscCs2|jst|j}|j|tjd|_dSdS)NT)rSrrQrz subscriberr@r>sessionrrrrY2s   z Script._subscribe_to_log_entriescCsD|jrtj|jjvr t|j}|j|tjd|_dSdSdSrP)rSrr@rQrUrrz unsubscriberrrrrc8s   z$Script._unsubscribe_from_log_entriescsfdd}|S)Ncs|jkr |dSdSr<)type_) log_entryr]r#rrr[?s  z3Script._handle_log_entry.._handle_log_entryr)r>r#r]r[rrrr[>szScript._handle_log_entryr<)NNNN)NNNNF)NNF)NN)r r rrEVENTSr?r^rareremove_javascript_error_handlerr.rhrkr1rzrtrrrrgrjrrr3rurr rrYrcr[rrrrrLs    ,6   +    4 )  rL)rr dataclassesrtypingrrselenium.common.exceptionsr%selenium.webdriver.common.bidi.commonr"selenium.webdriver.common.bidi.logr&selenium.webdriver.common.bidi.sessionrr rr r2r3r7rArIrLrrrrs(      %!$