fdZddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z m Z ddl mZddlmZmZmZdd lmZdd lmZGd d eZd ZdZdZ ej:j<j>Zeeege_ eee_!ej:jDjFZ#e e#_$e ge#_ ddZdZ&y#e%$reZYdZ&ywxYw)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_ENCODINGceZdZdZdZdZy) 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|_|jjd|_|jr!tj d|jSy)Nz\\zC:)osgetcwdpath startswith is_unc_pathchdir)selfs W/var/www/cvtools/html/venv/lib/python3.12/site-packages/IPython/utils/_process_win32.py __enter__zAvoidUNCPath.__enter__9sIIIK 99//6    HHTN99 c\|jr tj|jyy)N)rrrr)rexc_type exc_value tracebacks r__exit__zAvoidUNCPath.__exit__Es    HHTYY  rN)__name__ __module__ __qualname____doc__rr!rrrr%s&  rrctfd}fd}t|jt|j j}|t j dn|S+)zCallback for _system.ctjjD]/}|jd}t |t j1yNreplace)file)r stdout splitlinesdecodeprintsyslineencps r stdout_readz!_system_body..stdout_readNB%ahh/::< )D;;sI.D $SZZ ( )rctjjD]/}|jd}t |t j1yr))r stderrr-r.r/r0r1s r stderr_readz!_system_body..stderr_readSr6r)targetg{Gz?)rr startpolltimesleep)r4r5r9resultr3s` @r _system_bodyr@Jsa C) )  +$$& +$$&  > JJt M rcvt5}|d|d|}t|tcdddS#1swYyxYw)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  '+S1CsL1222s/8ct5}|d|d|}t|dt}dddd}tj|S#1swY"xYw)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 NrBrCc(|jdS)Nr) communicate)r4s rzgetoutput..sQ]]_Q-?rr)rr rrr.)rDrouts r getoutputrK~s` I4  '+S1Cc#?HI  {   C  IIs A  Ac|jdk(rgS|st|||St}tt j |j tj|}t|jz}|jtj|jDcgc]}|}}t|}|Scc}w)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. )posixstrict)strip py_arg_splitrCommandLineToArgvWr cast_unicodelstripctypesbyrefrvalue from_address addressofcontents LocalFree) commandlinerNrOargvnresult_pointerresult_array_typeargr?retvals rr r s    " $I 5H H+I,B,B;CUCUCW,XZ`ZfZfglZmn#ekk1!2!?!?@P@PQ_QhQh@i!jk##kk>* ls3 C cjttjjj dd|S)Nr r)boolrUwindllkernel32 OpenProcess)pids r check_pidrhs(  &&221Qs; <rts" !+-[Z& " 6" J820!0..AA$+WU^#< !(!1 &&00II!(I.=I=sA/CCC