a xd#@sHdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl 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 l0m2Z2dd!l3m4Z4dd"l5m6Z6dd#l7m8Z8dd$l9m:Z:dd%l;mZ>e?gd'Z@d(d)d*d+ZAdaBd,d-ZCd.d/ZDd;d1d2ZEeeed3d4d5ZFGd6d7d7ed8ZGGd9d:d:eGZHdS)).0c capabilitiesr0r1 z(get_remote_connection..) keep_aliveZ ignore_proxy)Z-selenium.webdriver.chromium.remote_connectionrJZ,selenium.webdriver.firefox.remote_connectionrKZ+selenium.webdriver.safari.remote_connectionrLr"next) rPcommand_executorrSignore_local_proxyrJrKrL candidateshandlerr0rOr1get_remote_connections     rYoptionsreturnc Csdii}g}|D]}||qt|}i}t|D]p}|}|d|kr8||}|D]J} | ||dvr\||| ||d| kr\|| ||| iq\q8i} |D]\} } | | | <q|D]}| D] } || =qq| |dd<||dd<|S)NrPrr;r:)appendto_capabilitieslenrangekeysupdater@) r[rPoptsoptZ opts_sizeZsamesiesiZ min_indexZ first_keysZkysalwaysrGrHr0r0r1create_matchess.       rgc@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__r0r0r0r1rhsrh) metaclassc@seZdZdZeZeZdee e e fddddZ dd Z d d Z ejejeejeejejd d dZeddZeedddZeedddZddZddZdeddddZddZ eedd d!Z!d"d#Z"deeed$d%d&Z#edd'd(d)Z$eedd*d+Z%dee&d,d-d.Z'e&dd/d0d1Z(e edd2d3Z)d4d5Z*ed6d7d8Z+eedd9d:Z,eedd;d<Z-ddd=d>Z.ddd?d@Z/eeddAdBZ0ee eddCdDZ1dddEdFZ2dddGdHZ3dddIdJZ4dee5edKdLdMZ6ee7ddNdOZ8dddPdQZ9dddRdSZ:dddTdUZ;e eddVdWZdddZd[Z?ddd\d]Z@ddd^d_ZAeBdd`dadbZCeBdd`dcddZDeBdd`dedfZEeeFddgdhZGeGjHdddidhZGeIjJdfeeedjdkdlZKeIjJdfeee edjdmdnZLeeddodpZMeeddqdrZNeOddsdtZPeOddudvZQeRddwdxZSeddydzZTdedd|d}d~ZUdeed|ddZVdeed|ddZWdedddZXedddZYdedddZZee[dddZ\e\jHddddZ\eddZ]e]jHddddZ]eddZ^eddZ_ddZ`eaddZbddZcedddddZeeedddZfegddddZhegeiddddZjege eidddZkegeeelfddddZmegddddZnegeOddddZodS) 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://github.com/SeleniumHQ/selenium/wiki/JsonWireProtocol. :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://github.com/SeleniumHQ/selenium/wiki/DesiredCapabilities - command_executor - remote_connection.RemoteConnection object used to execute commands. - error_handler - errorhandler.ErrorHandler object used to handle errors. http://127.0.0.1:4444NTrZc Cs&|rtjdtdd|r(tjdtdd|r)typerjrirvrr0r0r1__repr__ szWebDriver.__repr__cCs|SNr0rr0r0r1 __enter__#szWebDriver.__enter__)exc_typeexc tracebackcCs |dSr)quit)rrrrr0r0r1__exit__&szWebDriver.__exit__cosLd}t|j|s&|j}||i||_zdVW|rH||_n |rF||_0dS)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)rpr{)rZfile_detector_classargskwargsZ last_detectorr0r0r1file_detector_context.s zWebDriver.file_detector_context)r\cCs|jSr)rzrr0r0r1mobileJszWebDriver.mobilecCs d|jvr|jdStddS)zReturns the name of the underlying browser for this instance. :Usage: :: name = driver.name r(z1browserName not specified in session capabilitiesN)rFKeyErrorrr0r0r1nameNs  zWebDriver.namecCsdS)zxCalled before starting a new session. This method may be overridden to define custom startup behavior. Nr0rr0r0r1r}[szWebDriver.start_clientcCsdS)zCalled after executing a quit command. This method may be overridden to define custom shutdown behavior. Nr0rr0r0r1 stop_clientbszWebDriver.stop_client)rPr\cCst|tstd|r>d|vr.|j|dd<n|d|jit|}d|i}|tj|}d|vrl|d}|d|_ | d|_ |j s| d|_ dS) aCreates a new session with the desired capabilities. :Args: - capabilities - a capabilities dict to start the session with. - browser_profile - A selenium.webdriver.firefox.firefox_profile.FirefoxProfile object. Only used if Firefox is requested. z!Capabilities must be a dictionaryr8r9r3rP sessionIdvalueN) rprsr encodedrbrIexecuterZ NEW_SESSIONrvr>rF)rrPrZw3c_caps parametersresponser0r0r1r~js   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_valuerMitemrr0r1rQrRz(WebDriver._wrap_value..)rprsr@r_web_element_clsid_shadowroot_clsrq)rrZ convertedkeyvalr0rr1rs      zWebDriver._wrap_value) element_idr\cCs |||S)z6Creates a web element with the specified `element_id`.)r)rrr0r0r1create_web_elementszWebDriver.create_web_elementcst|tr\d|vr |dSd|vr8|dS|D]\}}|||<q@|St|tr|tfdd|DS|S)Nrrc3s|]}|VqdSr) _unwrap_valuerrr0r1rQrRz*WebDriver._unwrap_value..)rprsrrr@rrq)rrrrr0rr1rs  zWebDriver._unwrap_value)driver_commandparamsr\cCsx||}|jr2|s d|ji}nd|vr2|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)rrvrUrrxZcheck_responserr>)rrrrr0r0r1rs    zWebDriver.execute)urlr\cCs|tjd|idS)z0Loads a web page in the current browser session.rN)rrGET)rrr0r0r1r>sz WebDriver.getcCs|tjddS)zuReturns the title of the current page. :Usage: :: title = driver.title r)rrZ GET_TITLEr>rr0r0r1titles zWebDriver.title)scriptr\cCst|}||j|j<|S)zUStore common javascript scripts to be executed later by a unique hashable ID.)r#rwr)rr script_keyZscript_key_instancer0r0r1 pin_scripts zWebDriver.pin_script)rr\cCsBz|j|jWn*ty<td|d|jdYn0dS)z$Remove a pinned script from storage.zNo script with key: z existed in N)rwpoprr)rrr0r0r1unpins zWebDriver.unpincCs t|jSr)rqrwrr0r0r1get_pinned_scriptsszWebDriver.get_pinned_scriptscGsZt|tr6z|j|j}Wnty4tdYn0t|}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) rpr#rwrrr rqrZW3C_EXECUTE_SCRIPTrrrrZconverted_argscommandr0r0r1execute_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)rqrZW3C_EXECUTE_SCRIPT_ASYNCrrr0r0r1execute_async_scriptszWebDriver.execute_async_scriptcCs|tjdS)znGets the URL of the current page. :Usage: :: driver.current_url r)rrZGET_CURRENT_URLrr0r0r1 current_url s zWebDriver.current_urlcCs|tjdS)zqGets the source of the current page. :Usage: :: driver.page_source r)rrZGET_PAGE_SOURCErr0r0r1 page_sources zWebDriver.page_sourcecCs|tjdS)zcCloses the current window. :Usage: :: driver.close() N)rrZCLOSErr0r0r1closeszWebDriver.closecCs<z"|tjW||jn||j0dS)z|Quits the driver and closes every associated window. :Usage: :: driver.quit() N)rrZQUITrrUrrr0r0r1r)s  zWebDriver.quitcCs|tjdS)zReturns the handle of the current window. :Usage: :: driver.current_window_handle r)rrZW3C_GET_CURRENT_WINDOW_HANDLErr0r0r1current_window_handle7s zWebDriver.current_window_handlecCs|tjdS)zReturns the handles of all windows within the current session. :Usage: :: driver.window_handles r)rrZW3C_GET_WINDOW_HANDLESrr0r0r1window_handlesBs zWebDriver.window_handlescCstj}||ddS)z5Maximizes the current window that webdriver is using.N)rZW3C_MAXIMIZE_WINDOWr)rrr0r0r1maximize_windowMszWebDriver.maximize_windowcCs|tjdS)zrRz-WebDriver.get_window_size..r)rBrCget_window_rectr>rrr0rr1get_window_sizes   zWebDriver.get_window_sizecCs(|dkrtd|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) rrxyr )rrrrr0r0r1set_window_positions  zWebDriver.set_window_positioncs,|dkrtd|fdddDS)zGets the x,y position of the current window. :Usage: :: driver.get_window_position() rrcsi|]}||qSr0r0rpositionr0r1rrRz1WebDriver.get_window_position..r)rBrCrrr0rr1get_window_positions  zWebDriver.get_window_positioncCs|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_RECTrr0r0r1rs zWebDriver.get_window_rectcCs<|dur |dur |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)rrr r r)r rrZSET_WINDOW_RECT)rrrr r r0r0r1r szWebDriver.set_window_rectcCs|jSr)_file_detectorrr0r0r1r{ 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)rrprr)rdetectorr0r0r1r{s  cCs|tjdS)zGets the current orientation of the device. :Usage: :: orientation = driver.orientation r)rrZGET_SCREEN_ORIENTATIONrr0r0r1 orientation!s zWebDriver.orientationcCs4ddg}||vr(|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)rrZallowed_valuesr0r0r1r,s  cCst|S)zQReturns a ApplicationCache Object to interact with the browser app cache.rrr0r0r1application_cache>szWebDriver.application_cachecCs|tjdS)zGets a list of the available log types. This only works with w3c compliant browsers. :Usage: :: driver.log_types r)rrZGET_AVAILABLE_LOG_TYPESrr0r0r1 log_typesDs 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') rr)rrZGET_LOG)rZlog_typer0r0r1get_logPszWebDriver.get_logc Cst|jdr6|jd}|jddd}n |\}}|sNtdt|}t|4IdH}| |j IdH}|dj }| |4IdH$}t|t|VWdIdHq1IdHs0YWdIdHq1IdHs0YdS)Nzse:cdpz se:cdpVersionrrz2Unable to find url to connect to from capabilities)r2rFr>r_get_cdp_detailsrr/Zimport_devtoolsZopen_cdprtargetZ get_targets target_idZ open_sessionr)rZws_urlr-Zdevtoolsconntargetsr%sessionr0r0r1bidi_connection`s     zWebDriver.bidi_connectionc Csddl}ddl}|}d}|jddkrV|j|jd|jddd}nd}|jd }|d d |d }||j}|d }|d} ddl } |rd} n| d| d} | | fS)NrFr(Zchromer7OptionsZdebuggerAddressTzmoz:debuggerAddressrzhttp://z /json/versionZBrowserZwebSocketDebuggerUrlUz .*/(\d+)\.r) jsonurllib3 PoolManagerrFr>Z vendor_prefixrequestloadsdataresearchgroup) rr,r-httpZ_firefoxZdebugger_addressresr1Zbrowser_versionZ websocket_urlr2r-r0r0r1r#ts&"    zWebDriver._get_cdp_detailscCs|tj|d|_dS)z4Adds a virtual authenticator with the given options.rN)rrZADD_VIRTUAL_AUTHENTICATORrr|)rr[r0r0r1add_virtual_authenticatorsz#WebDriver.add_virtual_authenticatorcCs|jS)z,Returns the id of the virtual authenticator.)r|rr0r0r1virtual_authenticator_idsz"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_AUTHENTICATORr|rr0r0r1remove_virtual_authenticatorsz&WebDriver.remove_virtual_authenticator) credentialr\cCs$|tji|d|jidS)z,Injects a credential into the authenticator.r9N)rrZADD_CREDENTIALrr|)rr;r0r0r1add_credentialszWebDriver.add_credentialcCs&|tjd|ji}dd|dDS)z;Returns the list of credentials owned by the authenticator.r9cSsg|]}t|qSr0)r from_dict)rMr;r0r0r1 rRz-WebDriver.get_credentials..r)rrZGET_CREDENTIALSr|)rZcredential_datar0r0r1get_credentialsszWebDriver.get_credentials) credential_idr\cCs0t|trt|}|tj||jddS)z,Removes a credential from the authenticator.)Z credentialIdr9N)rp bytearrayrrrrZREMOVE_CREDENTIALr|)rr@r0r0r1remove_credentials   zWebDriver.remove_credentialcCs|tjd|jidS)z/Removes all credentials from the authenticator.r9N)rrZREMOVE_ALL_CREDENTIALSr|rr0r0r1remove_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. )r9ZisUserVerifiedN)rrZSET_USER_VERIFIEDr|)rrDr0r0r1set_user_verifiedszWebDriver.set_user_verified)roNNNTNN)N)N)N)N)r)r)r)r)NNNN)prirjrkrlr&rr$rr rr rrrtypingr Type BaseExceptiontypes TracebackTyperrrpropertyr!rrtrr}rrsr~rrrrr>rr#rrrrrrrrrrrrrrrrr%rrrrrrrrrrrrrrrr,setterrrrrrrPboolrrrurrr rrrrr rr{rr r!r"rr)r#rr7r8rr:rr<r?rArBrCrEr0r0r0r1rns  V                              rn)F)Irlrr<rrIrFrBabcrbase64rrrr importlibrrr r r Zselenium.common.exceptionsr r rrrZselenium.webdriver.common.byrZ1selenium.webdriver.common.html5.application_cacherZ!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{rr rr!Zremote_connectionr"rr#Z shadowrootr$rr%Z webelementr& frozensetrErAr/r2rIrYrgrhrnr0r0r0r1sh                                    ) "