a j,,@sdZddlmZmZmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z eeZeejdZdZdZGdddeZdZGd d d eZGd d d eZd dZeZdS)z(Easy to use python subprocess interface.) split_command unidecode uniencodeNlinkg?c@seZdZdddZddZdS)EasyProcessErrorcCs||_||_dSN) easy_processmsg)selfr r r BE:\soccer\ml_env\Lib\site-packages\sbvirtualdisplay/easyprocess.py__init__szEasyProcessError.__init__cCs|jdt|jS)N )r reprr r r r r __str__szEasyProcessError.__str__N)r)__name__ __module__ __qualname__rrr r r r rs rz)cmd=%s OSError=%s Program install error! c@s eZdZdZddZddZdS)EasyProcessCheckInstalledErrorzoThis exception is raised when a process run by check() returns a non-zero exit status or OSError is raised.cCs ||_dSr)r )r r r r r r)sz'EasyProcessCheckInstalledError.__init__cCsdt|jj|jjf}|jjr,|d|jj7}tddkr`|jjr`|d7}|d|jj7}|S)Nz home page: rZubuntuz! You can install it in terminal: zsudo apt-get install %s) templater cmdoserrorurlplatformdistlowerubuntu_packager r r r r r,sz&EasyProcessCheckInstalledError.__str__N)rrr__doc__rrr r r r r%src@seZdZdZd'ddZddZedd Zed d Zd(d dZ ddZ d)ddZ ddZ ddZ d*ddZddZddZddZdd Zd+d!d"Zd#d$Zd%d&ZdS), EasyProcessa .. module:: easyprocess simple interface for :mod:`subprocess` :param cmd: string ('ls -l') or list of strings (['ls','-l']) :param cwd: working directory :param use_temp_files: use temp files instead of pipes for stdout and stderr, pipes can cause deadlock in some cases (see unit tests) :param env: If *env* is not ``None``, it must be a mapping that defines the environment variables for the new process; these are used instead of inheriting the current process' environment, which is the default behavior. (check :mod:`subprocess` for more information) NTcCs||_d|_||_d|_d|_d|_d|_d|_||_||_ d|_ d|_ ||_ d|_ d|_d|_||_t|}||_d|j|_td|j td|jtd|jt|st|ddS)NFrz param: "%s" z command: %szjoined command: %szempty command!)use_temp_files_outputs_processedenvpopenstdoutstderr _stdout_file _stderr_filerr is_startedr cmd_param_thread _stop_threadtimeout_happenedcwdrrjoinZ cmd_as_stringlogdebuglenr)r rrrr/r"r$r r r rOs2 zEasyProcess.__init__c Cs.d|jj|j|j|j|j|j|j|jf}|S)Nz^<%s cmd_param=%s cmd=%s oserror=%s return_code=%s stdout="%s" stderr="%s" timeout_happened=%s>) __class__rr+rr return_coder&r'r.rr r r __repr__rszEasyProcess.__repr__cCs|jr|jjSdS)zH PID (:attr:`subprocess.Popen.pid`) :rtype: int N)r%pidrr r r r7szEasyProcess.pidcCs|jr|jjSdS)zV returncode (:attr:`subprocess.Popen.returncode`) :rtype: int N)r% returncoderr r r r5szEasyProcess.return_codercCs*|j}||k}|s&t|d||S)aRun command with arguments. Wait for command to complete. If the exit code was as expected and there is no exception then return, otherwise raise EasyProcessError. :param return_code: int, expected return code :rtype: self z$check error, return code is not {0}!)callr5rformat)r r5retokr r r checks zEasyProcess.checkcCs,z |Wnty&t|Yn0|S)aBUsed for testing if program is installed. Run command with arguments. Wait for command to complete. If OSError raised, then raise :class:`EasyProcessCheckInstalledError` with information about program installation. :param return_code: int, expected return code :rtype: self )r9 Exceptionrrr r r check_installeds   zEasyProcess.check_installedcCs$|j|d|r ||S)zRun command with arguments. Wait for command to complete. same as: 1. :meth:`start` 2. :meth:`wait` 3. :meth:`stop` :rtype: self timeout)startwaitis_alivestopr rAr r r r9szEasyProcess.callc Cs|jrt|d|jr@tjdd|_tjdd|_|j}|j}n tj}tj}t t t |j }ztj ||||j|jd|_Wn@ty}z(td|||_t|dWYd}~n d}~00d |_td |j|S) zSstart command in background and does not wait for it. :rtype: self zprocess was started twice!Zstdout_)prefixZstderr_)r&r'r/r$zOSError exception: %sz start errorNTzprocess was started (pid=%s))r*rr"tempfile TemporaryFiler(r) subprocessPIPElistmaprrPopenr/r$r%OSErrorr1r2rr7)r r&r'rrr r r rBs2    zEasyProcess.startcCs|jr|jduSdSdS)zW poll process using :meth:`subprocess.Popen.poll` :rtype: bool NF)r%pollrr r r rDszEasyProcess.is_alivecCsd|dur0|js0tj|jd|_d|j_|j|jrX|jj|d|jpR|j|_n||S)N)targetr@) r, threadingThread _wait4processdaemonrBr0r.ZisAliverFr r r rCs zEasyProcess.waitcCs|jr dSdd}|jr|jrtrL|jdur4qV|jr>dSttq$n |j d|_|j d|j d|j |_|j |_|j |j nd|_|j\|_|_td|t|j|_|t|j|_td|jtd|jtd|jdS) NcSs|dr|ddS|SdS)N )endswith)sr r r remove_ending_lfs  z3EasyProcess._wait4process..remove_ending_lfTrzprocess has endedzreturn code=%sz stdout=%sz stderr=%s)r#r%r"USE_POLLrPr-timesleep POLL_TIMErCr(seekr)readr&r'close communicater1r2rr5)r r[r r r rUs6        zEasyProcess._wait4processcCs |Sr)sendstoprCrr r r rEszEasyProcess.stopc Cs|jst|dtd|j|j|jr|rtdz8z|jWn$t ynt |jjt j Yn0Wqty}ztd|WYd}~qd}~00qtdn td|S)z Kill process (:meth:`subprocess.Popen.terminate`). Do not wait for command to complete. :rtype: self zprocess was not started!z"stopping process (pid=%s cmd="%s")z$process is active -> sending SIGTERMzexception in terminate:%sNzprocess was already stoppedzprocess was not started)r*rr1r2r7rr%rD terminateAttributeErroroskillsignalZSIGKILLrO)r rr r r rd"s    $  zEasyProcess.sendstopcCst||Sr)r]r^)r secr r r r^<s zEasyProcess.sleepcsfdd}|S)z returns a function which: 1. start process 2. call func, save result 3. stop process 4. returns result similar to :keyword:`with` statement :rtype: c srd}zXz }Wn@tyf}z(td||_tdWYd}~n d}~00Wn 0|S)NzOSError exception:%sz wrap error!)rBr^rOr1r2rrrE)xrdelayfuncr r r wrappedJs   "z!EasyProcess.wrap..wrappedr )r rnrmror rlr wrap@s zEasyProcess.wrapcCs ||S)%used by the :keyword:`with` statement)rBrr r r __enter__ZszEasyProcess.__enter__cGs |dS)rqN)rE)r exc_infor r r __exit___szEasyProcess.__exit__)NNNTN)r)N)N)r)rrrr rr6propertyr7r5r=r?r9rBrDrCrUrErdr^rprrrtr r r r r!=s2 #     & r!cCsj|dd}d}t|D]H}t|dkr|ddkrH|dd}d|vr|dr|}qfq|S) zQThis function tries to extract the version from the help text of any program.,rNrvrR.)replacesplitreversedr3risdigit)txtwordsversionrkr r r extract_versionds   r)r Zsbvirtualdisplay.unicodeutilrrrloggingos.pathrgrrirJrHrSr] getLoggerrr1setLevelERRORZ SECTION_LINKr_r\r>rrrobjectr!rZProcr r r r s,   )