Mhw dZddlZddlZddlZddlZddlZddlmZmZddlm Z m Z ddlm Z ddl m Z ddlmZddlmZmZd d lmZd d lmZd d lmZmZd d lmZGddZdejdefdZde deefdZ!de de fdZ" ej#Z#e#j$j%Z%e eege%_&ee e%_'e#j(j)Z)e e)_*e ge)_& dde de+de+dee fdZn #e,$reZYnwxYwdede+fdZ-dS)zWindows-specific implementation of process utilities. This file is only meant to be imported by process.py, not by end-users. N)POINTERc_int)HLOCALLPCWSTR)STDOUT)Thread) TracebackType)ListOptional) py3compat) arg_split)process_handlerread_no_interrupt)DEFAULT_ENCODINGcfeZdZdZdeefdZdeeedeede ddfdZ dS) 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) returnctj|_|jd|_|jrtjd|jSdS)Nz\\zC:)osgetcwdpath startswith is_unc_pathchdir)selfs \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/IPython/utils/_process_win32.py __enter__zAvoidUNCPath.__enter__.sNIKK 9//66    HTNNN9 4exc_type exc_value tracebackNcJ|jrtj|jdSdSN)rrrr)rr r!r"s r__exit__zAvoidUNCPath.__exit__:s1   HTY       r) __name__ __module__ __qualname____doc__r strrtype BaseExceptionr r%rrrrs( 8C=     4 ./ M* !         rrprcptdfd }dfd }t|}t|}|| }|t jdnn-|||S) zCallback for _system.rNc< jJtjD]3}|d}t |t j4dS#t $r)}t d|t jYd}~dSd}~wwxYw)NreplacefilezError reading stdout: )stdoutr splitlinesdecodeprintsys Exceptionstderr byte_linelineeencr.s r stdout_readz!_system_body..stdout_readK A8'''.qx88CCEE - -  ''Y77d,,,,, - - A A A .1..SZ @ @ @ @ @ @ @ @ @ @ AA#A(( B2BBc< jJtjD]3}|d}t |t j4dS#t $r)}t d|t jYd}~dSd}~wwxYw)Nr1r2zError reading stderr: )r:rr5r6r7r8r9r;s r stderr_readz!_system_body..stderr_readTrArB)targetTg{Gz?)rN)rrstartpolltimesleepjoin)r.r@rD stdout_thread stderr_threadresultr?s` @r _system_bodyrNDs C AAAAAAAAAAAAAA+...M+...M  > Jt       Mrcmdct5}|d|d|}t|t}t|tt dzsJ|cdddS#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  &&")rrrN isinstanceintr+)rOrress rsystemrVts& 4   '+ttSS1Cc<00#sT$ZZ/00000  4sAA$$A(+A(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 NrQrRc6|dS)Nr) communicate)r.s rzgetoutput..sQ]]__Q-?rr)rrrr r6)rOrouts r getoutputr\s I4   '+ttSS1Cc#?#?HHIIIIIIIIIIIIIII  {  C  s"=AAFT commandlineposixstrictc|dkrgS|st|||St}t|t j|} t|jz}d| t j |j D}t|}n#t|}wxYw|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. )r^r_cg|]}||Sr$r-).0args r zarg_split..s)? #??r) strip py_arg_splitrCommandLineToArgvWlstripctypesbyrefrvalue from_address addressofcontents LocalFree)r]r^r_argvnresult_pointerresult_array_typerM_s rrrs     " $ $I I 5HHH H+K,>,>,@,@&,uBUBUVV * '%+ 5 ,99$^%<==F.))AA .))A ) ) ) ) s 2ACCpidc^ttjdd|S)Nr r)boolwindllkernel32 OpenProcess)rus r check_pidr{s& ++Aq#66 7 77r)FT).r)rjr subprocessr8rHrrctypes.wintypesrrr threadingrtypesr typingr r rar _process_commonrrgrrencodingrrPopenrTrNr*rVr\rxshell32rh arg_typesrestyperyrpres_typerwAttributeErrorr{r-rrrs   !!!!!!!!++++++++!!!!!!!!666666????????&&&&&&( ( ( ( ( ( ( ( V-J$-----` 8!3!3!!!!2) ]F:$+WWU^^#< !(!1!1)II!(I?C!%7; c>III8384888888s A$C11C;:C;