U gW@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl Z ddl mZddlmZdd lmZdd lmZdd lmZeeZejd ejd ejdejdejdejdej dej!dej"dej#dej$dej%dej&dej'dej(dej)dej*dej+dej,dej-dej.d ej/d!ej0d"ej1d#ej2d$ej3d%ej4d&ej5d'ej6d(ej7d)ej8d*ej9d+ej:d,ej;d-ejd0ej?d1ej@d2ejAd3ejBd4ejCd5ejDd6ejEd7ejFd8ejGd9ejHd:ejId;ejJdejMd?ejNd@ejOdAejPdBejQdCejRdDejSdEejTdFejUdGejVdHejWdIejXdJejYdKejZdLej[dMej\dNej]dOej^dPej_dQej`dRejadSejbdTejcdUejddVejedWejfdXejgdYejhdZejid[ejjd\ejkd]ejld^ejmd_ejnd`ejodaejpdbejqdcejrddiYZsGdedfdfZtdS)gN) b64encode)Optional)parse)urlparse) __version__)utils) ClientConfig)Command) ErrorCode)POSTz/session)DELETEz/session/$sessionId)GET/session/$sessionId/window)rz"/session/$sessionId/window/handles)r /session/$sessionId/url)r z/session/$sessionId/forward)r z/session/$sessionId/back)r z/session/$sessionId/refresh)r z /session/$sessionId/execute/sync)r z!/session/$sessionId/execute/async)rr)rz/session/$sessionId/title)rz/session/$sessionId/source)rz/session/$sessionId/screenshot)rz*/session/$sessionId/element/$id/screenshot)r z/session/$sessionId/element)r z/session/$sessionId/elements)rz"/session/$sessionId/element/active)r z'/session/$sessionId/element/$id/element)r z(/session/$sessionId/element/$id/elements)r z%/session/$sessionId/element/$id/click)r z%/session/$sessionId/element/$id/clear)rz$/session/$sessionId/element/$id/text)r z%/session/$sessionId/element/$id/value)rz$/session/$sessionId/element/$id/name)rz(/session/$sessionId/element/$id/selected)rz'/session/$sessionId/element/$id/enabled)rz$/session/$sessionId/element/$id/rect)rz//session/$sessionId/element/$id/attribute/$name)rz./session/$sessionId/element/$id/property/$name)rz,/session/$sessionId/element/$id/computedrole)rz-/session/$sessionId/element/$id/computedlabel)rz&/session/$sessionId/element/$id/shadow)r z,/session/$sessionId/shadow/$shadowId/element)r z-/session/$sessionId/shadow/$shadowId/elements)r/session/$sessionId/cookie)r r)r /session/$sessionId/cookie/$name)r r)r r)r z/session/$sessionId/frame)r z /session/$sessionId/frame/parent)r r)r z/session/$sessionId/window/new)r r)rz1/session/$sessionId/element/$id/css/$propertyName)r z!/session/$sessionId/execute_async)r /session/$sessionId/timeouts)rr)r z!/session/$sessionId/alert/dismiss)r z /session/$sessionId/alert/accept)r /session/$sessionId/alert/text)rr)r /session/$sessionId/actions)r r)r /session/$sessionId/window/rect)rr)r z#/session/$sessionId/window/maximize)r /session/$sessionId/orientation)rr)r&/session/$sessionId/network_connection)r r)r z/session/$sessionId/se/log)rz /session/$sessionId/se/log/types)r/session/$sessionId/context)rz/session/$sessionId/contexts)r r)r z%/session/$sessionId/window/fullscreen)r z#/session/$sessionId/window/minimize)r z/session/$sessionId/print)r z*/session/$sessionId/webauthn/authenticator)r z;/session/$sessionId/webauthn/authenticator/$authenticatorId)r zF/session/$sessionId/webauthn/authenticator/$authenticatorId/credential)rG/session/$sessionId/webauthn/authenticator/$authenticatorId/credentials)r zU/session/$sessionId/webauthn/authenticator/$authenticatorId/credentials/$credentialId)r r)r z>/session/$sessionId/webauthn/authenticator/$authenticatorId/uv)r z/session/$sessionId/se/file)r/session/$sessionId/se/files)r r)r r)rz"/session/$sessionId/fedcm/gettitle)rz'/session/$sessionId/fedcm/getdialogtype)rz%/session/$sessionId/fedcm/accountlist)r z+/session/$sessionId/fedcm/clickdialogbutton)r z&/session/$sessionId/fedcm/canceldialog)r z'/session/$sessionId/fedcm/selectaccount)r z)/session/$sessionId/fedcm/setdelayenabled)r z'/session/$sessionId/fedcm/resetcooldownc@seZdZUdZdZddlZddlZddlZeddk rPe ede e ne Z dej krledneZdZeed<eZedkrdZdZd ed ed Zed d ZeddZeddZeddZeddZed1ddZddZ ddZ!ddZ"d2e#e e#e$e#e$e#e$e#e%e#ed d!d"Z&iZ'd#d$Z(e d%d&d'Z)d(d)Z*d3d*d+Z+d,d-Z,d4d/d0Z-dS)5RemoteConnectionzA connection with the Remote WebDriver server. Communicates with the server using the WebDriver wire protocol: https://github.com/SeleniumHQ/selenium/wiki/JsonWireProtocol NrZGLOBAL_DEFAULT_TIMEOUTREQUESTS_CA_BUNDLE_client_configdarwinmacz selenium/z (python )cCstjdtdd|jjS)zp:Returns: Timeout value in seconds for all http requests made to the Remote Connection z_get_timeout() in RemoteConnection is deprecated, get timeout from ClientConfig instance instead stacklevelwarningswarnDeprecationWarningrtimeoutclsr,O/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/remote/remote_connection.py get_timeouts zRemoteConnection.get_timeoutcCstjdtdd||j_dS)zyOverride the default timeout. :Args: - timeout - timeout value for http requests in seconds zlset_timeout() in RemoteConnection is deprecated, set timeout to ClientConfig instance in constructor insteadr"r#Nr%)r+r)r,r,r- set_timeouts zRemoteConnection.set_timeoutcCstjdtdd|jdS)zAReset the http request timeout to socket._GLOBAL_DEFAULT_TIMEOUT.zgreset_timeout() in RemoteConnection is deprecated, use reset_timeout() in ClientConfig instance insteadr"r#N)r&r'r(r reset_timeoutr*r,r,r-r0s zRemoteConnection.reset_timeoutcCstjdtdd|jjS)z:Returns: Paths of the .pem encoded certificate to verify connection to command executor. Defaults to certifi.where() or REQUESTS_CA_BUNDLE env variable if set. zpget_certificate_bundle_path() in RemoteConnection is deprecated, get ca_certs from ClientConfig instance insteadr"r#r&r'r(rca_certsr*r,r,r-get_certificate_bundle_paths z,RemoteConnection.get_certificate_bundle_pathcCstjdtdd||j_dS)zSet the path to the certificate bundle to verify connection to command executor. Can also be set to None to disable certificate validation. :Args: - path - path of a .pem encoded certificate chain. z}set_certificate_bundle_path() in RemoteConnection is deprecated, set ca_certs to ClientConfig instance in constructor insteadr"r#Nr1)r+pathr,r,r-set_certificate_bundle_paths z,RemoteConnection.set_certificate_bundle_pathFcCs|dd|jd}|jrTtjdddt|jd|j}|dd |i|rf|d d i|j rx||j |S) zGet headers for remote request. :Args: - parsed_url - The parsed url - keep_alive (Boolean) - Is this a keep-alive connection (default: False) zapplication/jsonzapplication/json;charset=UTF-8)Accept Content-Typez User-AgentzREmbedding username and password in URL could be insecure, use ClientConfig insteadr"r#: AuthorizationzBasic Connectionz keep-alive) user_agentusernamer&r'rpasswordencodeupdatedecode extra_headers)r+ parsed_url keep_aliveheadersZ base64stringr,r,r-get_remote_connection_headerss  z.RemoteConnection.get_remote_connection_headerscCst|j}|jr|jrdSdS)NT)r _proxy_urlr<r=)selfrBr,r,r-_identify_http_proxy_auths  z*RemoteConnection._identify_http_proxy_authcCs>t|j}|jd|jd|j}|jd|j}||fS)Nz://r8)rrFschemehostnameportr<r=)rGrBZproxy_without_authauthr,r,r-_separate_http_proxy_auths z*RemoteConnection._separate_http_proxy_authcCsd|jji}||jjdi|jjrBd|d<ttjj n|jj r^d|d<|jj |d<|j r|j drdd lm}||j f|S|r|\|_ |_tj|jd |d <tj|j f|Stjf|S) Nr)init_args_for_pool_manager CERT_NONE cert_reqs CERT_REQUIREDr2sockr)SOCKSProxyManager)proxy_basic_auth proxy_headers)rr)r?rNgetignore_certificatesurllib3disable_warnings exceptionsInsecureRequestWarningr2rFlower startswithZurllib3.contrib.socksrSrHrMZ_basic_proxy_auth make_headers ProxyManager PoolManager)rGZpool_manager_init_argsrSr,r,r-_get_connection_managers&   z(RemoteConnection._get_connection_managerT)remote_server_addrrC ignore_proxyrWrN client_configcCs|pt||||d|_|jjt_|jjt_|jt_|jjp>tjt_|jjpNtjt_|rft j dt dd|szt j dt dd|rt j dt dd|rt j dt dd|rt j dt ddd|_ n |j |_ |jjr||_t|_dS) N)rbrCrWrNzdsetting remote_server_addr in RemoteConnection() is deprecated, set in ClientConfig instance insteadr"r#z\setting keep_alive in RemoteConnection() is deprecated, set in ClientConfig instance insteadzesetting ignore_certificates in RemoteConnection() is deprecated, set in ClientConfig instance insteadzlsetting init_args_for_pool_manager in RemoteConnection() is deprecated, set in ClientConfig instance insteadz^setting ignore_proxy in RemoteConnection() is deprecated, set in ClientConfig instance instead)r rr)r_timeoutr2 _ca_certsrAr;r&r'r(rFZ get_proxy_urlrCra_connremote_commands _commands)rGrbrCrcrWrNrdr,r,r-__init__2s\     zRemoteConnection.__init__cCs||f|j|<dS)zRegister a new command.N)ri)rGnamemethodurlr,r,r- add_commanduszRemoteConnection.add_command)rkcCs |j|S)z Retrieve a command if it exists.)rirV)rGrkr,r,r- get_commandyszRemoteConnection.get_commandc Cs|j|p|j|}|dk s.td||d}t||}dd|dD}t|t rx|rx|D] }||=qlt |}|j j |} ||} td|d| t| |j|d| |d S) amSend a command to the remote server. Any path substitutions required for the URL mapped to the command should be included in the command parameters. :Args: - command - A string specifying the command to execute. - params - A dictionary of named parameters to send with the command as its JSON payload. NzUnrecognised command rcSs"h|]}|dr|ddqS)$rNr]).0wordr,r,r- s z+RemoteConnection.execute../z%s %s %sr)body)rirVextra_commandsAssertionErrorstringTemplate substitutesplit isinstancedictrZ dump_jsonrrb_trim_large_entriesLOGGERdebugstr_request) rGcommandparams command_infoZ path_stringr4Zsubstitute_paramsrsdatarmZtrimmedr,r,r-execute}s   zRemoteConnection.executec Cs4t|}|||jj}|j}|r2|||rB|dkrBd}|jjrl|jj|||||jj d}|j }n6| } | } | j|||||jj d}W5QRX|j }|j d} td|j | |jzVd|krdkrnn|d |jd dW$Sd |krd krNnn8|d kr.|ddWS|| s>t|n| dWSg} |jddrv|jddd} tdd| Ds zt| } WnRtk rd|krdkrnntj} ntj} | | dYW0SXd| krd| d<| WSd| d} | WStd|XdS)aqSend an HTTP request to the remote server. :Args: - method - A string for the HTTP method to send the request with. - url - A string for the URL to send the request to. - body - A string for request body. Ignored unless method is POST or PUT. :Returns: A dictionary with the server's parsed JSON response. )r PUTN)rvrDr)zUTF-8z1Remote response: status=%s | data=%s | headers=%szFinished Requesti,i0rlocationiiizAuthorization Required)statusvaluer7;cSsg|]}|dqS)z image/pngrq)rrxr,r,r- sz-RemoteConnection._request..rr)rrrErrCZget_auth_headerr?rgrequestr)rrarr@rrrDcloserrVrstripr|anyrZ load_json ValueErrorr SUCCESS UNKNOWN_ERROR)rGrlrmrvrBrD auth_headerresponseZ statuscodeconnhttpr content_typerr,r,r-rsR    "      zRemoteConnection._requestcCst|dr|jdS)zr)uloggingrryr&base64rtypingrurllibr urllib.parserrXZseleniumrrrdr rr Z errorhandlerr getLoggerrrZ NEW_SESSIONZQUITZW3C_GET_CURRENT_WINDOW_HANDLEZW3C_GET_WINDOW_HANDLESrZ GO_FORWARDZGO_BACKZREFRESHZW3C_EXECUTE_SCRIPTZW3C_EXECUTE_SCRIPT_ASYNCZGET_CURRENT_URLZ GET_TITLEZGET_PAGE_SOURCEZ SCREENSHOTZELEMENT_SCREENSHOTZ FIND_ELEMENTZ FIND_ELEMENTSZW3C_GET_ACTIVE_ELEMENTZFIND_CHILD_ELEMENTZFIND_CHILD_ELEMENTSZ CLICK_ELEMENTZ CLEAR_ELEMENTZGET_ELEMENT_TEXTZSEND_KEYS_TO_ELEMENTZGET_ELEMENT_TAG_NAMEZIS_ELEMENT_SELECTEDZIS_ELEMENT_ENABLEDZGET_ELEMENT_RECTZGET_ELEMENT_ATTRIBUTEZGET_ELEMENT_PROPERTYZGET_ELEMENT_ARIA_ROLEZGET_ELEMENT_ARIA_LABELZGET_SHADOW_ROOTZFIND_ELEMENT_FROM_SHADOW_ROOTZFIND_ELEMENTS_FROM_SHADOW_ROOTZGET_ALL_COOKIESZ ADD_COOKIEZ GET_COOKIEZDELETE_ALL_COOKIESZ DELETE_COOKIEZSWITCH_TO_FRAMEZSWITCH_TO_PARENT_FRAMEZSWITCH_TO_WINDOWZ NEW_WINDOWZCLOSEZ!GET_ELEMENT_VALUE_OF_CSS_PROPERTYZEXECUTE_ASYNC_SCRIPTZ SET_TIMEOUTSZ GET_TIMEOUTSZW3C_DISMISS_ALERTZW3C_ACCEPT_ALERTZW3C_SET_ALERT_VALUEZW3C_GET_ALERT_TEXTZ W3C_ACTIONSZW3C_CLEAR_ACTIONSZSET_WINDOW_RECTZGET_WINDOW_RECTZW3C_MAXIMIZE_WINDOWZSET_SCREEN_ORIENTATIONZGET_SCREEN_ORIENTATIONZGET_NETWORK_CONNECTIONZSET_NETWORK_CONNECTIONZGET_LOGZGET_AVAILABLE_LOG_TYPESZCURRENT_CONTEXT_HANDLEZCONTEXT_HANDLESZSWITCH_TO_CONTEXTZFULLSCREEN_WINDOWZMINIMIZE_WINDOWZ PRINT_PAGEZADD_VIRTUAL_AUTHENTICATORZREMOVE_VIRTUAL_AUTHENTICATORZADD_CREDENTIALZGET_CREDENTIALSZREMOVE_CREDENTIALZREMOVE_ALL_CREDENTIALSZSET_USER_VERIFIEDZ UPLOAD_FILEZGET_DOWNLOADABLE_FILESZ DOWNLOAD_FILEZDELETE_DOWNLOADABLE_FILESZGET_FEDCM_TITLEZGET_FEDCM_DIALOG_TYPEZGET_FEDCM_ACCOUNT_LISTZCLICK_FEDCM_DIALOG_BUTTONZCANCEL_FEDCM_DIALOGZSELECT_FEDCM_ACCOUNTZSET_FEDCM_DELAYZRESET_FEDCM_COOLDOWNrhrr,r,r,r-s          g