U -eQS @s`dZddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZee ZeeZeeZGdddejZeeZGdddejZeeZGd d d ejZeeZd Zd Zd ZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)ej*j+j,Z,ge,_-e e,_.ej*j+j/Z0ee e e e e ege0_-ee0_.ej*j+j1Z1eeeeee ge1_-e e1_.ej*j+j2Z3eeeee e e eeeg e3_-e e3_.ej*j+j4Z4eege4_-e e4_.ej*j+j5Z5ge5_-ee5_.ej*j+j6Z6ege6_-e e6_.ej*j+j7Z7ee e ee ge7_-e e7_.ej*j+j8Z8ee e ee ge8_-e e8_.ej*j+j9Z9eege9_-e e9_.ej*j+j:Z:ee ge:_-e e:_.ej*j+j;Z;ege;_-e e;_.ej*j+jZ>ee e ge>_-e e>_.ej*j+j?Z?ege?_-e e?_.ej*j@jAZAeeejBgeA_-eeeA_.ej*j+jCZCe geC_-e eC_.GdddeDZEGdddeDZFddZGeHdkr\eIdeGdeId dS)!zvWindows-specific implementation of process utilities with direct WinAPI. This file is meant to be used by process.py N)POINTER)HANDLEHLOCALLPVOIDWORDDWORDBOOLULONGLPCWSTRc@s"eZdZdefdefdefgZdS)SECURITY_ATTRIBUTESnLengthlpSecurityDescriptorbInheritHandleN)__name__ __module__ __qualname__rrr_fields_rrh/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/IPython/utils/_process_win32_controller.pyr sr c@s|eZdZdefdefdefdefdefdefdefdefd efd efd efd efd efdefdefdefdefdefgZdS) STARTUPINFOcbZ lpReservedZ lpDesktopZlpTitleZdwXZdwYZdwXSizeZdwYSizeZ dwXCountCharsZ dwYCountCharsZdwFillAttributedwFlagsZ wShowWindowZ cbReserved2Z lpReserved2 hStdInput hStdOutput hStdErrorN) rrrrr rrrrrrrrrs&rc@s(eZdZdefdefdefdefgZdS)PROCESS_INFORMATIONhProcesshThreadZ dwProcessIdZ dwThreadIdN)rrrrrrrrrrr2s r&miiic@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) cCs6t|_|jd|_|jr.td|jSdSdS)Nz\\zC:)osgetcwdpath startswith is_unc_pathchdirselfrrr __enter__s   zAvoidUNCPath.__enter__cCs|jrt|jdSN)r,r(r-r*r/exc_type exc_value tracebackrrr__exit__szAvoidUNCPath.__exit__N)rrr__doc__r0r6rrrrr's r'c@sleZdZdZdddZddZddZd d Zdd d ZddZ ddZ ddZ ddZ ddZ ddZd S)Win32ShellCommandControlleraRuns a shell command in a 'with' context. This implementation is Win32-specific. Example: # Runs the command interactively with default console stdin/stdout with ShellCommandController('python -i') as scc: scc.run() # Runs the command using the provided functions for stdin/stdout def my_stdout_func(s): # print or save the string 's' write_to_stdout(s) def my_stdin_func(): # If input is available, return it as a string. if input_available(): return get_input() # If no input available, return None after a short delay to # keep from blocking. else: time.sleep(0.01) return None with ShellCommandController('python -i') as scc: scc.run(my_stdout_func, my_stdin_func) TcCs||_||_dS)aUInitializes the shell command controller. The cmd is the program to execute, and mergeout is whether to blend stdout and stderr into one output in stdout. Merging them together in this fashion more reliably keeps stdout and stderr in the correct order especially for interactive shell usage. N)cmdmergeout)r/r9r:rrr__init__s z$Win32ShellCommandController.__init__c s|j}|j}d\|_|_|_d|_zddgd\}}}}}}tt _ d_ d_ fdd} | dd\}}|rt }tt|tt|ddtstn| dd\}}| dd\}}t} t} t | | _|| _|| _|| _t| _ttB} tdd |ddd| ddt| t|  s.tt|d}t|d}|dk r\t|d}||_d}||_d}|s||_d}| |_W5|rt||rt||rt||rt||rt||rt|X|S) N)NNNTcshttf}tt|dt|dtds.create_piper)rDr!z cmd.exe /c )r9r:hstdouthstdinhstderr piProcInfo CloseHandler r>Zsizeofr rr rDuplicateHandleGetCurrentProcessr?DUPLICATE_SAME_ACCESSr@rrrrrrSTARTF_USESTDHANDLESrCREATE_SUSPENDEDCREATE_NO_WINDOW CreateProcess) r/r9r:Zp_hstdinZc_hstdinZ p_hstdoutZ c_hstdoutZ p_hstderrZ c_hstderrrFrJZ siStartInfoZdwCreationFlagsrrErr0s       z%Win32ShellCommandController.__enter__cCst}td}|}|dkrbt|t|s4t|jtkrBdSt|ddt|dstqt|t rv| d}t|t st dt |dkrdS||t |dkrt||t |t|dsttkrdSt||jd}qqdS)Nrutf_8z$internal stdin function string error)rGetExitCodeProcessr>r?r@rC STILL_ACTIVE WriteFile isinstanceunicodeencodestr RuntimeErrorlen GetLastError ERROR_NO_DATA)r/handleZhprocessfunc stdout_funcZexitCodeZ bytesWrittendatarrr _stdin_threadEs<        z)Win32ShellCommandController._stdin_threadcCshtd}td}t||dt|dsBt}|tkr:dSt|jd|j}|| ddq dS)NirrTreplace) r>create_string_bufferrReadFiler?r^ERROR_BROKEN_PIPEr@rCdecode)r/r`rarc bytesReadlesrrr_stdout_threadws z*Win32ShellCommandController._stdout_threadNcCs(|dkr |dkr |dkr |S|dk r6|jr6tdd}g}|r`tj|j|j|jj||fd}| tj|j |j |fd|js|dkr|}| tj|j |j |fdt |jjdkrt|dk r||D] }|qt|jjttkrt|D]}|q|dk r$|dS)aRuns the process, using the provided functions for I/O. The function stdin_func should return strings whenever a character or characters become available. The functions stdout_func and stderr_func are called whenever something is printed to stdout or stderr, respectively. These functions are called from different threads (but not concurrently, because of the GIL). NzqShell command was initiated with merged stdin/stdout, but a separate stderr_func was provided to the run() method)targetargsr%) _run_stdior:r\ threadingThreadrdrHrJrappendrmrGrI ResumeThreadrr>r@startWaitForSingleObjectINFINITE WAIT_FAILEDjoin)r/rb stdin_func stderr_funcZ stdin_threadthreadsthreadrrrrunsH        zWin32ShellCommandController.runcCsttj}t|d}|tkr,tn|t krDt ddddSt d}t d}t dddt ||dt|dstt||j}|d d }|d d }t t|d dd|SdS) z>Use the raw Win32 handle of sys.stdin to do non-blocking readsd.rS)endNr"r?z    )msvcrtZ get_osfhandlesysstdinfilenorvrxr>r@ WAIT_TIMEOUTprintrfrrgr?FlushConsoleInputBufferrCrerepr)r/r`resultrcrjrrr_stdin_raw_nonblocks*       z/Win32ShellCommandController._stdin_raw_nonblockc Cs^ztjd}|dd}|WStk rX}z|jtkrDWY dS|W5d}~XYnXdS)zUse a blocking stdin readr!rrN)rrreadreZ WindowsErrorwinerrorr_)r/rcZwerrr_stdin_raw_blocks    z,Win32ShellCommandController._stdin_raw_blockcCst|dtjdtjdSzWrites the string to stdoutrS)rfileN)rrstdoutflushr/rlrrr _stdout_rawsz'Win32ShellCommandController._stdout_rawcCst|dtjdtjdSr)rrstderrrrrrr _stderr_rawsz'Win32ShellCommandController._stderr_rawcCs2|jr|j|j|jdS|j|j|j|jdSdS)zRuns the process using the system standard I/O. IMPORTANT: stdin needs to be asynchronous, so the Python sys.stdin object is not used. Instead, msvcrt.kbhit/getwch are used asynchronously. )rbrz)rbrzr{N)r:r~rrrr.rrrrpsz&Win32ShellCommandController._run_stdiocCsn|jrt|jd|_|jr,t|jd|_|jrBt|jd|_|jdk rjt|jjt|jjd|_dSr1)rHrKrGrIrJrrr2rrrr6s      z$Win32ShellCommandController.__exit__)T)NNN)rrrr7r;r0rdrmr~rrrrrpr6rrrrr8s `2 3r8c CsFt6}|dk rd||f}t|}|W5QRXW5QRXdS)aWin32 version of os.system() that works with network shares. Note that this implementation returns None, as meant for use in IPython. Parameters ---------- cmd : str 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. Nz"pushd %s &&"%s)r'r8r~)r9r*Zsccrrrsystem"s   r__main__zTest starting!z python -izTest finished!)Jr7r(rrqr>rrZctypes.wintypesrrrrrrr r ZLPDWORDZLPHANDLEZ ULONG_PTRZ Structurer ZLPSECURITY_ATTRIBUTESrZ LPSTARTUPINFOrZLPPROCESS_INFORMATIONZERROR_HANDLE_EOFrhr_rBrOrPZCREATE_NEW_CONSOLErQrVrrxrwrNZENABLE_ECHO_INPUTZENABLE_LINE_INPUTZENABLE_PROCESSED_INPUTwindllkernel32r^ZargtypesrestypeZ CreateFileWZ CreateFiler=ZCreateProcessWrRrUrMrtrgrWZGetConsoleModeZSetConsoleModerrvrLrArKZshell32ZCommandLineToArgvWc_intZ LocalFreeobjectr'r8rrrrrrrs (                         %g