ó æNXc@s}dZddlmZddlZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZmZmZdd lmZdd lmZd efd „ƒYZd „Zd„Zd„Zd„ZyjejjjZe eeƒge_ee ƒe_ ejj!j"Z"e e"_#e ge"_e$e%d„ZWne&k roeZnXd„Z'dS(sWindows-specific implementation of process utilities. This file is only meant to be imported by process.py, not by end-users. iÿÿÿÿ(tprint_functionN(tc_inttPOINTER(tLPCWSTRtHLOCAL(tSTDOUTi(tread_no_interrupttprocess_handlert arg_split(t py3compat(tDEFAULT_ENCODINGt AvoidUNCPathcBs eZdZd„Zd„ZRS(sÂA 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) cCsItjƒ|_|jjdƒ|_|jrAtjdƒ|jSdSdS(Ns\\sC:(R tgetcwdtpatht startswitht is_unc_pathtostchdirtNone(tself((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt __enter__8s   cCs |jrtj|jƒndS(N(RRRR (Rtexc_typet exc_valuet traceback((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt__exit__Ds (t__name__t __module__t__doc__RR(((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyR $s cCsµyddlm}Wntk r3tdƒ‚n~Xtjd}ddddg}d }x2|D]*}y||||ƒd }Wq`q`Xq`W|d kr­td |ƒ‚n|Sd S( s?Find the full path to a .bat or .exe using the win32api module.iÿÿÿÿ(t SearchPaths3you need to have pywin32 installed for this to worktPATHs.exes.coms.bats.pyiscommand %r not foundN(twin32apiRt ImportErrorRtenvironRtOSError(tcmdRRt extensionsR text((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt _find_cmdIs    cCsšt}xBt|jƒjƒD]+}|j|dƒ}t|dtjƒqWxBt|jƒjƒD]+}|j|dƒ}t|dtjƒqaW|jƒS(sCallback for _system.treplacetfile( R Rtstdoutt splitlinestdecodetprinttsyststderrtwait(tptenctline((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt _system_body^scCsBtƒ3}|dk r+d||f}nt|tƒSWdQXdS(sÜWin32 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 ------- None : we explicitly do NOT return the subprocess status code, as this utility is meant to be used extensively in IPython, where any return value would trigger :func:`sys.displayhook` calls. s"pushd %s &&"%sN(R RRR2(R"R ((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pytsystemls  cCshtƒ;}|dk r+d||f}nt|d„tƒ}WdQX|dkr[d}ntj|ƒS(sûReturn 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 s"pushd %s &&"%scSs|jƒdS(Ni(t communicate(R/((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt˜sNt(R RRRR t bytes_to_str(R"R tout((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt getoutput†s    c Cs­|jƒdkrgS|s2t|d|d|ƒStƒ}ttj|jƒƒtj|ƒƒ}t |j }g|j tj |j ƒƒD] }|^q‹}t|ƒ}|S(s;Split 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 paramter is ignored. If strict=False, process_common.arg_split(...strict=False) is used instead. R6tposixtstrict(tstript py_arg_splitRtCommandLineToArgvWR t cast_unicodetlstriptctypestbyrefRtvaluet from_addresst addressoftcontentst LocalFree( t commandlineR:R;targvntresult_pointertresult_array_typetargtresulttretval((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyR¦s  ' . cCsttjjjdd|ƒƒS(Nii(tboolRAtwindlltkernel32t OpenProcess(tpid((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyt check_pid½s((Rt __future__RRR,RARRtctypes.wintypesRRt subprocessRt_process_commonRRRR=R6R tencodingR tobjectR R%R2R3R9RPtshell32R>t arg_typestrestypeRQRGtres_typetFalsetTruetAttributeErrorRT(((sT/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/utils/_process_win32.pyts4    %