o 5@Hh%@sddlZddlZddlZddlZddlmZmZddlmZddl m Z ddlm Z ddl m Z ddlmZmZmZmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZe e!Z"Gd ddeZ#dS)N)ABCabstractmethod)IOBase)system)PIPE)sleep)IOAnyListMappingOptionalUnioncast)request)URLError)WebDriverException)SubprocessStdAlias)utilsc @seZdZdZ     d'deededeedeee e fdeed df d d Z e d efd d Z e d eefddZe d efddZejded dfddZd(ddZd(ddZd efddZd(ddZd(ddZd(ddZd(d d!Zd"ed dfd#d$Zd eefd%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. Nrexecutable_pathport log_outputenvdriver_path_env_keyreturncKst|trttt|ddd|_n,|tjkr#ttt t tfd|_n|dus,|tj kr:ttt t tftj |_n||_|pBt |_|di|_|jdd|_|pWtj|_||_|pa||_dS)Nza+zutf-8)encodingpopen_kwcreation_flagsr) isinstancestrrropenr subprocessSTDOUTr r intDEVNULLr free_portrpoprrosenvironrDRIVER_PATH_ENV_KEYenv_path_path)selfrrrrrkwargsr.a/home/air/sanwanet/gpt-api/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)rjoin_host_portrr,r.r.r/ service_urlKszService.service_urlcCstd)z7A List of program arguments (excluding the executable).z2This method needs to be implemented in a sub class)NotImplementedErrorr3r.r.r/command_line_argsPszService.command_line_argscCs |jpdS)N)r+r3r.r.r/pathUs z Service.pathvaluecCst||_dSN)rr+)r,r9r.r.r/r8YscCsp|jdur td||jd} ||rdSttdd|d|d7}|d kr7td |jq) 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.rTg{Gz?g?g?FzCan not connect to the Service )r+r_start_processassert_process_still_runningis_connectablermin)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)processpollrr+)r, return_coder.r.r/r?ss z$Service.assert_process_still_runningcCs t|jS)zfEstablishes a socket connection to determine if the service running on the port is accessible.)rr@rr3r.r.r/r@ys 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<)rurlopenr4rranger@r)r,_r.r.r/send_remote_shutdown_command~s   z$Service.send_remote_shutdown_commandcCs|jttjhvr t|jtr|jn t|jtr t|j|j durP|j durRzz| Wn t y<YnwW| dSW| dS| wdSdS)zStops the service.N)rrr!r$rrcloser#r'rDrErK TypeError_terminate_processr3r.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)rDstdinstdoutstderrrLAttributeError terminatewaitr!TimeoutExpiredloggererrorkillOSError)r,rRrSrTstreamr.r.r/rNs0     zService._terminate_processcCs$z|WdStyYdSwr:)rO Exceptionr3r.r.r/__del__s  zService.__del__r8c Cs4|g}|||jdtdk}zTd}tdkr+t}tjtjB|_ tj |_ tj |f|j |ttttttf|jttttttf|jt|j|d|j|_td|j|jj|j|jWdStypty}z|jtjkr|jdurt dt dt!j"#|jd|d}~ww) zmCreates a subprocess by executing the command provided. :param cmd: full command to execute close_fdsWindowsN)rr`rSrTrR creationflags startupinfozNStarted executable: `%s` in a child process with pid: %s using %s to output %sr;'z(' executable may have wrong permissions.)$extendr6rr&rr! STARTUPINFOCREATE_NEW_CONSOLESTARTF_USESHOWWINDOWdwFlagsSW_HIDE wShowWindowPopenrrr r r#rr rrrrDrYdebugr+pidrMr\errnoEACCESrr'r8basename)r,r8cmdclose_file_descriptors start_infoerrr.r.r/r>sV      zService._start_processcCs|jr t|jdSdSr:)r)r'getenvr3r.r.r/r*szService.env_path)NrNNN)rN)__name__ __module__ __qualname____doc__r rr#rr r r0propertyr4rr r6r8setterrCr?boolr@rKrOrNr_r>r*r.r.r.r/r%sJ        ,r)$rologgingr'r!abcrriorplatformrrtimertypingrr r r r r rurllibr urllib.errorrselenium.common.exceptionsrselenium.typesrselenium.webdriver.commonr getLoggerrwrYrr.r.r.r/s"    $