U g~@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl 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.m/Z/ddl0m1Z1dd l2m3Z3dd!l4m5Z5dd"l6m7Z7dd#l8m9Z9dd$l:m;Z;dd%l:mZ>dd'l?m@Z@dd(lAmBZBdd)lCmDZDdd*lEmFZFdd+lGmHZHdd,lImJZJdd-lKmLZLdaMdaNd.d/ZOd0d1ZPd=eQeeReBfeSeSee3eBd2d3d4ZTee#ed5d6d7ZUGd8d9d9e d:ZVGd;d<dzThe WebDriver implementation.N)ABCMeta) b64decode)urlsafe_b64encode)asynccontextmanager)contextmanager) import_module)Dict)List)Optional)Union)InvalidArgumentException)JavascriptException)NoSuchCookieException)NoSuchElementException)WebDriverException)Script)By) ArgOptions) BaseOptions) PrintOptions)Timeouts) Credential)VirtualAuthenticatorOptions)required_virtual_authenticator) RelativeBy)Dialog)BidiConnection) ClientConfig)Command) ErrorHandler)FedCM) FileDetector)LocalFileDetector)LocatorConverter)Mobile)RemoteConnection) ScriptKey) ShadowRoot)SwitchTo) WebElement)WebSocketConnectioncCsts tdadS)Nz"selenium.webdriver.common.bidi.cdp)cdprr.r.G/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/remote/webdriver.py import_cdpNsr0cCs8t|}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_capsTs  r:)r1command_executor keep_aliveignore_local_proxy client_configreturnc st|tr(|pt|d}||_t|d}ddlm}ddlm}ddl m }ddl m }||||g} t fdd | Dt} | ||||d S) N)remote_server_addr)r>r)ChromeRemoteConnection)EdgeRemoteConnection)FirefoxRemoteConnection)SafariRemoteConnectionc3s"|]}|jdkr|VqdS) browserNameN)Z browser_nameget).0cr1r.r/ xsz(get_remote_connection..)r@r<Z ignore_proxyr>) isinstancestrrr@r'Z+selenium.webdriver.chrome.remote_connectionrAZ)selenium.webdriver.edge.remote_connectionrBZ,selenium.webdriver.firefox.remote_connectionrCZ+selenium.webdriver.safari.remote_connectionrDnext) r1r;r<r=r>rArBrCrD candidateshandlerr.rIr/get_remote_connectiongs        rPoptionsr?c 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)Nr1rr3r2)appendto_capabilitieslenrangekeysupdater6) rRr1optsoptZ opts_sizeZsamesiesiZ min_indexZ first_keysZkysalwaysr8r9r.r.r/create_matchess.       r]c@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/r^sr^) metaclassc @seZdZdZeZeZdee e fe e e e eeeefe ee ee eddddZdd Zd d Zej ejeej eej ejd d dZeddZeedddZ ee dddZ!ddZ"ddZ#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/d4d5Z0e d6d7d8Z1ee dd9d:Z2ee dd;d<Z3ddd=d>Z4ddd?d@Z5ee ddAdBZ6eee ddCdDZ7dddEdFZ8dddGdHZ9dddIdJZ:de e;e dKdLdMZdddPdQZ?dddRdSZ@dddTdUZAee$ddVdWZBej ejCddXdYZDdddZd[ZEddd\d]ZFddd^d_ZGeHdd`dadbZIeHdd`dcddZJeHdd`dedfZKeeLddgdhZMeMjNdddidhZMeOjPdfe e edjdkdlZQeOjPdfe e eedjdmdnZRee$ddodpZSe ddqdrZTe ddsdtZUeVddudvZWe ddwdxZXde ddzd{d|ZYde e$dzd}d~ZZdeHeHe e$dddZ[de$dddZ\e ddzddZ]e$dddZ^de$dddZ_ee dddZ`e`jNddddZ`eddZaeajNddddZaeddZbddZcddZdeeddZfeddZgddZhddZiejddddZkee dddZlemddddZnemeoddddZpemeeodddZqemee erfddddZsemddddZteme ddddZue$dddZve e ddddZwddddZxeeydddZzee dddZ{ddZ|eddZ}dddĄZ~dS) 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)r;r< file_detectorrRlocator_converterweb_element_clsr>r?c Cst|trt|}d} n|}|j} ||_t|jttfrPt|||| |d|_d|_ d|_ i|_ i|_ t |_t||_t||_|pt|_|pt|_|p|j|_d|_|||t||_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 - (Deprecated) 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 - locator_converter - Custom locator converter to use. Defaults to None. - web_element_cls - Custom class to use for web elements. Defaults to WebElement. - client_config - Custom client configuration to use. Defaults to None. F)r;r<r=r>TN)rKlistr]rT_ignore_local_proxyr;rLbytesrPZ _is_remote session_idr7pinned_scriptsr! error_handlerr* _switch_tor&_mobiler$rfr%rg_web_element_cls_authenticator_id start_client start_sessionr"_fedcm_websocket_connection_script) selfr;r<rfrRrgrhr>r1rjr.r.r/__init__s<        zWebDriver.__init__cCs&dt|jdt|jd|jdS)N<.z (session="z")>)typer`r_rlrxr.r.r/__repr__szWebDriver.__repr__cCs|SNr.r}r.r.r/ __enter__szWebDriver.__enter__)exc_typeexc tracebackcCs |dSr)quit)rxrrrr.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)rKrf)rxZfile_detector_classargskwargsZ last_detectorr.r.r/file_detector_contexts   zWebDriver.file_detector_context)r?cCs|jSr)rpr}r.r.r/mobile!szWebDriver.mobilecCs d|jkr|jdStddS)zReturns the name of the underlying browser for this instance. :Usage: :: name = driver.name rEz1browserName not specified in session capabilitiesN)r7KeyErrorr}r.r.r/name%s  zWebDriver.namecCsdS)zxCalled before starting a new session. This method may be overridden to define custom startup behavior. Nr.r}r.r.r/rs2szWebDriver.start_clientcCsdS)zCalled after executing a quit command. This method may be overridden to define custom shutdown behavior. Nr.r}r.r.r/ stop_client9szWebDriver.stop_client)r1r?cCs6t|}|tj|d}|d|_|d|_dS)zCreates a new session with the desired capabilities. :Args: - capabilities - a capabilities dict to start the session with. value sessionIdr1N)r:executer Z NEW_SESSIONrFrlr7)rxr1r7responser.r.r/rtAs 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|]}|VqdSr) _wrap_valuerGitemr}r.r/rJXsz(WebDriver._wrap_value..)rKdictr6rrqid_shadowroot_clsri)rxrZ convertedkeyvalr.r}r/rMs      zWebDriver._wrap_value) element_idr?cCs |||S)z6Creates a web element with the specified `element_id`.)rq)rxrr.r.r/create_web_element[szWebDriver.create_web_elementcst|tr\d|kr |dSd|kr8|dS|D]\}}|||<q@|St|tr|tfdd|DS|S)Nrrc3s|]}|VqdSr) _unwrap_valuerr}r.r/rJisz*WebDriver._unwrap_value..)rKrrrr6rri)rxrrrr.r}r/r_s  zWebDriver._unwrap_value)driver_commandparamsr?cCsx||}|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)rrlr;rrnZcheck_responserrF)rxrrrr.r.r/rls    zWebDriver.execute)urlr?cCs|tjd|idS)z0Loads a web page in the current browser session.rN)rr GET)rxrr.r.r/rFsz WebDriver.getcCs|tjddS)zuReturns the title of the current page. :Usage: :: title = driver.title r)rr Z GET_TITLErFr}r.r.r/titles zWebDriver.title)scriptr?cCst|}||j|j<|S)zUStore common javascript scripts to be executed later by a unique hashable ID.)r(rmr)rxr script_keyZscript_key_instancer.r.r/ pin_scripts zWebDriver.pin_script)rr?cCsDz|j|jWn,tk r>td|d|jdYnXdS)z$Remove a pinned script from storage.zNo script with key: z existed in N)rmpoprr)rxrr.r.r/unpinszWebDriver.unpincCs t|jSr)rirmr}r.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 foundrrr) rKr(rmrrr rir ZW3C_EXECUTE_SCRIPTrrxrrZconverted_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)rir ZW3C_EXECUTE_SCRIPT_ASYNCrrr.r.r/execute_async_scriptszWebDriver.execute_async_scriptcCs|tjdS)znGets the URL of the current page. :Usage: :: driver.current_url r)rr ZGET_CURRENT_URLr}r.r.r/ current_urls zWebDriver.current_urlcCs|tjdS)zqGets the source of the current page. :Usage: :: driver.page_source r)rr ZGET_PAGE_SOURCEr}r.r.r/ page_sources zWebDriver.page_sourcecCs|tjdS)zcCloses the current window. :Usage: :: driver.close() N)rr ZCLOSEr}r.r.r/closeszWebDriver.closecCs*z|tjW5||jXdS)z|Quits the driver and closes every associated window. :Usage: :: driver.quit() N)rr;rrr ZQUITr}r.r.r/rszWebDriver.quitcCs|tjdS)zReturns the handle of the current window. :Usage: :: driver.current_window_handle r)rr ZW3C_GET_CURRENT_WINDOW_HANDLEr}r.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)rr ZW3C_GET_WINDOW_HANDLESr}r.r.r/window_handles s zWebDriver.window_handlescCstj}||ddS)z5Maximizes the current window that webdriver is using.N)r ZW3C_MAXIMIZE_WINDOWr)rxrr.r.r/maximize_windowszWebDriver.maximize_windowcCs|tjdS)ztd|j|dS|tj ||ddS)zFind an element given a By strategy and locator. :Usage: :: element = driver.find_element(By.ID, 'foo') :rtype: WebElement )byrz%Cannot locate relative element with: rZusingrr) rgconvertrKr find_elementsrrootrr Z FIND_ELEMENT)rxrrelementsr.r.r/ find_elements  zWebDriver.find_elementcCs~|j||\}}t|trbdtddd}t|d d}d|d}| || S| t j||dd p|gS) zFind elements given a By strategy and locator. :Usage: :: elements = driver.find_elements(By.CLASS_NAME, 'foo') :rtype: list of WebElement r{NzfindElements.jsutf8z/* findElements */return (z).apply(null, arguments);rr)rgrrKrjoinr_splitpkgutilget_datadecoderrrr Z FIND_ELEMENTS)rxrrZ_pkgZ raw_functionZfind_element_jsr.r.r/rs   zWebDriver.find_elementscCs|jS)z4Returns the drivers current capabilities being used.)r7r}r.r.r/r1szWebDriver.capabilitiesc Csxt|ds"tjdtdd|}zDz$t|d}||W5QRXWnt k rjYW dSXW5~XdS)aSaves a screenshot of the current window to a PNG image file. Returns False if there is any IOError, else returns True. Use full paths in your filename. :Args: - filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. :Usage: :: driver.get_screenshot_as_file('/Screenshots/foo.png') z.pngz^name used for saved screenshot does not match file type. It should end with a `.png` extensionr stacklevelwbFT) rLlowerendswithwarningswarn UserWarningget_screenshot_as_pngopenwriteOSError)rxfilenameZpngfr.r.r/get_screenshot_as_files z WebDriver.get_screenshot_as_filecCs ||S)aSaves a screenshot of the current window to a PNG image file. Returns False if there is any IOError, else returns True. Use full paths in your filename. :Args: - filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. :Usage: :: driver.save_screenshot('/Screenshots/foo.png') )r)rxrr.r.r/save_screenshot=szWebDriver.save_screenshotcCst|dS)zGets the screenshot of the current window as a binary data. :Usage: :: driver.get_screenshot_as_png() ascii)rget_screenshot_as_base64encoder}r.r.r/rMszWebDriver.get_screenshot_as_pngcCs|tjdS)zGets the screenshot of the current window as a base64 encoded string which is useful in embedded images in HTML. :Usage: :: driver.get_screenshot_as_base64() r)rr Z SCREENSHOTr}r.r.r/rWs z"WebDriver.get_screenshot_as_base64current) windowHandler?cCs$|||jt|t|ddS)a$Sets the width and height of the current window. (window.resizeTo) :Args: - width: the width in pixels to set the window to - height: the height in pixels to set the window to :Usage: :: driver.set_window_size(800,600) widthheightN"_check_if_window_handle_is_currentset_window_rectr)rxrrrr.r.r/set_window_sizebs zWebDriver.set_window_sizecs8|||ddr&dfdddDS)zGets the width and height of the current window. :Usage: :: driver.get_window_size() rNcsi|]}||qSr.r.rGr8sizer.r/ sz-WebDriver.get_window_size..r)rget_window_rectrFrxrr.rr/get_window_sizeqs  zWebDriver.get_window_size)xyrr?cCs |||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) r r r)rxr r rr.r.r/set_window_positions 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 )rrr r.rr/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)rrr r.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)rr ZGET_WINDOW_RECTr}r.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)r r rrr)r rr ZSET_WINDOW_RECT)rxr r rrr.r.r/rszWebDriver.set_window_rectcCs|jSr)_file_detectorr}r.r.r/rfszWebDriver.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)rrKr#r)rxdetectorr.r.r/rfs  cCs|tjdS)zGets the current orientation of the device. :Usage: :: orientation = driver.orientation r)rr ZGET_SCREEN_ORIENTATIONr}r.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)upperrr ZSET_SCREEN_ORIENTATIONr)rxrZallowed_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)rr ZGET_AVAILABLE_LOG_TYPESr}r.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') r|r)rr ZGET_LOG)rxZlog_typer.r.r/get_logszWebDriver.get_logcCs|jrt|jfStts|jdrJ|jd}|jddd}n |\}}|sbtdt |a|jd dkrt j dt d d t||_|jtj}|dj}|jtj|d }||j_t|jfSdS) Nse:cdp se:cdpVersionr{r2Unable to find url to connect to from capabilitiesrEZfirefoxznCDP support for Firefox is deprecated and will be removed in future versions. Please switch to WebDriver BiDi.rrT)rvdevtoolsr0r7rFr_get_cdp_detailsrr-import_devtoolsrrrDeprecationWarningr,rtarget get_targets target_idZattach_to_targetrl)rxws_urlversiontargetsr!sessionr.r.r/start_devtoolss.       zWebDriver.start_devtoolsc Cst|jdr6|jd}|jddd}n |\}}|sNtdt|}t|4IdHX}| |j IdH}|dj }| |4IdH}t|t|VW5QIdHRXW5QIdHRXdS)Nrrr{rr)r0r7rFrrrr-rZopen_cdprrr r!Z open_sessionr)rxr"r#rconnr$r!r%r.r.r/bidi_connection1s     zWebDriver.bidi_connectioncCs&|js||js t|j|_|jSr)rv _start_bidirwrr}r.r.r/rEs  zWebDriver.scriptcCs0|jdr|jd}ntdt||_dS)NZ webSocketUrlr)r7rFrr,rv)rxr"r.r.r/r)Os 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)NrFrEZchromezgoog:chromeOptionsZdebuggerAddressZ MicrosoftEdgezms:edgeOptionsTzmoz:debuggerAddressrzhttp://z /json/versionZBrowserZwebSocketDebuggerUrlUz .*/(\d+)\.r) jsonurllib3 PoolManagerr7rFrequestloadsdataresearchgroup) rxr+r,httpZ_firefoxZdebugger_addressresr0Zbrowser_versionZ websocket_urlr1r#r.r.r/rWs&    zWebDriver._get_cdp_detailsrQcCs|tj|d|_dS)z4Adds a virtual authenticator with the given options.rN)rr ZADD_VIRTUAL_AUTHENTICATORrrr)rxrRr.r.r/add_virtual_authenticatorwsz#WebDriver.add_virtual_authenticatorcCs|jS)z,Returns the id of the virtual authenticator.)rrr}r.r.r/virtual_authenticator_id{sz"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)rr ZREMOVE_VIRTUAL_AUTHENTICATORrrr}r.r.r/remove_virtual_authenticatorsz&WebDriver.remove_virtual_authenticator) credentialr?cCs |tj|d|jidS)z,Injects a credential into the authenticator.r8N)rr ZADD_CREDENTIALrrr)rxr:r.r.r/add_credentialszWebDriver.add_credentialcCs&|tjd|ji}dd|dDS)z;Returns the list of credentials owned by the authenticator.r8cSsg|]}t|qSr.)r from_dict)rGr:r.r.r/ sz-WebDriver.get_credentials..r)rr ZGET_CREDENTIALSrr)rxZcredential_datar.r.r/get_credentialsszWebDriver.get_credentials) credential_idr?cCs0t|trt|}|tj||jddS)z,Removes a credential from the authenticator.)Z credentialIdr8N)rK bytearrayrrrr ZREMOVE_CREDENTIALrr)rxr?r.r.r/remove_credentials    zWebDriver.remove_credentialcCs|tjd|jidS)z/Removes all credentials from the authenticator.r8N)rr ZREMOVE_ALL_CREDENTIALSrrr}r.r.r/remove_all_credentialssz WebDriver.remove_all_credentials)verifiedr?cCs|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. )r8ZisUserVerifiedN)rr ZSET_USER_VERIFIEDrr)rxrCr.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)r1rrr ZGET_DOWNLOADABLE_FILESr}r.r.r/get_downloadable_filess z WebDriver.get_downloadable_files) file_nametarget_directoryr?c 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. rErFrrcontentsz.ziprrN)r1rospathexistsmakedirsrr Z DOWNLOAD_FILEtempfileTemporaryDirectoryrrrbase64rzipfileZipFile extractall)rxrIrJrKZtmp_dirzip_filefileZzip_refr.r.r/ download_files     zWebDriver.download_filecCs"d|jkrtd|tjdS)zDeletes all downloadable files.rErFN)r1rrr ZDELETE_DOWNLOADABLE_FILESr}r.r.r/delete_downloadable_filess z#WebDriver.delete_downloadable_filescCs|jS)aw :Returns: - FedCM: an object providing access to all Federated Credential Management (FedCM) dialog commands. :Usage: :: title = driver.fedcm.title subtitle = driver.fedcm.subtitle dialog_type = driver.fedcm.dialog_type accounts = driver.fedcm.account_list driver.fedcm.select_account(0) driver.fedcm.accept() driver.fedcm.dismiss() driver.fedcm.enable_delay() driver.fedcm.disable_delay() driver.fedcm.reset_cooldown() )rur}r.r.r/fedcmszWebDriver.fedcmcCs|jtjdS)z8Returns whether the browser supports FedCM capabilities.F)r1rFrZFEDCM_CAPABILITYr}r.r.r/supports_fedcmszWebDriver.supports_fedcmcCs|jstddS)z.Raises an exception if FedCM is not supported.znThis browser does not support Federated Credential Management. Please ensure you're using a supported browser.N)r\rr}r.r.r/_require_fedcm_supportsz WebDriver._require_fedcm_supportcCs|t|S)z0Returns the FedCM dialog object for interaction.)r]rr}r.r.r/dialogszWebDriver.dialog?cs^ddlmddlm}|dkr.f}fdd||||d}|fdd S) aWaits for and returns the FedCM dialog. Args: timeout: How long to wait for the dialog poll_frequency: How frequently to poll ignored_exceptions: Exceptions to ignore while waiting Returns: The FedCM dialog object if found Raises: TimeoutException if dialog doesn't appear WebDriverException if FedCM not supported r)NoAlertPresentException) WebDriverWaitNcs4zt}|jr|ndWSk r.YdSXdSr)rr|)r^)rarxr.r/ _check_fedcms z,WebDriver.fedcm_dialog.._check_fedcm)poll_frequencyignored_exceptionscsSrr.)r)rcr.r/z(WebDriver.fedcm_dialog..)selenium.common.exceptionsraZselenium.webdriver.support.waitrbr]Zuntil)rxtimeoutrdrerbwaitr.)rarcrxr/ fedcm_dialogs  zWebDriver.fedcm_dialog)reTNNNNN)N)N)N)r)r)r)r)NNNN)r_r`N)r_r`rarbr+rqr)rr rLr'boolr r#rr r%r|rryr~rtypingType BaseExceptiontypes TracebackTyperrrpropertyr&rrrsrrrtrrrrrFrr(rrrrrrrrrrrrrrrrr*rrrrrrrrrrrrrrrrsetterrZIDrrr1rrrkrrrr rrrrrrfrrrr&rr(rr)rrr6r7rr9rr;r>r@rArBrDrHrYrZr"r[r\r]r^rkr.r.r.r/rds   ;                                 rd)N)XrbrSrr4rMrrQrprmrrTabcrrrrr importlibrrr r r rhr r rrrZ%selenium.webdriver.common.bidi.scriptrZselenium.webdriver.common.byrZ!selenium.webdriver.common.optionsrrZ,selenium.webdriver.common.print_page_optionsrZ"selenium.webdriver.common.timeoutsrZ/selenium.webdriver.common.virtual_authenticatorrrrZ+selenium.webdriver.support.relative_locatorrZcommon.fedcm.dialogrr(rr>rrr Z errorhandlerr!r[r"rfr#r$rgr%rr&Zremote_connectionr'rr(Z shadowrootr)rr*Z webelementr+Zwebsocket_connectionr,r-rr0r:rrLrlrPr]r^rdr.r.r.r/s                                           "