o Vh!@sddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddlm Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZeeZGd ddeZdS)N)ABC)abstractmethod)IOBase)system)PIPE)sleep)request)URLError)WebDriverException)SubprocessStdAlias)utilsc @seZdZdZ    d$dedededejej ej ej fddf d d Z e defd d Z edejefd dZe defddZejdeddfddZd%ddZd%ddZdefddZd%ddZd%ddZd%ddZd%dd Zd!eddfd"d#ZdS)&Servicea(The 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`. Nrexecutable_pathport log_outputenvreturncKst|trt|ddd|_n|tjkrd|_n|dus |tjkr%tj|_n||_||_|p0t |_ | di|_ |j dd|_ |pEtj|_dS)Nza+zutf-8)encodingpopen_kwcreation_flagsr) isinstancestropenr subprocessSTDOUTDEVNULL_pathr free_portrpoprrosenvironr)selfrrrrkwargsr#^/home/air/yokohama/back/venv/lib/python3.10/site-packages/selenium/webdriver/common/service.py__init__1s   zService.__init__cCsdtd|jS)zGets the url of the Service.zhttp:// localhost)r join_host_portrr!r#r#r$ service_urlIszService.service_urlcCstd)z7A List of program arguments (excluding the executable).z2This method needs to be implemented in a sub class)NotImplementedErrorr(r#r#r$command_line_argsNszService.command_line_argscCs|jSN)rr(r#r#r$pathSsz Service.pathvaluecCst||_dSr,)rr)r!r.r#r#r$r-WscCs^||jd} ||rd Sttdd|d|d7}|dkr.td|jq ) zStarts the Service. :Exceptions: - WebDriverException : Raised either when it can't start the service or when it can't connect to the service rTg{Gz?g?g?FzCan not connect to the Service N)_start_processrassert_process_still_runningis_connectablerminr )r!countr#r#r$start[s z Service.startcCs(|j}|rtd|jd|dS)z1Check if the underlying process is still running.zService z' unexpectedly exited. Status code was: N)processpollr r)r! return_coder#r#r$r2os z$Service.assert_process_still_runningcCs t|jS)zfEstablishes a socket connection to determine if the service running on the port is accessible.)r r3rr(r#r#r$r3us zService.is_connectablecCsTz t|jdWn tyYdSwtdD] }|s#dStdqdS)zcDispatch an HTTP request to the shutdown endpoint for the service in an attempt to stop it.z /shutdownNr/)rurlopenr)r ranger3r)r!_r#r#r$send_remote_shutdown_commandzs   z$Service.send_remote_shutdown_commandcCsz|jttjhvr t|jtr|jn t|jtr t|j|j dur;z| Wn t y4Ynw| dSdS)zStops the service.N) rrrrrrcloseintrr7r> TypeError_terminate_processr(r#r#r$stops        z Service.stopc CszM|jj|jj|jj}}}|||fD]}z|Wqty%Yqw|jz |jdWWdStj yMt j ddd|j YWdSwt y^t j dddYdSw)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)r7stdinstdoutstderrr?AttributeError terminatewaitrTimeoutExpiredloggererrorkillOSError)r!rFrGrHstreamr#r#r$rBs0     zService._terminate_processcCs$z|WdStyYdSwr,)rC Exceptionr(r#r#r$__del__s  zService.__del__r-c Cs|g}|||jdtdk}z>d}tdkr+t}tjtjB|_ tj |_ tj |f|j ||j|jt|j|d|j|_td|j|jj|j|jWdStyZtyz}z|jtjkrutdtj|jd|d}~ww)zmCreates a subprocess by executing the command provided. :param cmd: full command to execute close_fdsWindowsN)rrTrGrHrF creationflags startupinfozNStarted executable: `%s` in a child process with pid: %s using %s to output %s'z(' executable may have wrong permissions.)extendr+rrrr STARTUPINFOCREATE_NEW_CONSOLESTARTF_USESHOWWINDOWdwFlagsSW_HIDE wShowWindowPopenrrrrr7rMdebugrpidrArPerrnoEACCESr rr-basename)r!r-cmdclose_file_descriptors start_infoerrr#r#r$r1sR     zService._start_process)NrNN)rN)__name__ __module__ __qualname____doc__rr@r typingOptionalMappingAnyr%propertyr)rListr+r-setterr6r2boolr3r>rCrBrSr1r#r#r#r$r &sB        r )rcloggingrrrnabcrriorplatformrrtimerurllibr urllib.errorr selenium.common.exceptionsr selenium.typesr selenium.webdriver.commonr getLoggerrjrMr r#r#r#r$s$