:dc:6dZddlmZdZgdZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlmZmZmZddlmZdd lmZ dd lmZmZej6dZd ZdxZZ ej@jCedZej@jCedZdZ#dZ$e%e jLddxsdZ'e'jQdk(rdZ'd"dZ)edZ*GddZ+ee$e$e'dfdZ,e'dfdZ-da.e j^Z0e'dfdZ1dZ2da3d Z4d!Z5y#e$rd Zd ZYwxYw#e"$rR ej@jCedZej@jCedZn#e"$r e\ZZYnwxYwYwxYw)#zhCapture C-level FD output on pipes Use `wurlitzer.pipes` or `wurlitzer.sys_pipes` as context managers. )print_functionz3.0.3)pipes sys_pipessys_pipes_foreverstop_sys_pipesPIPESTDOUT WurlitzerN)contextmanager)F_GETFLF_SETFLfcntl) lru_cache)Queue) F_GETPIPE_SZ F_SETPIPE_SZiic ddl} |j}|jd|jd|j dj gd}t j|j}t j|j}||fS#t$r tdwxYw#t$r.}tjdj|Yd}~y d}~wwxYw) zeUse CFFI to lookup stdout/stderr pointers Should work ~everywhere, but requires compilation rNz@Failed to lookup stdout symbols in libc. Fallback requires cffi.zconst size_t c_stdout_p();zconst size_t c_stderr_p(); )z#include z=const size_t c_stdout_p() { return (size_t) (void*) stdout; }z=const size_t c_stderr_p() { return (size_t) (void*) stderr; }zBFailed to lookup stdout with cffi: {}. Streams may not be flushed.)NN)cffi ImportErrorFFIcdefverifyjoinctypesc_void_p c_stdout_p c_stderr_p Exceptionwarningswarnformat)r_ffi_librres +/usr/lib/python3/dist-packages/wurlitzer.py_get_streams_cffir',s   &xxz ./ ./{{ II  __T__%67 __T__%67 :%%9   N   (  Q X X   s#B%BB=%B:= C4$C//C4stdoutstderr __stdoutp __stderrpencodingutf8asciicd}ttd|zD]} tj||cS|r|y#t$r@}|}|j t j k(rtjdnYd}~fd}~wwxYw)z Like os.dup2, but retry on EBUSYN g?) rangeintosdup2OSErrorerrnoEBUSYtimesleep)abtimeoutdup_errir%s r&r6r6hsG 3rG|$ % 771a=    Gww%++% 3  s< B6BBctjdk7ry tdd5}t|j }ddddkDry|dkry|S#1swYxYw#t $rYywxYw)aGet max pipe size Reads /proc/sys/fs/pipe-max-size on Linux. Always returns None elsewhere. Returns integer (up to 1MB), or None if no value can be determined. Adapted from wal-e, (c) 2018, WAL-E Contributors used under BSD-3-clause LinuxNz/proc/sys/fs/pipe-max-sizerii)platformsystemopenr4readr)f pipe_max_sizes r&_get_max_pipe_sizerJys}G#! . 4 *MM * ; & e #  * *    s( A"AA"AA"" A.-A.cdeZdZdZdZddeefdZdZdZ dZ dZ d Z d Z d Zd Zd Zy)r zcClass for Capturing Process-level FD output via dup2 Typically used via `wurlitzer.pipes` g?Nc||_|tk(r|j|_n||_||_| t }||_i|_i|_i|_|j|jd<|j|jd<y)a 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. bufsize: int or None Set pipe buffer size using fcntl F_SETPIPE_SZ (linux only) default: use /proc/sys/fs/pipe-max-size up to a max of 1MB if 0, will do nothing. Nr)r() _stdoutr _stderrr.rJ_bufsize _save_fds _real_fds _handlers_handle_stderr_handle_stdout)selfr(r)r.bufsizes r&__init__zWurlitzer.__init__s}* V <|j|jf|_y)zSetup handle for output, if anyN)rMrNhandlerUs r& _setup_handlezWurlitzer._setup_handles||T\\2 rXcy)zCFinish handle, if anything should be done when it's all wrapped up.Nrxs r&_finish_handlezWurlitzer._finish_handles rXcj|jr.tjrtjj|jr.tj rtj jt tjt ttjtyy)z&flush sys.stdout/err and low-level FDsN) rMr[r(flushrNr)rlibcfflushrrxs r&_flushzWurlitzer._flushse <.flush_mains*!oo'&= rX)targetTcd}d}tj}D]"}|j|tj$r(|j |}|rd}n"|rn j dj };|D]\}}|j}|jk(rXd}jj|jjtjjy|}tj|d} | s8j||j|tj|td|z} | | snr( j d jD cgc]} tj| c} |jycc} w) z*Forward bytes on a pipe to stream messagesFrr~Tiz _handle_%srN) selectorsDefaultSelectorregister EVENT_READselectputflush_intervalfd _control_rremove unregisterr5rarGrZr)drainingrpollerpipe_events selector_keyrirrcrohandlerr^r flush_threadnamesrrUs r& forwarderz&Wurlitzer.__enter__..forwarder(sHN..0F =y';';< =~6&'N  $0)-)<)< +1&'L%%BT__,#' T__5))$//:1  9D772t,D R())"- ")$ t0C"D #&$OR OOF #    (- .RXXd^ . LLN /sG)rryr5r^r _control_wrMrjappendrNr threadingThreaddaemonstartthreadrw)rUr^rrrrrrs` @@@@r& __enter__zWurlitzer.__enter__s  +-779(!), <<##H-D LL "E$K <<##H-D LL "E$Kg  !''z: " 6 6 p &&i8 !  {{rXc|jtj|jd|jj tj |j|jjD]5\}}|j|}t||tj |7|jy)N) rr5rsrrrrarQitemsrPr6r|)rUexc_type exc_value tracebackrcrdres r&__exit__zWurlitzer.__exit__fs  %(  !"^^113 MD'nnT*G ' " HHW   rX)__name__ __module__ __qualname____doc__r_default_encodingrJrWrjrprTrSryr|rrrr{rXr&r r sT N""$ "7H0333  $^@rXr c#Kdx}}|rtj}ntj}|tk(r |x}}d}n|x}}|tk(rd} |} n|tk(r |x} } d}n|x} } t || ||} | 5|| fddd|r|j d|r| j dyy#1swY1xYw#|r|j d|r| j dwwxYww)ahCapture C-level stdout/stderr in a context manager. The return value for the context manager is (stdout, stderr). .. versionchanged:: 3.0 when using `PIPE` (default), the type of captured output is `io.StringIO/BytesIO` instead of an OS pipe. This eliminates max buffer size issues (and hang when output exceeds 65536 bytes), but also means the buffer cannot be read with `.read()` methods until after the context exits. Examples -------- >>> with capture() as (stdout, stderr): ... printf("C-level stdout") ... output = stdout.read() FTN)r(r)r.rVr)ioStringIOBytesIOrr r seek) r(r)r.rV stdout_pipe stderr_pipePipeIOstdout_rstdout_wstderr_rstderr_wws r&rrxs*!&%K+ ~$h&8 $$8  4$h&8 $$8(XwWA   %H$ $ %  MM!   MM!   % %  MM!   MM!  s6A7C+:B?<B3B? (C+3B<8B??)C((C+cXttjtj||S)zRedirect 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.rV)rr[r(r)rs r&rrs SZZ(G LLrXct5t t||atjdddy#1swYyxYw)zvRedirect all C output to sys.stdout/err This is not a context manager; it turns on C-forwarding permanently. N) _mighty_lock_mighty_wurlitzerrrrs r&rrs; *  $ )(G <   ' ' )***s '7Acxt5ttjddddadddy#1swYyxYw)z7Stop permanent redirection started by sys_pipes_foreverN)rrrr{rXr&rrs: %  (  & &tT4 8 $ %%%s 09Fc*t|ddstjdydD])}tt|tjdy|jj dt |jj dtday) zRegister me as an IPython extension Captures all C output during execution and forwards to sys. Does nothing on terminal IPython. Use: %load_ext wurlitzer kernelNz;wurlitzer extension doesn't do anything in terminal IPython) __stdout__ __stderr__z:sys.{} is None. Wurlitzer can't capture output without it. pre_execute post_executeT) rZr r!r[rrrr_extension_enabled)iprcs r&load_ipython_extensionrs} 2x & ST, 3  % MMV W  II}&78II~~6rXctsy|jjdt|jjdtt day)zFUnload me as an IPython extension Use: %unload_ext wurlitzer NrrF)rrrrr)rs r&unload_ipython_extensionrsA II(9:II8rX)r-)6r __future__r __version____all__rr8rr5rDrr[rr:r contextlibr rr r functoolsrqueuerrrrCDLLrr'rrrin_dll ValueErrorr rrZstdinrlowerr6rJr rrrLockrrrrrrr{rXr&rs&     %))0 v{{4#&L Z 5''h7J''h7J CIIz48BF'" ! !DYYxd->33l)$My~~  1$ *%0SLLd55__++D+> __++D+> 5!2!4 J5 5sHD268E2 D>=D>F8F?FFFFFF