o 5@Hh@s^ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z GdddZ dS)N)SeleniumManagercszeZdZdZdddZddZd d Zfd d Zfd dZddZ ddZ dddZ dddZ ddZ ddZZS)ServeraManage a Selenium Grid (Remote) Server in standalone mode. This class contains functionality for downloading the server and starting/stopping it. For more information on Selenium Grid, see: - https://www.selenium.dev/documentation/grid/getting_started/ Parameters: ----------- host : str Hostname or IP address to bind to (determined automatically if not specified) port : int or str Port to listen on (4444 if not specified) path : str Path/filename of existing server .jar file (Selenium Manager is used if not specified) version : str Version of server to download (latest version if not specified) log_level : str Logging level to control logging output ("INFO" if not specified) Available levels: "SEVERE", "WARNING", "INFO", "CONFIG", "FINE", "FINER", "FINEST" env: collections.abc.Mapping Mapping that defines the environment variables for the server process N\INFOcCsf|r|rtd||_|||_|||_|||_|||_ | ||_ d|_ | |_dS)NzCNot allowed to specify a version when using an existing server path) TypeErrorhost_validate_portport_validate_pathpath_validate_versionversion_validate_log_level log_level _validate_envenvprocess_get_status_url status_url)selfrr r r rrr`/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/selenium/webdriver/remote/server.py__init__7s     zServer.__init__cCs(|jdur|jnd}d|d|jdS)N localhostzhttp://:z/status)rr )rrrrrrEszServer._get_status_urlcCs"|rtj|std||S)Nz"Can't find server .jar located at )osr existsOSError)rr rrrr IszServer._validate_pathcsZzt|}Wntytjd|dwd|kr&dks+tdtd|S)Nz".__init__() got an invalid port: ''rizport must be 0-65535)int ValueErrorr__name__)rr  __class__rrrNs  zServer._validate_portcs.|rtdt|stjd|d|S)Nz^\d+\.\d+\.\d+$z%.__init__() got an invalid version: 'r)rematchstrrr!)rr r"rrr WszServer._validate_versioncCs$d}||vrtdd||S)N)SEVEREWARNINGrCONFIGFINEFINERFINESTzlog_level must be one of: z, )rjoin)rrlevelsrrrr]szServer._validate_log_levelcCs"|durt|tjjstd|S)Nz.env must be a mapping of environment variables) isinstance collectionsabcMappingr)rrrrrrcszServer._validate_env cCsbt}t||kr/z tj|jWdStjjy&tdYnwt||ks dS)NTg?F)timeurllibrequesturlopenrerrorURLErrorsleep)rtimeoutstartrrr_wait_for_serverhszServer._wait_for_servercCs(dg}|dur ||t|dS)zqDownload the server if it doesn't already exist. Latest version is downloaded unless specified. z--gridN driver_path)appendr binary_paths)rr argsrrrdownload_if_neededrs zServer.download_if_neededc Cs<|jdur ||jn|j}td}|durtd|d|ddt|jd|jdd d d g }|j dur:| d |j g|j durB|j nd }z(t t j t j }|||jfWdn1s`wYtd |jtytdtj||jd|_td|jj|std|jtdY|jSw)zStart the server. Selenium Manager will detect the server location and download it if necessary, unless an existing server path was specified. NjavazJCan't find java on system PATH. JRE is required to run the Selenium serverz-jar standalonez--portz --log-levelz--selenium-managertruez--enable-managed-downloadsz--hostrzDSelenium server is already running, or something else is using port zStarting Selenium server...)rz$Selenium server running as process: z)Timed out waiting for Selenium server at zSelenium server is ready)r rBr shutilwhichrr&r rrextendsocketAF_INET SOCK_STREAMconnectConnectionErrorConnectionRefusedErrorprint subprocessPopenrrpidr= TimeoutErrorr)rr java_pathcommandrsockrrrr<|sD    z Server.startcCsF|jdur td|jdur|j|jd|_tddS)zStop the server.NzSelenium server isn't runningz#Selenium server has been terminated)r RuntimeErrorpoll terminatewaitrO)rrrrstops    z Server.stop)NrNNrN)r3)N)r! __module__ __qualname____doc__rrr rr rrr=rBr<r[ __classcell__rrr"rrs      ,r) r0rr$rFrIrPr4r5*selenium.webdriver.common.selenium_managerrrrrrrs