&edZddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z m Z ddl mZddlmZmZmZdd lmZdd lmZGd d eZd ZdZdZ ejjjZeeege_ eee_!ejj"j#Z#e e#_$e ge#_ ddZn #e%$reZYnwxYwdZ&dS)zWindows-specific implementation of process utilities. This file is only meant to be imported by process.py, not by end-users. N)c_intPOINTER)LPCWSTRHLOCAL)STDOUTTimeoutExpired)Thread)read_no_interruptprocess_handler arg_split) py3compat)DEFAULT_ENCODINGceZdZdZdZdZdS) AvoidUNCPathaA context manager to protect command execution from UNC paths. In the Win32 API, commands can't be invoked with the cwd being a UNC path. This context manager temporarily changes directory to the 'C:' drive on entering, and restores the original working directory on exit. The context manager returns the starting working directory *if* it made a change and None otherwise, so that users can apply the necessary adjustment to their system calls in the event of a change. Examples -------- :: cmd = 'dir' with AvoidUNCPath() as path: if path is not None: cmd = '"pushd %s &&"%s' % (path, cmd) os.system(cmd) ctj|_|jd|_|jrtjd|jSdS)Nz\\zC:)osgetcwdpath startswith is_unc_pathchdir)selfs .stdout_readN]%ah//::<< ) )D;;sI..D $SZ ( ( ( ( ( ) )rctjD]3}|d}t |t j4dSr))r stderrr-r.r/r0r1s r stderr_readz!_system_body..stderr_readSr6r)targetTNg{Gz?)rr startpolltimesleep)r4r5r9resultr3s` @r _system_bodyr@Js C)))))) ))))))  +$$&&& +$$&&&  > Jt    M rct5}|d|d|}t|tcdddS#1swxYwYdS)a7Win32 version of os.system() that works with network shares. Note that this implementation returns None, as meant for use in IPython. Parameters ---------- cmd : str or list A command to be executed in the system shell. Returns ------- int : child process' exit code. N"pushd  &&")rr r@)cmdrs rsystemrEfs& 24   '+ttSS1CsL11222222222222222222s ;??ct5}|d|d|}t|dt}dddn #1swxYwY|d}tj|S)zReturn standard output of executing cmd in a shell. Accepts the same arguments as os.system(). Parameters ---------- cmd : str or list A command to be executed in the system shell. Returns ------- stdout : str NrBrCc6|dS)Nr) communicate)r4s rzgetoutput..sQ]]__Q-?rr)rr rrr.)rDrouts r getoutputrK~s I4   '+ttSS1Cc#?#?HHIIIIIIIIIIIIIII  {  C  s"=AAFTc|dkrgS|st|||St}tt j|tj|}t|j z}d| tj |j D}t|}|S)a4Split a command line's arguments in a shell-like manner. This is a special version for windows that use a ctypes call to CommandLineToArgvW to do the argv splitting. The posix parameter is ignored. If strict=False, process_common.arg_split(...strict=False) is used instead. )posixstrictcg|]}|Sr&r&).0args r zarg_split..skkk##kkkr)strip py_arg_splitrCommandLineToArgvWr cast_unicodelstripctypesbyrefrvalue from_address addressofcontents LocalFree) commandlinerNrOargvnresult_pointerresult_array_typer?retvals rr r s     " $ $I I 5HHH H+I,B;CUCUCWCW,X,XZ`ZfglZmZmnn#ek1kk!2!?!?@PQ_Qh@i@i!j!jkkk>** rchttjjdd|S)Nr r)boolrYwindllkernel32 OpenProcess)pids r check_pidrks)  &221Qs;; < <rws" !!!!!!!!++++++++--------[ZZZZZZZZZ&&&&&& " " " " " 6" " " J82220!!!0.A$+WWU^^#< !(!1!1 &0II!(I(III=====sAB77CC