o h@s dZddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z m Z ddl mZddlmZmZmZdd lmZdd lmZGd d d eZd dZddZddZz$ejjjZeeege_ eee_!ejj"j#Z#e e#_$e ge#_ dddZWn e%yeZYnwddZ&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_ENCODINGc@s eZdZdZddZddZdS) 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) cCs2t|_|jd|_|jrtd|jSdS)Nz\\zC:)osgetcwdpath startswith is_unc_pathchdir)selfrRc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/utils/_process_win32.py __enter__9s  zAvoidUNCPath.__enter__cCs|jr t|jdSdS)N)rrrr)rexc_type exc_value tracebackrrr__exit__EszAvoidUNCPath.__exit__N)__name__ __module__ __qualname____doc__rrrrrrr%s rcs`tfdd}fdd}t|dt|d }|dur-tdn|Sq) zCallback for _system.c2tjD]}|d}t|tjdqdSNreplace)file)r stdout splitlinesdecodeprintsyslineencprr stdout_readN z!_system_body..stdout_readcr"r#)r stderrr'r(r)r*r+r-rr stderr_readSr1z!_system_body..stderr_read)targetTNg{Gz?)rrstartpolltimesleep)r/r0r3resultrr-r _system_bodyJs r:cCsHt}|durd||f}t|tWdS1swYdS)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 %s &&"%s)rr r:)cmdrrrrsystemfs  $r=cCs`t}|durd||f}t|ddt}Wdn1s wY|dur+d}t|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 Nr;cSs |dS)Nr) communicate)r/rrrs zgetoutput..)rr rr r()r<routrrr getoutput~s  rBFTcCsn|dkrgS|st|||dSt}t|t|}t|j}dd| t |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. )posixstrictcSsg|]}|qSrr).0argrrr szarg_split..) strip py_arg_splitrCommandLineToArgvWlstripctypesbyrefrvalue from_address addressofcontents LocalFree)Z commandlinerDrEZargvnZresult_pointerZresult_array_typer9retvalrrrr s  r cCsttjjdd|S)Nr r)boolrMwindllkernel32 OpenProcess)pidrrr check_pidsrZ)FT)'r!rr*rMr7rrZctypes.wintypesrr subprocessrr threadingrZ_process_commonr r r rJrCr encodingrobjectrr:r=rBrVshell32rKZ arg_typesrestyperWrSZres_typeAttributeErrorrZrrrrs8   %