hQK2ddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZGd d ZGd d ZeGd dZeGddZeGddZGddZGddZGddZGddZy)N) dataclass)AnyOptional)WebDriverException)command_builder) LogEntryAdded)SessionceZdZdZdZdZy)ResultOwnershipz/Represents the possible result ownership types.nonerootN)__name__ __module__ __qualname____doc__NONEROOTg/var/www/html/w1hd.aprs.trek/venv/lib/python3.12/site-packages/selenium/webdriver/common/bidi/script.pyr r s9 D Drr c0eZdZdZdZdZdZdZdZdZ dZ d Z y ) RealmTypez$Represents the possible realm types.windowzdedicated-workerz shared-workerzservice-workerworkerz paint-workletz audio-workletworkletN) rrrrWINDOWDEDICATED_WORKER SHARED_WORKERSERVICE_WORKERWORKER PAINT_WORKLET AUDIO_WORKLETWORKLETrrrrr%s.. F)#M%N F#M#MGrrc|eZdZUdZeed<eed<eed<dZeeed<dZeeed<e de ee fd dfd Z y) RealmInfoz%Represents information about a realm.realmorigintypeNcontextsandboxjsonreturnc d|vr tdd|vr tdd|vr td||d|d|d|jd|jd 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,s r from_jsonzRealmInfo.from_json<sz $ JK K 4 KL L  IJ Jw->fHHY'HHY'   r) rrrrstr__annotations__r*rr+ classmethoddictrr4rrrr&r&2sX/ J K I!GXc]!!GXc]! T#s(^   rr&cTeZdZUdZeed<dZeeed<ede ee fddfdZ y)Sourcez*Represents the source of a script message.r'Nr*r,r-cXd|vr td||d|jdS)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/r2s rr4zSource.from_json_s8 $ GH Hw-HHY'  r) rrrrr5r6r*rr7r8rr4rrrr:r:Xs@4 J!GXc]! T#s(^   rr:creZdZUdZeed<eed<dZeeed<dZ eeed<e deee fddfd Z y) EvaluateResultz+Represents the result of script evaluation.r)r'Nresultexception_detailsr,r-cd|vr tdd|vr td||d|d|jd|jdS)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 EvaluateResultr>exceptionDetails)r)r'r>r?r/r2s rr4zEvaluateResult.from_json}s` $ OP P  NO Ofw-88H%"hh'9:   r) rrrrr5r6r>rr8r?r7rr4rrrr=r=tsU5 I J!FHTN!(,x~, T#s(^ 0@  rr=cLeZdZdZdZdededefdZe deee fddfd Z y ) ScriptMessagez"Represents a script message event.script.messagechanneldatasourcec.||_||_||_yNrErFrG)selfrErFrGs r__init__zScriptMessage.__init__s   rr,r-cd|vr tdd|vr tdd|vr td||d|dtj|dS)aCreates a ScriptMessage instance from a dictionary. Parameters: ----------- json: A dictionary containing the script message. Returns: ------- ScriptMessage: A new instance of ScriptMessage. rEz1Missing required field 'channel' in ScriptMessagerFz.Missing required field 'data' in ScriptMessagerGz0Missing required field 'source' in ScriptMessagerJ)r0r:r4r2s rr4zScriptMessage.from_jsonsm D PQ Q  MN N 4 OP POf##DN3  rN) rrrr event_classr5r8r:rLr7rr4rrrrCrCsN,"K4  T#s(^   rrCcDeZdZdZdZdefdZedee e fddfdZ y) RealmCreatedz!Represents a realm created event.script.realmCreated realm_infoc||_yrIrR)rKrRs rrLzRealmCreated.__init__s $rr,r-c:|tj|S)aCreates a RealmCreated instance from a dictionary. Parameters: ----------- json: A dictionary containing the realm created event. Returns: ------- RealmCreated: A new instance of RealmCreated. rT)r&r4r2s rr4zRealmCreated.from_jsonsi11$788rN) rrrrrNr&rLr7r8r5rr4rrrrPrPs@+'K%9% 9T#s(^ 9 9 9rrPcDeZdZdZdZdefdZedeee fddfdZ y) RealmDestroyedz#Represents a realm destroyed event.script.realmDestroyedr'c||_yrIr')rKr's rrLzRealmDestroyed.__init__s  rr,r-c8d|vr td||dS)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 RealmDestroyedrZ)r0r2s rr4zRealmDestroyed.from_jsons' $ OP Pg''rN) rrrrrNr5rLr7r8rr4rrrrWrWsA-)Kc(T#s(^(0@((rrWceZdZdZddddZd*dZdZd Zd ZeZ d e d e fd Z de d dfdZ d e d e fdZd e fdZ d+de deee e efdeee deee dee d e f dZde d dfdZdee de d dfdZ d,de dede deee dee dee dee d ed efd!Z d-d"e de dedee dee d ed efd#Z d.d$ee d%ee d eefd&Zd'Zd(Zd)Zy)/Scriptz)BiDi implementation of the script module.rDrQrX)message realm_createdrealm_destroyedNcJ||_||_d|_i|_i|_yNF)conndriverlog_entry_subscribed subscriptions callbacks)rKrcrds rrLzScript.__init__s'  $)!rc|j|jjt|j d|S)Nconsole_subscribe_to_log_entriesrc add_callbackr _handle_log_entryrKhandlers radd_console_message_handlerz"Script.add_console_message_handlers3 &&(yy%%mT5K5KIW^5_``rc|j|jjt|j d|S)N javascriptrjrns radd_javascript_error_handlerz#Script.add_javascript_error_handlers3 &&(yy%%mT5K5KLZa5bccrcd|jjt||jyrI)rcremove_callbackr _unsubscribe_from_log_entries)rKids rremove_console_message_handlerz%Script.remove_console_message_handlers" !!-4 **,rscriptr-c$|j|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)rKrys rpinz Script.pin s''//r script_idc&|j|y)zUnpins a script from the current browsing context. Parameters: ----------- script_id: The ID of the pinned script to unpin. N)_remove_preload_script)rKr}s runpinz Script.unpins ##I.rc|j td|jj}g}|D]"}|j|j |$d|i}|j |d||r|nd}|j dk(r |jSd}|jrQd|jvr |d |jdz }t|d |jvr|d |jd z }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: r^) rdrcurrent_window_handleappend_Script__convert_to_local_value_call_functionr)r>r?) rKryargsbrowsing_context_idrargrr> error_messages rexecutezScript.execute!s#" ;; $%XY Y"kk??  AC   T::3? @ A01$$!'tFclV_rv%  ;;) #== :M''V555!r&*B*B6*J)K%LLM%]3 3&":"::!r&*B*B9*M)N%OOM$]3 3rc B|ddiSt|trd|dSt|ttfrt|trat j |rdddSt j |r|dkDrdddSdd dS|d k(rt jd |dkrdd dSd }t|tr||kDs|| krdt|dSd|dSt|trd|dSt|tjr4d|j|jdzdS|jdSt|tjrstjj|tjjj!tj"j$}d|jdSt|t&r"d|Dcgc]}|j)|c}dSt|t*t,fr"d|Dcgc]}|j)|c}dSt|t.rFd|j1Dcgc]'\}}|j)||j)|g)c}}dSdt|dScc}wcc}wcc}}w)zD Converts a Python value to BiDi LocalValue format. r)nullboolean)r)valuenumberNaNrInfinityz -Infinitygg?z-0lbigintstringdateZ)tzinfosetarrayobject) isinstanceboolintfloatmathisnanisinfcopysignr5datetimer isoformatrcombinetimeminreplacetimezoneutcrrlisttupler8items)rKrJS_MAX_SAFE_INTEGERdtitemkvs r__convert_to_local_valuezScript.__convert_to_local_valueMsg =F# # t $%6 6 U| ,%'::e$$,u==ZZ&qy(0:FF(0;GGc\dmmC&?!&C$,t<<"2 %%53F+F%SfRfJf (3u:>>$u5 5 s #$u5 5 x00 1" H\U__->-Dt tbgbqbqbst t x}} -""**5(--2C2CDLLT\TeTeTiTiLjB"R\\^< < s #!]b,cUYT-J-J4-P,cd d e} -#_d.eW[t/L/LT/R.ef f t $ ejepeper]a]^`aT2215t7T7TUV7WX %s5z: :-d.esJJ,Jrrcontexts user_contextsr+c| | tdd|i}|||d<|||d<|||d<|||d<|jjtd|}|dS) 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. z.Cannot specify both contexts and user_contextsfunctionDeclarationrr userContextsr+zscript.addPreloadScriptry)r0rcrr)rKrrrrr+paramsr>s rr{zScript._add_preload_scripts4  M$=MN N"79M!N  "+F;   !)F:   $%2F> "   'F9 ""?3Lf#UVhrcVd|i}|jjtd|y)zRemoves a preload script. Parameters: ----------- script_id: The preload script ID to remove. ryzscript.removePreloadScriptNrcrr)rKr}rs rrzScript._remove_preload_scripts(I& /*FOPrhandlesrcX||d}|jjtd|y)zDisowns the given handles. Parameters: ----------- handles: The handles to disown. target: The target realm or context. )rrz script.disownNr)rKrrrs r_disownzScript._disowns,  //6BCrrresult_ownershipserialization_optionsthisuser_activationc ||||d} ||| d<||| d<||| d<||| d<|jjtd| } tj | 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 awaitPromiseruserActivationrresultOwnershipserializationOptionsrzscript.callFunctionrcrrr=r4) rKrrrrrrrrrr>s rrzScript._call_functions:$8)-    "+F;   '(8F$ % ,-BF) *  !F6N""?3H&#QR''//r expressionc||||d}|||d<|||d<|jjtd|}tj |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. )rrrrrrzscript.evaluater) rKrrrrrrrr>s r _evaluatezScript._evaluatesh2%)-    '(8F$ % ,-BF) *""?3Df#MN''//rr*r)ci}|||d<|||d<|jjtd|}|dDcgc]}tj |c}Scc}w)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. r*r)zscript.getRealmsrealms)rcrrr&r4)rKr*r)rr>r's r _get_realmszScript._get_realmssh    'F9   !F6N""?3Ev#NO8>x8HIu ##E*IIIsAc|jsUt|j}|jj|j t j d|_yy)NT)rer rcr subscriber rNrKsessions rrkz Script._subscribe_to_log_entries3sI((dii(G II  g// 0I0IJ K(,D %)rc|jr|tj|jjvrUt |j}|jj |jtjd|_yyyrb)rer rNrcrgr r unsubscribers rrvz$Script._unsubscribe_from_log_entries9sd  $ $)B)B$))J]J])]dii(G II  g11-2K2KL M(-D %*^ $rcfd}|S)Nc6|jk(r |yyrI)type_) log_entryror)s rrmz3Script._handle_log_entry.._handle_log_entry@s$& "'rr)rKr)rorms `` rrmzScript._handle_log_entry?s #! rrI)NNNN)NNNNF)NNF)NN)rrrrEVENTSrLrprsrxremove_javascript_error_handlerr5r|rr8rrrrrr{rrrr=rrr&rrkrvrmrrrr]r]sT3$.2F ad-'E# 0# 0# 0/s/t/*4c*4T*4X0;0;l59(,-1!% ) !) Dc3h01) 49% )  S * ) # )  ) VQQQ DtCy D$ D4 D&+/*.04# %-0!-0-0 -0 DJ' -0 #3- -0 (~-0tn-0-0 -0h+/04 %%0%0%0 %0 #3- %0 (~ %0%0 %0R"&"J#JsmJ i J4- . !rr])rr dataclassesrtypingrrselenium.common.exceptionsr%selenium.webdriver.common.bidi.commonrlogr rr r rr&r:r=rCrPrWr]rrrrs$ ! 9A   " "  " J     6     B! ! H99.((4Y!Y!r