o ѷ6iH@sddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gdd d ZGd d d ZeGd d d ZeGdddZeGdddZGdddZGdddZGdddZGdddZdS)N) dataclass)Any)WebDriverException)command_builder) LogEntryAdded)Sessionc@seZdZdZdZdZdS)ResultOwnershipz/Represents the possible result ownership types.nonerootN)__name__ __module__ __qualname____doc__NONEROOTrrj/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/selenium/webdriver/common/bidi/script.pyrsrc@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) RealmTypez$Represents the possible realm types.windowzdedicated-workerz shared-workerzservice-workerworkerz paint-workletz audio-workletworkletN) r r r rWINDOWDEDICATED_WORKER SHARED_WORKERSERVICE_WORKERWORKER PAINT_WORKLET AUDIO_WORKLETWORKLETrrrrr$src@sheZdZUdZeed<eed<eed<dZedBed<dZedBed<ede 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. Args: 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 r rstr__annotations__r#r$ classmethoddictrr,rrrrr1s  rc@sHeZdZUdZeed<dZedBed<edeee 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. Args: 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 r rr-r.r#r/r0rr,rrrrr1Us  r1c@s`eZdZUdZeed<eed<dZedBed<dZedBed<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)zCreates an EvaluateResult instance from a dictionary. Args: 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 EvaluateResultr3exceptionDetails)r"r r3r4r'r*rrrr,xs zEvaluateResult.from_json) r r r rr-r.r3r0r4r/rr,rrrrr2os  r2c@sHeZdZdZdZdededefddZe deee fd dfd d Z d S) ScriptMessagez"Represents a script message event.script.messagechanneldatasourcecCs||_||_||_dSNr8r9r:)selfr8r9r:rrr__init__s zScriptMessage.__init__r%r&cCsPd|vrtdd|vrtdd|vrtd||d|dt|ddS)zCreates a ScriptMessage instance from a dictionary. Args: json: A dictionary containing the script message. Returns: ScriptMessage: A new instance of ScriptMessage. r8z1Missing required field 'channel' in ScriptMessager9z.Missing required field 'data' in ScriptMessager:z0Missing required field 'source' in ScriptMessager<)r(r1r,r*rrrr,s  zScriptMessage.from_jsonN) r r r r event_classr-r0r1r>r/rr,rrrrr6s  r6c@s@eZdZdZdZdefddZedee e fddfdd Z d S) RealmCreatedz!Represents a realm created event.script.realmCreated realm_infocC ||_dSr;rB)r=rBrrrr> zRealmCreated.__init__r%r&cCs|t|dS)zCreates a RealmCreated instance from a dictionary. Args: json: A dictionary containing the realm created event. Returns: RealmCreated: A new instance of RealmCreated. rDrr,r*rrrr,s zRealmCreated.from_jsonN) r r r rr?rr>r/r0r-rr,rrrrr@  r@c@s@eZdZdZdZdefddZedeee fddfdd Z d S) RealmDestroyedz#Represents a realm destroyed event.script.realmDestroyedr cCrCr;r )r=r rrrr>rEzRealmDestroyed.__init__r%r&cCsd|vrtd||ddS)zCreates a RealmDestroyed instance from a dictionary. Args: json: A dictionary containing the realm destroyed event. Returns: RealmDestroyed: A new instance of RealmDestroyed. r z0Missing required field 'realm' in RealmDestroyedrJ)r(r*rrrr,s zRealmDestroyed.from_jsonN) r r r rr?r-r>r/r0rr,rrrrrHrGrHc@seZdZdZddddZdde d(ed$e dee dBd)e dBd*e dBd+e dBd,edefd-d.Z   'd?d/e d$e d(ed)e dBd*e dBd,edefd0d1Z  d@d2e dBd3e dBdeefd4d5Zd6d7Zd8d9Zd:d;ZdS)AScriptz)BiDi implementation of the script module.r7rArI)message realm_createdrealm_destroyedNcCs"||_||_d|_i|_i|_dSNF)conndriverlog_entry_subscribed subscriptions callbacks)r=rPrQrrrr>s  zScript.__init__cC||jt|d|S)Nconsole_subscribe_to_log_entriesrP add_callbackr_handle_log_entryr=handlerrrradd_console_message_handlerz"Script.add_console_message_handlercCrU)N javascriptrWr[rrradd_javascript_error_handlerr^z#Script.add_javascript_error_handlercCs|jt||dSr;)rPremove_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. Args: script: The script to pin. Returns: str: The ID of the pinned script. )_add_preload_script)r=rerrrpins z Script.pin script_idcCs||dS)zUnpins a script from the current browsing context. Args: script_id: The ID of the pinned script to unpin. N)_remove_preload_script)r=rhrrrunpin sz 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|) a[Executes a script in the current browsing context. Args: 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: rL) rQrcurrent_window_handleappend_Script__convert_to_local_value_call_functionr"r3r4) r=reargsbrowsing_context_idrnargrmr3 error_messagerrrexecutes(    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)z2Converts a Python value to BiDi LocalValue format.Nr"nullboolean)r"valuenumberNaNrInfinityz -Infinitygg?z-0lbigintstringdateZ)tzinfosetcg|]}|qSrrs.0itemr=rr [z3Script.__convert_to_local_value..arraycrrrrrrrr]robjectcs$g|]\}}||gqSrr)rkvrrrras) isinstanceboolintfloatmathisnanisinfcopysignr-datetimer isoformatrcombinetimeminreplacetimezoneutcrlisttupler0items)r=r|JS_MAX_SAFE_INTEGERdtrrr__convert_to_local_value9sH             (    zScript.__convert_to_local_valuerkrncontexts user_contextsr$cCsz|dur |dur tdd|i}|dur||d<|dur ||d<|dur(||d<|dur0||d<|jtd|}|d S) aAdds a preload script. Args: 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_contextsfunctionDeclarationrnr userContextsr$zscript.addPreloadScriptre)r(rPryr)r=rkrnrrr$paramsr3rrrrfjszScript._add_preload_scriptcCsd|i}|jtd|dS)ziRemoves a preload script. Args: script_id: The preload script ID to remove. rezscript.removePreloadScriptNrPryr)r=rhrrrrriszScript._remove_preload_scripthandlesrmcCs ||d}|jtd|dS)zDisowns the given handles. Args: handles: The handles to disown. target: The target realm or context. )rrmz script.disownNr)r=rrmrrrr_disownszScript._disownFrlresult_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. Args: 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 awaitPromisermuserActivationNrnresultOwnershipserializationOptionsrzscript.callFunctionrPryrr2r,) r=rkrlrmrnrrrrrr3rrrrts zScript._call_function expressionc CsJ||||d}|dur||d<|dur||d<|jtd|}t|S)aEvaluates a provided script in a given realm. Args: 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. )rrmrrNrrzscript.evaluater) r=rrmrlrrrrr3rrr _evaluates zScript._evaluater#r"cCsHi}|dur ||d<|dur||d<|jtd|}dd|dDS)aReturns a list of all realms, optionally filtered. Args: 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|qSrrF)rr rrrrrz&Script._get_realms..realmsr)r=r#r"rr3rrr _get_realmsszScript._get_realmscCs2|jst|j}|j|tjd|_dSdS)NT)rRrrPry subscriberr?r=sessionrrrrXs   z Script._subscribe_to_log_entriescCsD|jrtj|jjvr t|j}|j|tjd|_dSdSdSrO)rRrr?rPrTrry unsubscriberrrrrbs   z$Script._unsubscribe_from_log_entriescsfdd}|S)Ncs|jkr |dSdSr;)type_) log_entryr\r"rrrZs  z3Script._handle_log_entry.._handle_log_entryr)r=r"r\rZrrrrZszScript._handle_log_entryr;)NNNN)NNNNF)NNF)NN)r r r rEVENTSr>r]r`rdremove_javascript_error_handlerr-rgrjr0ryrsrrrfrirrr2rtrrrrXrbrZrrrrrKs   (4   (    2 '  rK)rr dataclassesrtypingrselenium.common.exceptionsr%selenium.webdriver.common.bidi.commonr"selenium.webdriver.common.bidi.logr&selenium.webdriver.common.bidi.sessionrrrrr1r2r6r@rHrKrrrrs(       #"