U g %@sddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddlm Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZeeZGdddeZdS)N)ABC)abstractmethod)IOBase)system)PIPE)sleep)Optional)cast)request)URLError)WebDriverException)SubprocessStdAlias)utilsc@seZdZdZd#eeeejej ej ej feddddZ e eddd Z eejedd d Ze edd d Zejedddd ZddddZddddZedddZddddZddddZddddZddddZedddd Zeedd!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`. Nr)executable_pathport log_outputenvdriver_path_env_keyreturncKst|tr"ttt|ddd|_n`|tjkrJttj tj t tfd|_n8|dks\|tj kr|ttj tj t tftj |_n||_|pt |_|di|_|jdd|_|ptj|_||_|p||_dS)Nza+zutf-8)encodingpopen_kwcreation_flagsr) isinstancestrr ropenr subprocessSTDOUTtypingrUnionintDEVNULLrZ free_portrpoprrosenvironrDRIVER_PATH_ENV_KEYenv_path_path)selfrrrrrkwargsr*E/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/service.py__init__3s    zService.__init__)rcCsdtd|jS)zGets the url of the Service.zhttp:// localhost)rZjoin_host_portrr(r*r*r+ service_urlMszService.service_urlcCs tddS)z7A List of program arguments (excluding the executable).z2This method needs to be implemented in a sub classN)NotImplementedErrorr.r*r*r+command_line_argsRszService.command_line_argscCs |jpdS)N)r'r.r*r*r+pathWsz Service.path)valuercCst||_dSN)rr')r(r4r*r*r+r3[scCsp|jdkrtd||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+start_s  z Service.startcCs(|j}|r$td|jd|dS)z1Check if the underlying process is still running.zService z' unexpectedly exited. Status code was: N)processpollr r')r(Z return_coder*r*r+r:us z$Service.assert_process_still_runningcCs t|jS)zfEstablishes a socket connection to determine if the service running on the port is accessible.)rr;rr.r*r*r+r;{szService.is_connectablecCsTzt|jdWntk r,YdSXtdD]}|sFqPtdq6dS)zcDispatch an HTTP request to the shutdown endpoint for the service in an attempt to stop it.z /shutdownNr7)r urlopenr/r ranger;r)r(_r*r*r+send_remote_shutdown_commands z$Service.send_remote_shutdown_commandcCsx|jttjhkr@t|jtr(|jnt|jtr@t|j|j dk rtz | Wnt k rjYnX| dS)zStops the service.N) rrrr!rrcloser r#r?rE TypeError_terminate_processr.r*r*r+stops      z Service.stopc Csz|jj|jj|jj}}}|||fD]&}z |Wq(tk rLYq(Xq(|jz|jdWn.tj k rt j ddd|j YnXWn"t k rt j dddYnXdS)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)r?stdinstdoutstderrrFAttributeError terminatewaitrTimeoutExpiredloggererrorkillOSError)r(rLrMrNstreamr*r*r+rHs*   zService._terminate_processcCs&z |Wntk r YnXdSr5)rI Exceptionr.r*r*r+__del__s zService.__del__)r3rc Cs\|g}|||jdtdk}zd}tdkrVt}tjtjB|_ tj |_ tj |f|j |ttjtjttjtjf|jttjtjttjtjf|jt|j|d|j|_td|j|jj|j|jWnxtk rYndtk rV}zD|jtj krD|jdkr(t!dt!dt"j#$|jd|W5d}~XYnXdS) zmCreates a subprocess by executing the command provided. :param cmd: full command to execute close_fdsWindowsN)rrZrMrNrL creationflags startupinfozNStarted executable: `%s` in a child process with pid: %s using %s to output %sr6'z(' executable may have wrong permissions.)%extendr1rr"rrZ STARTUPINFOZCREATE_NEW_CONSOLEZSTARTF_USESHOWWINDOWZdwFlagsZSW_HIDEZ wShowWindowPopenrr rrrr IOAnyrrrr?rSdebugr'pidrGrVerrnoEACCESr r#r3basename)r(r3cmdZclose_file_descriptorsZ start_infoerrr*r*r+r9sR ""   zService._start_processcCst|jdSr5)r#getenvr%r.r*r*r+r&szService.env_path)NrNNN)__name__ __module__ __qualname____doc__rr r rrMappingrbr,propertyr/rListr1r3setterr>r:boolr;rErIrHrYr9r&r*r*r*r+r(s>    ,r)reloggingr#rrabcrriorplatformrrtimerrr urllibr urllib.errorr Zselenium.common.exceptionsr Zselenium.typesr Zselenium.webdriver.commonr getLoggerrkrSrr*r*r*r+s&