o 5@Hh W @sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl Z ddl mZddlmZdd lmZdd lmZdd lmZeeZiejd ejd ejdejdejdejdej dej!dej"dej#dej$dej%dej&dej'dej(dej)dej*diej+dej,dej-dej.d ej/d!ej0d"ej1d#ej2d$ej3d%ej4d&ej5d'ej6d(ej7d)ej8d*ej9d+ej:d,ej;d-iejd0ej?d1ej@d2ejAd3ejBd4ejCd5ejDd6ejEd7ejFd8ejGd9ejHd:ejId;ejJd<ejKd=ejLd>iejMd?ejNd@ejOdAejPdBejQdCejRdDejSdEejTdFejUdGejVdHejWdIejXdJejYdKejZdLej[dMej\dNej]dOiej^dPej_dQej`dRejadSejbdTejcdUejddVejedWejfdXejgdYejhdZejid[ejjd\ejkd]ejld^ejmd_ejnd`ejodaejpdbejqdcejrddiZsGdedfdfZtdS)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@seZdZUdZdZddlZddlZddlZeZ dej vr#e dne Z dZeeed<eZedkr;dZdZded ed Zed d Zed dZeddZeddZeddZeddZed7ddZ ddZ!ddZ"ddZ#     d8d!ee$d"ee%d#ee%d$ee%d%ee&d&eef d'd(Z'iZ(d)d*Z)d+e$fd,d-Z*d.d/Z+d9d0d1Z,d2d3Z-d:d5d6Z.dS);RemoteConnectionzA connection with the Remote WebDriver server. Communicates with the server using the WebDriver wire protocol: https://github.com/SeleniumHQ/selenium/wiki/JsonWireProtocol NrREQUESTS_CA_BUNDLE_client_configdarwinmacz selenium/z (python )cCs|jSN)rselfr%k/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/selenium/webdriver/remote/remote_connection.py client_configszRemoteConnection.client_configcCtjdtdd|jjS)zp:Returns: Timeout value in seconds for all http requests made to the Remote Connection zWget_timeout() in RemoteConnection is deprecated, get timeout from client_config instead stacklevelwarningswarnDeprecationWarningrtimeoutclsr%r%r& get_timeouts zRemoteConnection.get_timeoutcCtjdtdd||j_dS)zyOverride the default timeout. :Args: - timeout - timeout value for http requests in seconds zUset_timeout() in RemoteConnection is deprecated, set timeout in client_config insteadr)r*Nr,)r2r0r%r%r& set_timeouts  zRemoteConnection.set_timeoutcCstjdtdd|jdS)zAReset the http request timeout to socket._GLOBAL_DEFAULT_TIMEOUT.z_reset_timeout() in RemoteConnection is deprecated, use reset_timeout() in client_config insteadr)r*N)r-r.r/r reset_timeoutr1r%r%r&r6s zRemoteConnection.reset_timeoutcCr()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. zhget_certificate_bundle_path() in RemoteConnection is deprecated, get ca_certs from client_config insteadr)r*r-r.r/rca_certsr1r%r%r&get_certificate_bundle_paths z,RemoteConnection.get_certificate_bundle_pathcCr4)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. zfset_certificate_bundle_path() in RemoteConnection is deprecated, set ca_certs in client_config insteadr)r*Nr7)r2pathr%r%r&set_certificate_bundle_paths   z,RemoteConnection.set_certificate_bundle_pathFcCs|dd|jd}|jr*tjdddt|jd|j}|dd |i|r3|d d i|j r<||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)r2 parsed_url keep_aliveheaders base64stringr%r%r&get_remote_connection_headerss  z.RemoteConnection.get_remote_connection_headerscCs"t|j}|jr |jrdSdSdS)NT)r _proxy_urlrBrC)r$rHr%r%r&_identify_http_proxy_auths  z*RemoteConnection._identify_http_proxy_authcCs>t|j}|jd|jd|j}|jd|j}||fS)Nz://r>)rrMschemehostnameportrBrC)r$rHproxy_without_authauthr%r%r&_separate_http_proxy_auths z*RemoteConnection._separate_http_proxy_authcCsd|jji}||jjdi|jjr!d|d<ttjj n|jj r/d|d<|jj |d<|j rh|j drIdd lm}||j fi|S|r^|\|_ |_tj|jd |d <tj|j fi|Stjd i|S) Nr0init_args_for_pool_manager CERT_NONE cert_reqs CERT_REQUIREDr8sockr)SOCKSProxyManager)proxy_basic_auth proxy_headersr%)rr0rErUgetignore_certificatesurllib3disable_warnings exceptionsInsecureRequestWarningr8rMlower startswithurllib3.contrib.socksrZrNrT_basic_proxy_auth make_headers ProxyManager PoolManager)r$pool_manager_init_argsrZr%r%r&_get_connection_managers&   z(RemoteConnection._get_connection_managerTremote_server_addrrI ignore_proxyr^rUr'cCs|p t||||d|_|jjt_|jjt_|jt_|jjptjt_|jjp'tjt_|r3t j dt dd|s=t j dt dd|rGt j dt dd|rQt j dt dd|r_t j dt ddd|_ n|j |_ |jjrn||_t|_dS) N)rlrIr^rUz\setting remote_server_addr in RemoteConnection() is deprecated, set in client_config insteadr)r*zTsetting keep_alive in RemoteConnection() is deprecated, set in client_config insteadz]setting ignore_certificates in RemoteConnection() is deprecated, set in client_config insteadzdsetting init_args_for_pool_manager in RemoteConnection() is deprecated, set in client_config insteadzVsetting ignore_proxy in RemoteConnection() is deprecated, set in client_config instead)r rr0r_timeoutr8 _ca_certsrGrAr-r.r/rM get_proxy_urlrIrk_connremote_commands _commands)r$rlrIrmr^rUr'r%r%r&__init__2s\      zRemoteConnection.__init__cCs||f|j|<dS)zRegister a new command.N)rs)r$namemethodurlr%r%r& add_commanduszRemoteConnection.add_commandrucCs |j|S)z Retrieve a command if it exists.)rsr])r$rur%r%r& get_commandys zRemoteConnection.get_commandc Cs|j|p |j|}|dusJd||d}t||}dd|dD}t|tr<|r<|D]}||=q6t |}|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)$rNrd).0wordr%r%r& s"z+RemoteConnection.execute../z%s %s %sr)body)rsr]extra_commandsstringTemplate substitutesplit isinstancedictr dump_jsonrrl_trim_large_entriesLOGGERdebugstr_request) r$commandparams command_info path_stringr:substitute_paramsr}datarwtrimmedr%r%r&execute}s   zRemoteConnection.executec Cst|}|||jj}|j}|r|||r!|dvr!d}|jjr6|jj|||||jj d}|j }n%| } | } | j|||||jj d}Wdn1sSwY|j }|j d} td|j | |jzd|krvdkrnn|d|jd dWtd |Sd |krd krnn*|d kr|ddWtd |S|| st|n| dWtd |Sg} |jddr|jddd} tdd| Ds.z t| } Wn/tyd|krdkrnntj} ntj} | | dYWtd |Swd| vr"d| d<| Wtd |Sd| d} | Wtd |Std |w)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)rrJr0zUTF-8z1Remote response: status=%s | data=%s | headers=%si,i0rlocationzFinished RequestiiizAuthorization Required)statusvaluer=;cSsg|]}|dqS)z image/pngr{)r|xr%r%r& sz-RemoteConnection._request..rr)rrrLrrIget_auth_headerrErqrequestr0rrkrrFrrrJrr]closerstripranyr load_json ValueErrorr SUCCESS UNKNOWN_ERROR)r$rvrwrrHrJ auth_headerresponse statuscodeconnhttpr content_typerr%r%r&rsn                    zRemoteConnection._requestcCst|dr |jdSdS)zr)uloggingrrr-base64rtypingrurllibr urllib.parserr_seleniumrrr'r rr errorhandlerr getLoggerrr NEW_SESSIONQUITW3C_GET_CURRENT_WINDOW_HANDLEW3C_GET_WINDOW_HANDLESr GO_FORWARDGO_BACKREFRESHW3C_EXECUTE_SCRIPTW3C_EXECUTE_SCRIPT_ASYNCGET_CURRENT_URL GET_TITLEGET_PAGE_SOURCE SCREENSHOTELEMENT_SCREENSHOT FIND_ELEMENT FIND_ELEMENTSW3C_GET_ACTIVE_ELEMENTFIND_CHILD_ELEMENTFIND_CHILD_ELEMENTS CLICK_ELEMENT CLEAR_ELEMENTGET_ELEMENT_TEXTSEND_KEYS_TO_ELEMENTGET_ELEMENT_TAG_NAMEIS_ELEMENT_SELECTEDIS_ELEMENT_ENABLEDGET_ELEMENT_RECTGET_ELEMENT_ATTRIBUTEGET_ELEMENT_PROPERTYGET_ELEMENT_ARIA_ROLEGET_ELEMENT_ARIA_LABELGET_SHADOW_ROOTFIND_ELEMENT_FROM_SHADOW_ROOTFIND_ELEMENTS_FROM_SHADOW_ROOTGET_ALL_COOKIES ADD_COOKIE GET_COOKIEDELETE_ALL_COOKIES DELETE_COOKIESWITCH_TO_FRAMESWITCH_TO_PARENT_FRAMESWITCH_TO_WINDOW NEW_WINDOWCLOSE!GET_ELEMENT_VALUE_OF_CSS_PROPERTYEXECUTE_ASYNC_SCRIPT SET_TIMEOUTS GET_TIMEOUTSW3C_DISMISS_ALERTW3C_ACCEPT_ALERTW3C_SET_ALERT_VALUEW3C_GET_ALERT_TEXT W3C_ACTIONSW3C_CLEAR_ACTIONSSET_WINDOW_RECTGET_WINDOW_RECTW3C_MAXIMIZE_WINDOWSET_SCREEN_ORIENTATIONGET_SCREEN_ORIENTATIONGET_NETWORK_CONNECTIONSET_NETWORK_CONNECTIONGET_LOGGET_AVAILABLE_LOG_TYPESCURRENT_CONTEXT_HANDLECONTEXT_HANDLESSWITCH_TO_CONTEXTFULLSCREEN_WINDOWMINIMIZE_WINDOW PRINT_PAGEADD_VIRTUAL_AUTHENTICATORREMOVE_VIRTUAL_AUTHENTICATORADD_CREDENTIALGET_CREDENTIALSREMOVE_CREDENTIALREMOVE_ALL_CREDENTIALSSET_USER_VERIFIED UPLOAD_FILEGET_DOWNLOADABLE_FILES DOWNLOAD_FILEDELETE_DOWNLOADABLE_FILESGET_FEDCM_TITLEGET_FEDCM_DIALOG_TYPEGET_FEDCM_ACCOUNT_LISTCLICK_FEDCM_DIALOG_BUTTONCANCEL_FEDCM_DIALOGSELECT_FEDCM_ACCOUNTSET_FEDCM_DELAYRESET_FEDCM_COOLDOWNrrrr%r%r%r&s                !"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHLMNRVWXYZ\]^_`g