U g(@s~ddlZddlZddlZddlmZddlmZddlmZddl Z ddl m Z ddl m Z GdddeZ Gd d d ZdS) N)Enum)Optional)parse)Proxy) ProxyTypec@seZdZdZdZdZdS)AuthTypeZBasicZBearerz X-API-KeyN)__name__ __module__ __qualname__BASICBEARER X_API_KEYrrK/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/remote/client_config.pyrsrc@seZdZdedejidddddddejdddf ee e e ee e e e e e e ee ee ee ee ee ee e ddddZ eed d d Zejedd d d Zee d ddZeje dd ddZeed ddZejeddddZee d ddZeje ddddZee d ddZeje ddddZee d ddZeje ddd dZdd d!d"Zeed d#d$Zejedd%d&d$Zeed d'd(Zejedd d)d(Zeed d*d+Zejedd d,d+Zeed d-d.Zejedd d/d.Zeed d0d1Zejedd d2d1Zeed d3d4Zejedd d5d4Zee d d6d7Zeje dd d8d7Ze ed d9d:Ze e d d;d<ZdS)= ClientConfigTZ proxyType)rawFN)remote_server_addr keep_aliveproxyignore_certificatesinit_args_for_pool_managertimeoutca_certsusernamepassword auth_typetoken user_agent extra_headersreturncCs||_||_||_||_|pi|_||_||_| |_| |_| |_ | |_ | |_ |dkrt ddk rztt dttqtn||_|dkrdt jkrt dqtn||_dS)NZGLOBAL_DEFAULT_TIMEOUTREQUESTS_CA_BUNDLE)rrrrrrrrrrrrosgetenvfloatstrsocketgetdefaulttimeoutenvironcertifiwherer)selfrrrrrrrrrrrrrrrr__init__%s0   zClientConfig.__init__)rcCs|jS)z+:Returns: The address of the remote server.Z_remote_server_addrr*rrrrRszClientConfig.remote_server_addr)valuercCs ||_dS)z*Provides the address of the remote server.Nr,r*r.rrrrWscCs|jS)z:Returns: The keep alive value. _keep_aliver-rrrr\szClientConfig.keep_alivecCs ||_dS)zrToggles the keep alive value. :Args: - value: whether to keep the http connection alive Nr0r/rrrrascCs|jS)z@:Returns: The proxy used for communicating to the driver/server._proxyr-rrrrjszClientConfig.proxy)rrcCs ||_dS)zProvides the information for communicating with the driver or server. For example: Proxy(raw={"proxyType": ProxyType.SYSTEM}) :Args: - value: the proxy information to use to communicate with the driver or server Nr2)r*rrrrros cCs|jS)z-:Returns: The ignore certificate check value.Z_ignore_certificatesr-rrrrzsz ClientConfig.ignore_certificates)rrcCs ||_dS)zrToggles the ignore certificate check. :Args: - value: value of ignore certificate check Nr4)r*rrrrrscCs|jS)zc:Returns: The dictionary of arguments will be appended while initializing the pool manager.Z_init_args_for_pool_managerr-rrrrsz'ClientConfig.init_args_for_pool_manager)rrcCs ||_dS)a%Provides dictionary of arguments will be appended while initializing the pool manager. For example: {"init_args_for_pool_manager": {"retries": 3, "block": True}} :Args: - value: the dictionary of arguments will be appended while initializing the pool manager Nr5)r*rrrrrscCs|jS)zW:Returns: The timeout (in seconds) used for communicating to the driver/server._timeoutr-rrrrszClientConfig.timeout)rrcCs ||_dS)zProvides the timeout (in seconds) for communicating with the driver or server. :Args: - value: the timeout (in seconds) to use to communicate with the driver or server Nr6)r*rrrrrscCst|_dS)z2Resets the timeout to the default value of socket.N)r%r&r7r-rrr reset_timeoutszClientConfig.reset_timeoutcCs|jS)z0:Returns: The path to bundle of CA certificates.Z _ca_certsr-rrrrszClientConfig.ca_certs)rrcCs ||_dS)zProvides the path to bundle of CA certificates for establishing secure connections. :Args: - value: the path to bundle of CA certificates for establishing secure connections Nr9)r*rrrrrscCs|jS)zPReturns the username used for basic authentication to the remote server.Z _usernamer-rrrrszClientConfig.usernamecCs ||_dS)zMSets the username used for basic authentication to the remote server.Nr:r/rrrrscCs|jS)zPReturns the password used for basic authentication to the remote server.Z _passwordr-rrrrszClientConfig.passwordcCs ||_dS)zMSets the password used for basic authentication to the remote server.Nr;r/rrrrscCs|jS)z8Returns the type of authentication to the remote server.Z _auth_typer-rrrrszClientConfig.auth_typecCs ||_dS)zSets the type of authentication to the remote server if it is not using basic with username and password. :Args: value - AuthType enum value. For others, please use `extra_headers` instead Nr<r/rrrrscCs|jS)z?Returns the token used for authentication to the remote server._tokenr-rrrrszClientConfig.tokencCs ||_dS)z^Sets the token used for authentication to the remote server if auth_type is not basic.Nr=r/rrrrscCs|jS)z6Returns user agent to be added to the request headers.Z _user_agentr-rrrrszClientConfig.user_agentcCs ||_dS)z3Sets user agent to be added to the request headers.Nr?r/rrrrscCs|jS)z1Returns extra headers to be added to the request._extra_headersr-rrrrszClientConfig.extra_headerscCs ||_dS)z.Sets extra headers to be added to the request.Nr@r/rrrrscCs|jj}t|j}|tjkr"dS|tjkrtj dtj d}|rt t j |dD]F}|dkrjdSt|}|jr|j|jkrdS|j|jkrXdSqXtj |jdrdndtj |jdrd nd S|tjkr|jdr|jjS|jjSdS) z0Returns the proxy URL to use for the connection.Nno_proxyZNO_PROXY,*zhttps://Z https_proxy http_proxyZ HTTPS_PROXYZ HTTP_PROXY)r proxy_typerurlparserrZDIRECTSYSTEMr!r'getmapr$stripsplitnetlocpath startswithZMANUALZsslProxyrE)r*rFZ remote_addZ _no_proxyentryZn_urlrrr get_proxy_urls,      zClientConfig.get_proxy_urlcCs|jtjkrV|jrV|jrV|jd|j}t|dd}dtjj d|iS|jtj kr|j rdtj j d|j iS|jtj kr|j rtj j |j iSdS)z8Returns the authorization to add to the request headers.:zutf-8 Authorization N) rrr rrbase64 b64encodeencodedecoder.r rr )r* credentialsZencoded_credentialsrrrget_auth_headerszClientConfig.get_auth_header) rr r rrrHrr r$rbooldictintr+propertyrsetterrrrrrr8rrrrrrrrQrZrrrrr$s -    r)rUr!r%enumrtypingrurllibrr(Zselenium.webdriver.common.proxyrrrrrrrrs