a 5j%@sddlZddlZddlZddlZddlmZddlmZddlmZddl m Z ddlm Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZe e!Z"GdddeZ#dS)N)ABC)abstractmethod)IOBase)system)PIPE)sleep)IO)Any)List)Mapping)Optional)Union)cast)request)URLError)WebDriverException)SubprocessStdAlias)utilsc@s eZdZdZd#eeeeee e feddddZ e eddd Z e eedd d Ze edd d Zejedddd ZddddZddddZedddZddddZddddZddddZddddZedddd Zeedd!d"ZdS)$ServiceaThe abstract base class for all service objects. Services typically launch a child program in a new process as an interim process to communicate with a browser. :param executable: install path of the executable. :param port: Port for the service to run on, defaults to 0 where the operating system will decide. :param log_output: (Optional) int representation of STDOUT/DEVNULL, any IO instance or String path to file. :param env: (Optional) Mapping of environment variables for the new process, defaults to `os.environ`. :param driver_path_env_key: (Optional) Environment variable to use to get the path to the driver executable. Nr)executable_pathport log_outputenvdriver_path_env_keyreturncKst|tr"ttt|ddd|_nX|tjkrFttt t tfd|_n4|dusX|tj krtttt t tftj |_n||_|pt |_|di|_|jdd|_|ptj|_||_|p||_dS)Nza+zutf-8)encodingpopen_kwcreation_flagsr) isinstancestrrropenr subprocessSTDOUTr r intDEVNULLrZ free_portrpoprrosenvironrDRIVER_PATH_ENV_KEYenv_path_path)selfrrrrrkwargsr-GE:\soccer\ml_env\Lib\site-packages\selenium/webdriver/common/service.py__init__8s   zService.__init__)rcCsdtd|jS)zGets the url of the Service.zhttp:// localhost)rZjoin_host_portrr+r-r-r. service_urlRszService.service_urlcCs tddS)z7A List of program arguments (excluding the executable).z2This method needs to be implemented in a sub classN)NotImplementedErrorr1r-r-r.command_line_argsWszService.command_line_argscCs |jpdS)N)r*r1r-r-r.path\sz Service.path)valuercCst||_dSN)rr*)r+r7r-r-r.r6`scCsp|jdurtd||jd}||r4qlttdd|d|d7}|dkr"td |jq"dS) zStarts the Service. :Exceptions: - WebDriverException : Raised either when it can't start the service or when it can't connect to the service NService path cannot be None.rg{Gz?g?g?FzCan not connect to the Service )r*r_start_processassert_process_still_runningis_connectablermin)r+countr-r-r.startds  z Service.startcCs(|j}|r$td|jd|dS)z1Check if the underlying process is still running.zService z' unexpectedly exited. Status code was: N)processpollrr*)r+Z return_coder-r-r.r=zs z$Service.assert_process_still_runningcCs t|jS)zfEstablishes a socket connection to determine if the service running on the port is accessible.)rr>rr1r-r-r.r>szService.is_connectablecCsRzt|jdWnty*YdS0tdD]}|sDqNtdq4dS)zcDispatch an HTTP request to the shutdown endpoint for the service in an attempt to stop it.z /shutdownNr:)rurlopenr2rranger>r)r+_r-r-r.send_remote_shutdown_commands  z$Service.send_remote_shutdown_commandcCs|jttjhvr@t|jtr(|jnt|jtr@t|j|j dur|j durz,z | Wnt yxYn0W| n | 0dS)zStops the service.N)rrr!r$rrcloser#r&rBrCrH TypeError_terminate_processr1r-r-r.stops      z Service.stopc Csz|jj|jj|jj}}}|||fD]$}z |Wq(tyJYq(0q(|jz|jdWn,tj yt j ddd|j Yn0Wn t yt j dddYn0dS)a#Terminate the child process. On POSIX this attempts a graceful SIGTERM followed by a SIGKILL, on a Windows OS kill is an alias to terminate. Terminating does not raise itself if something has gone wrong but (currently) silently ignores errors here. <zTService process refused to terminate gracefully with SIGTERM, escalating to SIGKILL.T)exc_infoz"Error terminating service process.N)rBstdinstdoutstderrrIAttributeError terminatewaitr!TimeoutExpiredloggererrorkillOSError)r+rOrPrQstreamr-r-r.rKs*     zService._terminate_processcCs$z |WntyYn0dSr8)rL Exceptionr1r-r-r.__del__s  zService.__del__)r6rc CsL|g}|||jdtdk}zd}tdkrVt}tjtjB|_ tj |_ tj |f|j |ttttttf|jttttttf|jt|j|d|j|_td|j|jj|j|jWnxtyYnftyF}zL|jtjkr0|jdurt dt dt!j"#|jd|WYd}~n d}~00dS) zmCreates a subprocess by executing the command provided. :param cmd: full command to execute close_fdsWindowsN)rr]rPrQrO creationflags startupinfozNStarted executable: `%s` in a child process with pid: %s using %s to output %sr9'z(' executable may have wrong permissions.)$extendr4rr%rr! STARTUPINFOCREATE_NEW_CONSOLESTARTF_USESHOWWINDOWdwFlagsSW_HIDE wShowWindowPopenrrr r r#rr rrrrBrVdebugr*pidrJrYerrnoEACCESrr&r6basename)r+r6cmdZclose_file_descriptorsZ start_infoerrr-r-r.r<sR     zService._start_processcCs|jrt|jdSdSr8)r(r&getenvr1r-r-r.r)szService.env_path)NrNNN)__name__ __module__ __qualname____doc__rr#rr r r r/propertyr2rr r4r6setterrAr=boolr>rHrLrKr\r<r)r-r-r-r.r,s>    ,r)$rlloggingr&r!abcrriorplatformrrtimertypingrr r r r r rurllibr urllib.errorrZselenium.common.exceptionsrZselenium.typesrZselenium.webdriver.commonr getLoggerrrrVrr-r-r-r.s.