o Vh@s>dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddlm Z ddlm Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddl 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/ddl0m1Z1ddl0m2Z2dd l3m4Z4dd!l5m6Z6dd"l7m8Z8dd#l9m:Z:dd$l;mZ>da?d&d'Z@d(d)ZAd6d+d,ZBd-eed.efd/d0ZCGd1d2d2e d3ZDGd4d5d5eDZEdS)7zThe WebDriver implementation.N)ABCMeta) b64decode)urlsafe_b64encode)asynccontextmanager)contextmanager) import_module)Dict)List)Optional)Union)InvalidArgumentException)JavascriptException)NoSuchCookieException)NoSuchElementException)WebDriverException)By) BaseOptions) PrintOptions)Timeouts) Credential)VirtualAuthenticatorOptions)required_virtual_authenticator) RelativeBy)BidiConnection)Command) ErrorHandler) FileDetector)LocalFileDetector)Mobile)RemoteConnection) ScriptKey) ShadowRoot)SwitchTo) WebElementcCststdadSdS)Nz"selenium.webdriver.common.bidi.cdp)cdprr&r&`/home/air/yokohama/back/venv/lib/python3.10/site-packages/selenium/webdriver/remote/webdriver.py import_cdpEs r(cCs8t|}i}|D]\}}|||<q dig|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)caps always_matchkvr&r&r' _create_capsKs  r3Fc sbddlm}ddlm}ddlm}ddlm}||||g}tfdd|Dt } | |||dS) Nr)ChromeRemoteConnection)EdgeRemoteConnection)FirefoxRemoteConnection)SafariRemoteConnectionc3s$|] }|jdkr|VqdS) browserNameN) browser_nameget).0cr)r&r' es"z(get_remote_connection..) keep_alive ignore_proxy) +selenium.webdriver.chrome.remote_connectionr4)selenium.webdriver.edge.remote_connectionr5,selenium.webdriver.firefox.remote_connectionr6+selenium.webdriver.safari.remote_connectionr7nextr ) r)command_executorr?ignore_local_proxyr4r5r6r7 candidateshandlerr&r=r'get_remote_connection^s     rJoptionsreturnc Csdii}g}|D] }||qt|}i}t|D]8}|}|d|krT||}|D]%} | ||dvrS||| ||d| krS|| ||| iq.qi} |D]\} } | | | <q[|D] }| D]} || =qjqf| |dd<||dd<|S)Nr)rr+r*)appendto_capabilitieslenrangekeysupdater.) rKr)optsopt opts_sizesamesiesi min_index first_keyskysalwaysr1r2r&r&r'create_matchesjs2      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@seZdZdZeZeZ    ddee e e fddfddZ d d Z d d Z d ejejedejedejejfddZeddZedefddZedefddZddZddZdeddfddZdd Z d!edefd"d#Z!d$d%Z"dd&ed'edefd(d)Z#d*eddfd+d,Z$edefd-d.Z%dd/ede&fd0d1Z'd2e&ddfd3d4Z(de efd5d6Z)d7d8Z*d/efd9d:Z+edefd;d<Z,edefd=d>Z-dd?d@Z.ddAdBZ/edefdCdDZ0ede efdEdFZ1ddGdHZ2ddIdJZ3ddKdLZ4ddMee5defdNdOZ6ede7fdPdQZ8ddRdSZ9ddTdUZ:ddVdWZ;de efdXdYZdd\d]Z?dd^d_Z@dd`daZAdbeBddfdcddZCdbeBddfdedfZDdbeBddfdgdhZEedeFfdidjZGeGjHddkdjZGeIjJdfdleedefdmdnZKeIjJdfdleede efdodpZLedefdqdrZMdeNfdsdtZOdeNfdudvZPdeQfdwdxZRdefdydzZSdd|eddfd}d~ZTdd|edefddZUdd|edefddZVddefddZWd|eddfddZXdefddZYddefddZZede[fddZ\e\jHdddZ\eddZ]e]jHdddZ]eddZ^ddZ_e`ddZaddZbdecddfddZdedefddZeefdddZgefdehddfddZiefde ehfddZjefdeeekfddfddZlefdddZmefdeNddfddZndefddZodededdfddZpdddZqdS) 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:4444TNrKrLcCst|tr t|}d}n|}|j}||_t|jttfr't||||d|_d|_ d|_ i|_ i|_ t |_t||_t||_|pEt|_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)rFr?rGTN) isinstancelistr\rN_ignore_local_proxyrFstrbytesrJ _is_remote session_idr/pinned_scriptsr error_handlerr# _switch_tor_mobiler file_detector_authenticator_id start_client start_session)selfrFr?rprKr)rgr&r&r'__init__s0    zWebDriver.__init__cCs&dt|jdt|jd|jdS)N<.z (session="z")>)typer_r^rkrtr&r&r'__repr__s&zWebDriver.__repr__cCs|SNr&ryr&r&r' __enter__szWebDriver.__enter__exc_typeexc tracebackcCs |dSr{)quit)rtr}r~rr&r&r'__exit__s zWebDriver.__exit__cosPd}t|j|s|j}||i||_z dVW|r ||_dSdS|r'||_w)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)rerp)rtfile_detector_classargskwargs last_detectorr&r&r'file_detector_contexts  zWebDriver.file_detector_contextcC|jSr{)roryr&r&r'mobilezWebDriver.mobilecCsd|jvr |jdStd)zReturns the name of the underlying browser for this instance. :Usage: :: name = driver.name r8z1browserName not specified in session capabilities)r/KeyErrorryr&r&r'names zWebDriver.namecCdS)zxCalled before starting a new session. This method may be overridden to define custom startup behavior. Nr&ryr&r&r'rr szWebDriver.start_clientcCr)zCalled after executing a quit command. This method may be overridden to define custom shutdown behavior. Nr&ryr&r&r' stop_clientszWebDriver.stop_clientr)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 sessionIdr)N)r3executer NEW_SESSIONr:rkr/)rtr)r/responser&r&r'rss zWebDriver.start_sessioncst|tri}|D] \}}|||<q |St|jr$d|jiSt|jr/d|jiSt|tr?tfdd|DS|S)N#element-6066-11e4-a52e-4f735466cecf"shadow-6066-11e4-a52e-4f735466cecfc3|]}|VqdSr{) _wrap_valuer;itemryr&r'r>3z(WebDriver._wrap_value..)redictr.r_web_element_clsid_shadowroot_clsrf)rtr convertedkeyvalr&ryr'r(s      zWebDriver._wrap_value element_idcCs |||S)z6Creates a web element with the specified `element_id`.)r)rtrr&r&r'create_web_element6s zWebDriver.create_web_elementcst|tr.d|vr|dSd|vr|dS|D] \}}|||<q |St|tr>tfdd|DS|S)Nrrc3rr{) _unwrap_valuerryr&r'r>Drz*WebDriver._unwrap_value..)rerrrr.rrf)rtrrrr&ryr'r:s  zWebDriver._unwrap_valuedriver_commandparamscCsx||}|jr|sd|ji}n d|vr|j|d<|j||}|r5|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)rrkrFrrmcheck_responserr:)rtrrrr&r&r'rGs    zWebDriver.executeurlcC|tjd|idS)z0Loads a web page in the current browser session.rN)rrGET)rtrr&r&r'r:bsz WebDriver.getcCs|tjddS)zuReturns the title of the current page. :Usage: :: title = driver.title r)rr GET_TITLEr:ryr&r&r'titlefs zWebDriver.titlescriptcCst|}||j|j<|S)zUStore common javascript scripts to be executed later by a unique hashable ID.)r!rlr)rtr script_keyscript_key_instancer&r&r' pin_scriptqs zWebDriver.pin_scriptrcCs<z |j|jWdStytd|d|jdw)z$Remove a pinned script from storage.zNo script with key: z existed in N)rlpoprr)rtrr&r&r'unpinxs  zWebDriver.unpincCs t|jSr{)rfrlryr&r&r'get_pinned_scriptss zWebDriver.get_pinned_scriptscGsVt|trz|j|j}Wn tytdwt|}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) rer!rlrrr rfrW3C_EXECUTE_SCRIPTrrtrrconverted_argscommandr&r&r'execute_scripts  zWebDriver.execute_scriptcGs$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)rfrW3C_EXECUTE_SCRIPT_ASYNCrrr&r&r'execute_async_scriptszWebDriver.execute_async_scriptcC|tjdS)znGets the URL of the current page. :Usage: :: driver.current_url r)rrGET_CURRENT_URLryr&r&r' current_url zWebDriver.current_urlcCr)zqGets the source of the current page. :Usage: :: driver.page_source r)rrGET_PAGE_SOURCEryr&r&r' page_sourcerzWebDriver.page_sourcecC|tjdS)zcCloses the current window. :Usage: :: driver.close() N)rrCLOSEryr&r&r'closezWebDriver.closecCs:z|tjW||jdS||jw)z|Quits the driver and closes every associated window. :Usage: :: driver.quit() N)rrQUITrrFrryr&r&r'rs  zWebDriver.quitcCr)zReturns the handle of the current window. :Usage: :: driver.current_window_handle r)rrW3C_GET_CURRENT_WINDOW_HANDLEryr&r&r'current_window_handlerzWebDriver.current_window_handlecCr)zReturns the handles of all windows within the current session. :Usage: :: driver.window_handles r)rrW3C_GET_WINDOW_HANDLESryr&r&r'window_handlesrzWebDriver.window_handlescCstj}||ddS)z5Maximizes the current window that webdriver is using.N)rW3C_MAXIMIZE_WINDOWr)rtrr&r&r'maximize_windowszWebDriver.maximize_windowcCr)zkz-WebDriver.get_window_size..r8)r<get_window_rectr:rtr7r&rAr'get_window_size\s  zWebDriver.get_window_sizecCs |||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) xyr;)rtrIrJr7r&r&r'set_window_positionms zWebDriver.set_window_positioncs$|||fdddDS)zGets the x,y position of the current window. :Usage: :: driver.get_window_position() cr?r&r&r@positionr&r'rCrDz1WebDriver.get_window_position..rH)r<rErFr&rLr'get_window_position|s zWebDriver.get_window_positioncCs|dkr tjddddSdS)z5Warns if the window handle is not equal to `current`.r6z?Only 'current' window is supported for W3C compatible browsers.r r!N)r&r'rFr&r&r'r<sz,WebDriver._check_if_window_handle_is_currentcCr)zGets the x, y coordinates of the window as well as height and width of the current window. :Usage: :: driver.get_window_rect() r)rrGET_WINDOW_RECTryr&r&r'rErzWebDriver.get_window_rectcCs<|dur|dur|s|std|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)rIrJr9r:r)r rrSET_WINDOW_RECT)rtrIrJr9r:r&r&r'r=szWebDriver.set_window_rectcCrr{)_file_detectorryr&r&r'rprzWebDriver.file_detectorcCs(|stdt|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)rrerrQ)rtdetectorr&r&r'rps    cCr)zGets the current orientation of the device. :Usage: :: orientation = driver.orientation r)rrGET_SCREEN_ORIENTATIONryr&r&r' orientationrzWebDriver.orientationcCs2ddg}||vr|tjd|idStd)zSets the current orientation of the device. :Args: - value: orientation to set it to. :Usage: :: driver.orientation = 'landscape' LANDSCAPEPORTRAITrTz>You can only set the orientation to 'LANDSCAPE' and 'PORTRAIT'N)upperrrSET_SCREEN_ORIENTATIONr)rtrallowed_valuesr&r&r'rTs cCr)zGets a list of the available log types. This only works with w3c compliant browsers. :Usage: :: driver.log_types r)rrGET_AVAILABLE_LOG_TYPESryr&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') rxr)rrGET_LOG)rtlog_typer&r&r'get_logszWebDriver.get_logc Cs t|jdr|jd}|jddd}n|\}}|s(tdt|}t|4IdHB}| |j IdH}|dj }| |4IdH}t|t|VWdIdHn 1IdHshwYWdIdHdS1IdHs~wYdS)Nzse:cdpz se:cdpVersionrwrz2Unable to find url to connect to from capabilities)r(r/r:r_get_cdp_detailsrr%import_devtoolsopen_cdprtarget get_targets target_id open_sessionr)rtws_urlversiondevtoolsconntargetsrdsessionr&r&r'bidi_connections      *.zWebDriver.bidi_connectionc Csddl}ddl}|}d}|jddkr |jdd}n|jddkr2|jdd}nd }|jd }|d d |d }||j}|d}|d} ddl} |r`d} | | fS| d| d} | | fS)NrFr8chromezgoog:chromeOptionsdebuggerAddressmsedgezms:edgeOptionsTzmoz:debuggerAddressrzhttp://z /json/versionBrowserwebSocketDebuggerUrlUz .*/(\d+)\.r) jsonurllib3 PoolManagerr/r:requestloadsdataresearchgroup) rtrsrthttp_firefoxdebugger_addressresrxbrowser_version websocket_urlryrgr&r&r'r_s(    zWebDriver._get_cdp_detailscCs|tj|d|_dS)z4Adds a virtual authenticator with the given options.rN)rrADD_VIRTUAL_AUTHENTICATORrrq)rtrKr&r&r'add_virtual_authenticator0sz#WebDriver.add_virtual_authenticatorcCr)z,Returns the id of the virtual authenticator.)rqryr&r&r'virtual_authenticator_id4rz"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)rrREMOVE_VIRTUAL_AUTHENTICATORrqryr&r&r'remove_virtual_authenticator9s z&WebDriver.remove_virtual_authenticator credentialcCs$|tji|d|jidS)z,Injects a credential into the authenticator.rN)rrADD_CREDENTIALrrq)rtrr&r&r'add_credentialCs$zWebDriver.add_credentialcCs&|tjd|ji}dd|dDS)z;Returns the list of credentials owned by the authenticator.rcSsg|]}t|qSr&)r from_dict)r;rr&r&r' LrDz-WebDriver.get_credentials..r)rrGET_CREDENTIALSrq)rtcredential_datar&r&r'get_credentialsHszWebDriver.get_credentials credential_idcCs0t|tr t|}|tj||jddS)z,Removes a credential from the authenticator.) credentialIdrN)re bytearrayrrrrREMOVE_CREDENTIALrq)rtrr&r&r'remove_credentialNs  zWebDriver.remove_credentialcCs|tjd|jidS)z/Removes all credentials from the authenticator.rN)rrREMOVE_ALL_CREDENTIALSrqryr&r&r'remove_all_credentialsYsz WebDriver.remove_all_credentialsverifiedcCs|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. )risUserVerifiedN)rrSET_USER_VERIFIEDrq)rtrr&r&r'set_user_verified^szWebDriver.set_user_verifiedcCs&d|jvr td|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)rrrGET_DOWNLOADABLE_FILESryr&r&r'get_downloadable_filesgs z WebDriver.get_downloadable_files file_nametarget_directorycCsd|jvr tdtj|st||tjd|idd}tj ||}t |d}| t |Wdn1s@wYt|d}||WddS1s\wYdS) 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. rrrrcontentsr#Nr)r)rospathexistsmakedirsrr DOWNLOAD_FILErr*r+base64rzipfileZipFile extractall)rtrrr target_filefilezip_refr&r&r' download_fileos     "zWebDriver.download_filecCs"d|jvr td|tjdS)zDeletes all downloadable files.rrN)r)rrrDELETE_DOWNLOADABLE_FILESryr&r&r'delete_downloadable_filess z#WebDriver.delete_downloadable_files)rdTNNr{)rLN)r6)NNNN)rr^r_r`rar$rr"rr rr rurzr|typingr Type BaseExceptiontypes TracebackTyperrrpropertyrrrhrrrrrrsrrrrr:rr!rrrrrrrrrrrrrrrrr#rrrrrrrrrrrrrrrrsetterrr rr r)boolr0r1rir)r3r>rGrKrNr<rEr=rrprTr[r^rrlr_rrrrrrrrrrrrrrrr&r&r&r'rcs  .                   "             rc)F)Frarrr,rrrrr&rabcrrrrr importlibrrr r r selenium.common.exceptionsr r rrrselenium.webdriver.common.byr!selenium.webdriver.common.optionsr,selenium.webdriver.common.print_page_optionsr"selenium.webdriver.common.timeoutsr/selenium.webdriver.common.virtual_authenticatorrrr+selenium.webdriver.support.relative_locatorrrlrrr errorhandlerrrprrrrremote_connectionr rr! shadowrootr"rr# webelementr$r%r(r3rJr\r]rcr&r&r&r'sf                                    "