U Pf@s`dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlm Z ddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl'm)Z)ddl'm*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4dd l3m5Z5dd!l6m7Z7dd"l8m9Z9dd#l:m;Z;dd$lm?Z?dd&l@mAZAdd'lBmCZCdaDdaEd(d)ZFd*d+ZGd7d-d.ZHee"ed/d0d1ZIGd2d3d3e d4ZJGd5d6d6eJZKdS)8zThe WebDriver implementation.N)ABCMeta) b64decode)urlsafe_b64encode)asynccontextmanager)contextmanager) import_module)Dict)List)Optional)Union)InvalidArgumentException)JavascriptException)NoSuchCookieException)NoSuchElementException)WebDriverException)Script)By) BaseOptions) PrintOptions)Timeouts) Credential)VirtualAuthenticatorOptions)required_virtual_authenticator) RelativeBy)BidiConnection)Command) ErrorHandler) FileDetector)LocalFileDetector)Mobile)RemoteConnection) ScriptKey) ShadowRoot)SwitchTo) WebElement)WebSocketConnectioncCsts tdadS)Nz"selenium.webdriver.common.bidi.cdp)cdprr(r(G/tmp/pip-unpacked-wheel-ds5t_0qj/selenium/webdriver/remote/webdriver.py import_cdpIsr*cCs8t|}i}|D]\}}|||<qdig|diS)aMakes a W3C alwaysMatch capabilities object. Filters out capability names that are not in the W3C spec. Spec-compliant drivers will reject requests containing unknown capability names. Moves the Firefox profile, if present, from the old location to the new Firefox options object. :Args: - caps - A dictionary of capabilities requested by the caller. capabilities) firstMatch alwaysMatch)copydeepcopyitems)capsZ always_matchkvr(r(r) _create_capsOs  r4Fc sbddlm}ddlm}ddlm}ddlm}||||g}tfdd|Dt } | |||dS) Nr)ChromeRemoteConnection)EdgeRemoteConnection)FirefoxRemoteConnection)SafariRemoteConnectionc3s"|]}|jdkr|VqdS) browserNameN)Z browser_nameget).0cr+r(r) isz(get_remote_connection..) keep_aliveZ ignore_proxy) Z+selenium.webdriver.chrome.remote_connectionr5Z)selenium.webdriver.edge.remote_connectionr6Z,selenium.webdriver.firefox.remote_connectionr7Z+selenium.webdriver.safari.remote_connectionr8nextr!) r+command_executorr?ignore_local_proxyr5r6r7r8 candidateshandlerr(r=r)get_remote_connectionbs     rEoptionsreturnc Csdii}g}|D]}||qt|}i}t|D]p}|}|d|kr8||}|D]J} | ||dkr\||| ||d| kr\|| ||| iq\q8i} |D]\} } | | | <q|D]}| D] } || =qq| |dd<||dd<|S)Nr+rr-r,)appendto_capabilitieslenrangekeysupdater0) rGr+optsoptZ opts_sizeZsamesiesiZ min_indexZ first_keysZkysalwaysr2r3r(r(r)create_matchesns.       rSc@seZdZdZdS) BaseWebDriverzAbstract Base Class for all Webdriver subtypes. ABC's allow custom implementations of Webdriver to be registered so that isinstance type checks will succeed. N)__name__ __module__ __qualname____doc__r(r(r(r)rTsrT) metaclassc @sZeZdZdZeZeZdee e fe e e e eeeefddddZdd Zd d Zej ejeej eej ejd d dZeddZeedddZee dddZddZddZ e!ddddZ"ddZ#e edd d!Z$d"d#Z%de e!e!d$d%d&Z&e dd'd(d)Z'ee dd*d+Z(de e)d,d-d.Z*e)dd/d0d1Z+ee dd2d3Z,d4d5Z-e d6d7d8Z.ee dd9d:Z/ee dd;d<Z0ddd=d>Z1ddd?d@Z2ee ddAdBZ3eee ddCdDZ4dddEdFZ5dddGdHZ6dddIdJZ7de e8e dKdLdMZ9ee:ddNdOZ;dddPdQZee!ddVdWZ?ej ej@ddXdYZAdddZd[ZBddd\d]ZCddd^d_ZDeEdd`dadbZFeEdd`dcddZGeEdd`dedfZHeeIddgdhZJeJjKdddidhZJeLjMdfe e edjdkdlZNeLjMdfe e eedjdmdnZOee!ddodpZPe ddqdrZQe ddsdtZReSddudvZTe ddwdxZUde ddzd{d|ZVde e!dzd}d~ZWdeEeEe e!dddZXde!dddZYe ddzddZZe!dddZ[de!dddZ\ee dddZ]e]jKddddZ]eddZ^e^jKddddZ^eddZ_ddZ`ddZaebddZceddZdddZeddZfegddddZhee dddZiejddddZkejelddddZmejeeldddZnejee eofddddZpejddddZqeje ddddZre!dddZse e ddddZtddddZudS) WebDriveraControls a browser by sending commands to a remote server. This server is expected to be running the WebDriver wire protocol as defined at https://www.selenium.dev/documentation/legacy/json_wire_protocol/. :Attributes: - session_id - String ID of the browser session started and controlled by this WebDriver. - capabilities - Dictionary of effective capabilities of this browser session as returned by the remote server. See https://www.selenium.dev/documentation/legacy/desired_capabilities/ - command_executor - remote_connection.RemoteConnection object used to execute commands. - error_handler - errorhandler.ErrorHandler object used to handle errors. http://127.0.0.1:4444TN)rAr? file_detectorrGrHcCst|trt|}d}n|}|j}||_t|jttfrNt||||d|_d|_ d|_ i|_ i|_ t |_t||_t||_|pt|_d|_|||d|_d|_dS)aCreate a new driver that will issue commands using the wire protocol. :Args: - command_executor - Either a string representing URL of the remote server or a custom remote_connection.RemoteConnection object. Defaults to 'http://127.0.0.1:4444/wd/hub'. - keep_alive - Whether to configure remote_connection.RemoteConnection to use HTTP keep-alive. Defaults to True. - file_detector - Pass custom file detector object during instantiation. If None, then default LocalFileDetector() will be used. - options - instance of a driver options.Options class F)rAr?rBTN) isinstancelistrSrJ_ignore_local_proxyrAstrbytesrEZ _is_remote session_idr1pinned_scriptsr error_handlerr$ _switch_tor _mobilerr\_authenticator_id start_client start_session_websocket_connection_script)selfrAr?r\rGr+r_r(r(r)__init__s4     zWebDriver.__init__cCs&dt|jdt|jd|jdS)N<.z (session="z")>)typerVrUrbrlr(r(r)__repr__szWebDriver.__repr__cCs|SNr(rqr(r(r) __enter__szWebDriver.__enter__)exc_typeexc tracebackcCs |dSrs)quit)rlrurvrwr(r(r)__exit__szWebDriver.__exit__cos>d}t|j|s"|j}||||_z dVW5|r8||_XdS)aOverrides the current file detector (if necessary) in limited context. Ensures the original file detector is set afterwards. Example:: with webdriver.file_detector_context(UselessFileDetector): someinput.send_keys('/etc/hosts') :Args: - file_detector_class - Class of the desired file detector. If the class is different from the current file_detector, then the class is instantiated with args and kwargs and used as a file detector during the duration of the context manager. - args - Optional arguments that get passed to the file detector class during instantiation. - kwargs - Keyword arguments, passed the same way as args. N)r]r\)rlZfile_detector_classargskwargsZ last_detectorr(r(r)file_detector_contexts   zWebDriver.file_detector_context)rHcCs|jSrs)rfrqr(r(r)mobileszWebDriver.mobilecCs d|jkr|jdStddS)zReturns the name of the underlying browser for this instance. :Usage: :: name = driver.name r9z1browserName not specified in session capabilitiesN)r1KeyErrorrqr(r(r)names  zWebDriver.namecCsdS)zxCalled before starting a new session. This method may be overridden to define custom startup behavior. Nr(rqr(r(r)rhszWebDriver.start_clientcCsdS)zCalled after executing a quit command. This method may be overridden to define custom shutdown behavior. Nr(rqr(r(r) stop_clientszWebDriver.stop_client)r+rHcCs6t|}|tj|d}|d|_|d|_dS)zCreates a new session with the desired capabilities. :Args: - capabilities - a capabilities dict to start the session with. value sessionIdr+N)r4executerZ NEW_SESSIONr:rbr1)rlr+r1responser(r(r)ri#s zWebDriver.start_sessioncst|tr2i}|D]\}}|||<q|St|jrHd|jiSt|jr^d|jiSt|tr~tfdd|DS|S)N#element-6066-11e4-a52e-4f735466cecf"shadow-6066-11e4-a52e-4f735466cecfc3s|]}|VqdSrs) _wrap_valuer;itemrqr(r)r>:sz(WebDriver._wrap_value..)r]dictr0r_web_element_clsid_shadowroot_clsr^)rlrZ convertedkeyvalr(rqr)r/s      zWebDriver._wrap_value) element_idrHcCs |||S)z6Creates a web element with the specified `element_id`.)r)rlrr(r(r)create_web_element=szWebDriver.create_web_elementcst|tr\d|kr |dSd|kr8|dS|D]\}}|||<q@|St|tr|tfdd|DS|S)Nrrc3s|]}|VqdSrs) _unwrap_valuerrqr(r)r>Ksz*WebDriver._unwrap_value..)r]rrrr0rr^)rlrrrr(rqr)rAs  zWebDriver._unwrap_value)driver_commandparamsrHcCsx||}|jr2|s d|ji}nd|kr2|j|d<|j||}|rj|j|||dd|d<|Sdd|jdS)aHSends a command to be executed by a command.CommandExecutor. :Args: - driver_command: The name of the command to execute as a string. - params: A dictionary of named parameters to send with the command. :Returns: The command's JSON response loaded into a dictionary object. rrNr)successrr)rrbrArrdZcheck_responserr:)rlrrrr(r(r)rNs    zWebDriver.execute)urlrHcCs|tjd|idS)z0Loads a web page in the current browser session.rN)rrGET)rlrr(r(r)r:isz WebDriver.getcCs|tjddS)zuReturns the title of the current page. :Usage: :: title = driver.title r)rrZ GET_TITLEr:rqr(r(r)titlems zWebDriver.title)scriptrHcCst|}||j|j<|S)zUStore common javascript scripts to be executed later by a unique hashable ID.)r"rcr)rlr script_keyZscript_key_instancer(r(r) pin_scriptxs zWebDriver.pin_script)rrHcCsDz|j|jWn,tk r>td|d|jdYnXdS)z$Remove a pinned script from storage.zNo script with key: z existed in N)rcpoprr~)rlrr(r(r)unpinszWebDriver.unpincCs t|jSrs)r^rcrqr(r(r)get_pinned_scriptsszWebDriver.get_pinned_scriptscGs\t|tr8z|j|j}Wntk r6tdYnXt|}tj}| |||ddS)a'Synchronously Executes JavaScript in the current window/frame. :Args: - script: The JavaScript to execute. - \*args: Any applicable arguments for your JavaScript. :Usage: :: driver.execute_script('return document.title;') z Pinned script could not be foundrrzr) r]r"rcrr~r r^rZW3C_EXECUTE_SCRIPTrrlrrzZconverted_argscommandr(r(r)execute_scripts zWebDriver.execute_script)rcGs$t|}tj}||||ddS)aAsynchronously Executes JavaScript in the current window/frame. :Args: - script: The JavaScript to execute. - \*args: Any applicable arguments for your JavaScript. :Usage: :: script = "var callback = arguments[arguments.length - 1]; " \ "window.setTimeout(function(){ callback('timeout') }, 3000);" driver.execute_async_script(script) rr)r^rZW3C_EXECUTE_SCRIPT_ASYNCrrr(r(r)execute_async_scriptszWebDriver.execute_async_scriptcCs|tjdS)znGets the URL of the current page. :Usage: :: driver.current_url r)rrZGET_CURRENT_URLrqr(r(r) current_urls zWebDriver.current_urlcCs|tjdS)zqGets the source of the current page. :Usage: :: driver.page_source r)rrZGET_PAGE_SOURCErqr(r(r) page_sources zWebDriver.page_sourcecCs|tjdS)zcCloses the current window. :Usage: :: driver.close() N)rrZCLOSErqr(r(r)closeszWebDriver.closecCs*z|tjW5||jXdS)z|Quits the driver and closes every associated window. :Usage: :: driver.quit() N)rrArrrZQUITrqr(r(r)rxszWebDriver.quitcCs|tjdS)zReturns the handle of the current window. :Usage: :: driver.current_window_handle r)rrZW3C_GET_CURRENT_WINDOW_HANDLErqr(r(r)current_window_handles zWebDriver.current_window_handlecCs|tjdS)zReturns the handles of all windows within the current session. :Usage: :: driver.window_handles r)rrZW3C_GET_WINDOW_HANDLESrqr(r(r)window_handless zWebDriver.window_handlescCstj}||ddS)z5Maximizes the current window that webdriver is using.N)rZW3C_MAXIMIZE_WINDOWr)rlrr(r(r)maximize_windowszWebDriver.maximize_windowcCs|tjdS)zrsz-WebDriver.get_window_size..r)rget_window_rectr:rlrr(rr)get_window_sizecs  zWebDriver.get_window_size)xyrrHcCs |||jt|t|dS)a.Sets the x,y position of the current window. (window.moveTo) :Args: - x: the x-coordinate in pixels to set the window position - y: the y-coordinate in pixels to set the window position :Usage: :: driver.set_window_position(0,0) rrr)rlrrrr(r(r)set_window_positionts zWebDriver.set_window_positioncs$|||fdddDS)zGets the x,y position of the current window. :Usage: :: driver.get_window_position() csi|]}||qSr(r(rpositionr(r)rsz1WebDriver.get_window_position..r )rrrr(r r)get_window_positions zWebDriver.get_window_positioncCs|dkrtjddddS)z5Warns if the window handle is not equal to `current`.rz?Only 'current' window is supported for W3C compatible browsers.rrN)rrrr(r(r)rsz,WebDriver._check_if_window_handle_is_currentcCs|tjdS)zGets the x, y coordinates of the window as well as height and width of the current window. :Usage: :: driver.get_window_rect() r)rrZGET_WINDOW_RECTrqr(r(r)rs zWebDriver.get_window_rectcCs<|dkr |dkr |s |s td|tj||||ddS)aSets the x, y coordinates of the window as well as height and width of the current window. This method is only supported for W3C compatible browsers; other browsers should use `set_window_position` and `set_window_size`. :Usage: :: driver.set_window_rect(x=10, y=10) driver.set_window_rect(width=100, height=200) driver.set_window_rect(x=10, y=10, width=100, height=200) Nz'x and y or height and width need values)rrrrr)r rrZSET_WINDOW_RECT)rlrrrrr(r(r)rszWebDriver.set_window_rectcCs|jSrs)_file_detectorrqr(r(r)r\szWebDriver.file_detectorcCs(|s tdt|tstd||_dS)a0Set the file detector to be used when sending keyboard input. By default, this is set to a file detector that does nothing. see FileDetector see LocalFileDetector see UselessFileDetector :Args: - detector: The detector to use. Must not be None. z,You may not set a file detector that is nullz+Detector has to be instance of FileDetectorN)rr]rr)rldetectorr(r(r)r\s  cCs|tjdS)zGets the current orientation of the device. :Usage: :: orientation = driver.orientation r)rrZGET_SCREEN_ORIENTATIONrqr(r(r) orientations zWebDriver.orientationcCs4ddg}||kr(|tjd|intddS)zSets the current orientation of the device. :Args: - value: orientation to set it to. :Usage: :: driver.orientation = 'landscape' Z LANDSCAPEZPORTRAITrz>You can only set the orientation to 'LANDSCAPE' and 'PORTRAIT'N)upperrrZSET_SCREEN_ORIENTATIONr)rlrZallowed_valuesr(r(r)rs  cCs|tjdS)zGets a list of the available log types. This only works with w3c compliant browsers. :Usage: :: driver.log_types r)rrZGET_AVAILABLE_LOG_TYPESrqr(r(r) log_typess zWebDriver.log_typescCs|tjd|idS)a>Gets the log for a given log type. :Args: - log_type: type of log that which will be returned :Usage: :: driver.get_log('browser') driver.get_log('driver') driver.get_log('client') driver.get_log('server') rpr)rrZGET_LOG)rlZlog_typer(r(r)get_logszWebDriver.get_logcCs|jrt|jfSttsl|jdrJ|jd}|jddd}n |\}}|sbtdt |at ||_|j tj }|dj}|j tj |d}||j_t|jfSdS)Nse:cdp se:cdpVersionror2Unable to find url to connect to from capabilitiesT)rjdevtoolsr*r1r:r_get_cdp_detailsrr'import_devtoolsr&rtarget get_targets target_idZattach_to_targetrb)rlws_urlversiontargetsrsessionr(r(r)start_devtoolss"       zWebDriver.start_devtoolsc Cst|jdr6|jd}|jddd}n |\}}|sNtdt|}t|4IdHX}| |j IdH}|dj }| |4IdH}t|t|VW5QIdHRXW5QIdHRXdS)Nrrrorr)r*r1r:rrrr'rZopen_cdprrrrZ open_sessionr)rlrrrconnrrr r(r(r)bidi_connections     zWebDriver.bidi_connectioncCs&|js||js t|j|_|jSrs)rj _start_bidirkrrqr(r(r)r1s  zWebDriver.scriptcCs0|jdr|jd}ntdt||_dS)NZ webSocketUrlr)r1r:rr&rj)rlrr(r(r)r$;s zWebDriver._start_bidic Csddl}ddl}|}d}|jddkr@|jdd}n4|jddkrd|jdd}nd }|jd }|d d |d }||j}|d}|d} ddl} |rd} n| d| d} | | fS)NrFr9Zchromezgoog:chromeOptionsZdebuggerAddressZ MicrosoftEdgezms:edgeOptionsTzmoz:debuggerAddressrzhttp://z /json/versionZBrowserZwebSocketDebuggerUrlUz .*/(\d+)\.r) jsonurllib3 PoolManagerr1r:requestloadsdataresearchgroup) rlr&r'httpZ_firefoxZdebugger_addressresr+Zbrowser_versionZ websocket_urlr,rr(r(r)rCs&    zWebDriver._get_cdp_detailsrFcCs|tj|d|_dS)z4Adds a virtual authenticator with the given options.rN)rrZADD_VIRTUAL_AUTHENTICATORrrg)rlrGr(r(r)add_virtual_authenticatorcsz#WebDriver.add_virtual_authenticatorcCs|jS)z,Returns the id of the virtual authenticator.)rgrqr(r(r)virtual_authenticator_idgsz"WebDriver.virtual_authenticator_idcCs|tjd|jid|_dS)zRemoves a previously added virtual authenticator. The authenticator is no longer valid after removal, so no methods may be called. authenticatorIdN)rrZREMOVE_VIRTUAL_AUTHENTICATORrgrqr(r(r)remove_virtual_authenticatorlsz&WebDriver.remove_virtual_authenticator) credentialrHcCs |tj|d|jidS)z,Injects a credential into the authenticator.r3N)rrZADD_CREDENTIALrrg)rlr5r(r(r)add_credentialvszWebDriver.add_credentialcCs&|tjd|ji}dd|dDS)z;Returns the list of credentials owned by the authenticator.r3cSsg|]}t|qSr()r from_dict)r;r5r(r(r) sz-WebDriver.get_credentials..r)rrZGET_CREDENTIALSrg)rlZcredential_datar(r(r)get_credentials{szWebDriver.get_credentials) credential_idrHcCs0t|trt|}|tj||jddS)z,Removes a credential from the authenticator.)Z credentialIdr3N)r] bytearrayrrrrZREMOVE_CREDENTIALrg)rlr:r(r(r)remove_credentials    zWebDriver.remove_credentialcCs|tjd|jidS)z/Removes all credentials from the authenticator.r3N)rrZREMOVE_ALL_CREDENTIALSrgrqr(r(r)remove_all_credentialssz WebDriver.remove_all_credentials)verifiedrHcCs|tj|j|ddS)zSets whether the authenticator will simulate success or fail on user verification. verified: True if the authenticator will pass user verification, False otherwise. )r3ZisUserVerifiedN)rrZSET_USER_VERIFIEDrg)rlr>r(r(r)set_user_verifiedszWebDriver.set_user_verifiedcCs&d|jkrtd|tjddS)z]Retrieves the downloadable files as a map of file names and their corresponding URLs.se:downloadsEnabledCYou must enable downloads in order to work with downloadable files.rnames)r+rrrZGET_DOWNLOADABLE_FILESrqr(r(r)get_downloadable_filess z WebDriver.get_downloadable_files) file_nametarget_directoryrHc Csd|jkrtdtj|s(t||tjd|idd}t `}tj ||d}t |d}| t|W5QRXt|d}||W5QRXW5QRXd S) zDownloads a file with the specified file name to the target directory. file_name: The name of the file to download. target_directory: The path to the directory to save the downloaded file. r@rArrcontentsz.ziprrN)r+rospathexistsmakedirsrrZ DOWNLOAD_FILEtempfileTemporaryDirectoryrrrbase64rzipfileZipFile extractall)rlrDrErFZtmp_dirzip_filefileZzip_refr(r(r) download_files     zWebDriver.download_filecCs"d|jkrtd|tjdS)zDeletes all downloadable files.r@rAN)r+rrrZDELETE_DOWNLOADABLE_FILESrqr(r(r)delete_downloadable_filess z#WebDriver.delete_downloadable_files)r[TNN)N)N)N)r)r)r)r)NNNN)vrUrVrWrXr%rr#rr r`r!boolr rrr rmrrrttypingType BaseExceptiontypes TracebackTyperyrr|propertyr r}rrhrrrirrrrr:rr"rrrrrrrrrxrrrrrrrr$rrrrrrrrrrrrrrrrsetterrrrrr+rrrarrrrr r rrrr\rrrr!rr#rr$rrr1r2rr4rr6r9r;r<r=r?rCrTrUr(r(r(r)rZs   1                               rZ)F)LrXrNrr.rHrrLrZrWrrOabcrrrrr importlibrrr r r Zselenium.common.exceptionsr r rrrZ%selenium.webdriver.common.bidi.scriptrZselenium.webdriver.common.byrZ!selenium.webdriver.common.optionsrZ,selenium.webdriver.common.print_page_optionsrZ"selenium.webdriver.common.timeoutsrZ/selenium.webdriver.common.virtual_authenticatorrrrZ+selenium.webdriver.support.relative_locatorrr#rrrZ errorhandlerrr\rrr}r Zremote_connectionr!rr"Z shadowrootr#rr$Z webelementr%Zwebsocket_connectionr&r'rr*r4rErSrTrZr(r(r(r)sl                                     "