B 7Óþ[*ã@s®dZddlmZdZdddddgZdd lmZdd lZdd lZdd l m Z m Z m Z dd l Z dd l Z ydd lmZWn ek r’dd lmZYnXdd lZdd lZdd lZdd lZdd lZe d ¡Zy ej ed ¡Zej ed¡ZWn2ek rej ed¡Zej ed¡ZYnXdZdZeej dd ƒp4dZ!e! "¡dkrHdZ!d!dd„Z#Gdd„de$ƒZ%eeee!fdd„ƒZ&e!fdd„Z'd a(e!fdd„Z)dd„Z*dd„Z+dd „Z,d S)"zhCapture C-level FD output on pipes Use `wurlitzer.pipes` or `wurlitzer.sys_pipes` as context managers. é)Úprint_functionz1.0.2ÚpipesÚ sys_pipesÚsys_pipes_foreverÚstop_sys_pipesÚ Wurlitzer)ÚcontextmanagerN)ÚfcntlÚF_GETFLÚF_SETFL)ÚQueueÚstdoutÚstderrZ __stdoutpZ __stderrpééÚencodingÚutf8Úasciic Csxd}xfttd|ƒƒD]R}y t ||¡Stk rf}z"|}|jtjkrTt d¡n‚Wdd}~XYqXqW|rt|‚dS)z Like os.dup2, but retry on EBUSYNé gš™™™™™¹?) ÚrangeÚintÚosÚdup2ÚOSErrorÚerrnoZEBUSYÚtimeZsleep)ÚaÚbZtimeoutZdup_errÚiÚe©r ú(lib/python3.7/site-packages/wurlitzer.pyr4s   rc@sdeZdZdZdZddefdd„Zdd„Zdd „Zd d „Z d d „Z dd„Z dd„Z dd„Z dd„ZdS)rziClass for Capturing Process-level FD output via dup2 Typically used via `wurlitzer.capture` gš™™™™™É?NcCsR||_|tkr|j|_n||_||_i|_i|_i|_|j|jd<|j|jd<dS)a3 Parameters ---------- stdout: stream or None The stream for forwarding stdout. stderr = stream or None The stream for forwarding stderr. encoding: str or None The encoding to use, if streams should be interpreted as text. rr N) Ú_stdoutÚSTDOUTÚ_stderrrÚ _save_fdsÚ _real_fdsZ _handlersÚ_handle_stderrÚ_handle_stdout)Úselfr rrr r r!Ú__init__Ls   zWurlitzer.__init__cCspttd|ƒ ¡}t |¡}||j|<t ¡\}}t||ƒt |¡||j |<t |t ƒ}t |t |tj Bƒ|S)Nz__%s__)ÚgetattrÚsysÚfilenorÚdupr%ÚpiperÚcloser&r r r Ú O_NONBLOCK)r)ÚnameÚreal_fdÚsave_fdZpipe_outZpipe_inÚflagsr r r!Ú _setup_pipecs       zWurlitzer._setup_pipecCs|jr| |jd¡}|S)z\Decode data, if any Called before passing to stdout/stderr streams Úreplace)rÚdecode)r)Údatar r r!Ú_decoderszWurlitzer._decodecCs|jr|j | |¡¡dS)N)r"Úwriter:)r)r9r r r!r({szWurlitzer._handle_stdoutcCs|jr|j | |¡¡dS)N)r$r;r:)r)r9r r r!r'szWurlitzer._handle_stderrcCs|j|jf|_dS)zSetup handle for output, if anyN)r"r$Úhandle)r)r r r!Ú _setup_handleƒszWurlitzer._setup_handlecCsdS)zCFinish handle, if anything should be done when it's all wrapped up.Nr )r)r r r!Ú_finish_handle‡szWurlitzer._finish_handlecsèt t¡t t¡ˆ ¡t ¡\ˆ_ˆ_ˆjg‰ˆjdi‰ˆj r`ˆ  d¡}ˆ  |¡dˆ|<ˆj r‚ˆ  d¡}ˆ  |¡dˆ|<t ƒ‰‡fdd„}tj|d‰dˆ_ˆ ¡‡‡‡‡‡fdd „}tj|dˆ_dˆj_ˆj ¡ˆjS) NZcontrolr rcs2x,ˆ ¡}|dkrdSt t¡t t¡qWdS)NÚstop)ÚgetÚlibcÚfflushÚ c_stdout_pÚ c_stderr_p)Úmsg)Ú flush_queuer r!Ú flush_main£s  z'Wurlitzer.__enter__..flush_main)ÚtargetTc sðd}d}xˆrÊt ˆgg|¡\}}}|r.d}n|r6Pnˆ d¡ˆj}q xx|D]p}|ˆjkrzd}t ˆj¡ˆ ˆj¡qNˆ|}t |d¡}|s¨t |¡ˆ |¡qNtˆd|ƒ}||ƒqNWˆs Pq Wˆ d¡ˆ  ¡dd „ˆDƒd S) z*Forward bytes on a pipe to stream messagesFrÚflushTiz _handle_%sr?cSsg|]}t |¡‘qSr )rr0)Ú.0r/r r r!ú Ûsz:Wurlitzer.__enter__..forwarder..N) ÚselectZputÚflush_intervalÚ _control_rrr0ÚremoveÚreadr+Újoin) ZdrainingrMÚrÚwÚxr/r2r9Zhandler)rFÚ flush_threadÚnamesrr)r r!Ú forwarder¯s:          z&Wurlitzer.__enter__..forwarder)rArBrCrDr=rr/rNÚ _control_wr"r6Úappendr$r Ú threadingZThreadZdaemonÚstartÚthreadr<)r)r/rGrWr )rFrUrVrr)r!Ú __enter__‹s0         . zWurlitzer.__enter__cCszt t¡t t¡t |jd¡|j ¡t  |j¡x4|j   ¡D]&\}}|j |}t ||ƒt  |¡qDW| ¡dS)Nó)rArBrCrDrr;rXr\rQr0r&Úitemsr%rr>)r)Úexc_typeÚ exc_valueÚ tracebackr2r3r4r r r!Ú__exit__ãs      zWurlitzer.__exit__)Ú__name__Ú __module__Ú __qualname__Ú__doc__rMÚ_default_encodingr*r6r:r(r'r=r>r]rcr r r r!rEs Xc csd}}|tkrPt ¡\}}t |d¡}|r>tj|d|d}n t |d¡}d}n|}}|tkrjd}|}nP|tkr²t ¡\}}t |d¡}|r tj|d|d}n t |d¡}d}n|}}|sÂ|rÈd} n|} t||| d} z| ||fVWdQRXWd|r| ¡|r| ¡XdS) aCapture C-level stdout/stderr in a context manager. The return value for the context manager is (stdout, stderr). Examples -------- >>> with capture() as (stdout, stderr): ... printf("C-level stdout") ... output = stdout.read() FÚwbrR)rÚrbTN)r rr) ÚPIPErr/ÚfdopenÚioÚopenr#rr0) r rrZ stdout_pipeZ stderr_pipeZstdout_rZstdout_wZstderr_rZstderr_wZcapture_encodingrSr r r!rõs>       cCsttjtj|dS)zâRedirect C-level stdout/stderr to sys.stdout/stderr This is useful of sys.sdout/stderr are already being forwarded somewhere. DO NOT USE THIS if sys.stdout and sys.stderr are not already being forwarded. )r)rr,r r)rr r r!r,scCstdkrt|ƒat ¡dS)zzRedirect all C output to sys.stdout/err This is not a context manager; it turns on C-forwarding permanently. N)Ú_mighty_wurlitzerrr])rr r r!r8scCstdk rt ddd¡dadS)z7Stop permanent redirection started by sys_pipes_foreverN)rorcr r r r!rCscCs8t|dƒst d¡dS|j dt¡|j dt¡dS)zºRegister me as an IPython extension Captures all C output during execution and forwards to sys. Does nothing on terminal IPython. Use: %load_ext wurlitzer Úkernelz;wurlitzer extension doesn't do anything in terminal IPythonNÚ pre_executeÚ post_execute)r+ÚwarningsÚwarnÚeventsÚregisterrr)Úipr r r!Úload_ipython_extensionKs rxcCs.t|dƒsdS|j dt¡|j dt¡dS)zJUnload me as an IPython extension Use: %unload_ext wurlitzer rpNrqrr)r+ruZ unregisterrr)rwr r r!Úunload_ipython_extension]s ry)r)-rgZ __future__rÚ __version__Ú__all__Ú contextlibrZctypesrr r r rmrZqueuer Ú ImportErrorrLr,rZrrsZCDLLrAZc_void_pZin_dllrCrDÚ ValueErrorr#rkr+ÚstdinrhÚlowerrÚobjectrrrrorrrxryr r r r!ÚsZ    16