ACe8: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.2)pipes sys_pipessys_pipes_foreverstop_sys_pipes 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 )lib/python3.12/site-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$r4r4fsG 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)platformsystemopenr2readr)f pipe_max_sizes r$_get_max_pipe_sizerHws}G#! . 4 *MM * ; & e #  * *    s( A"AA"AA"" A.-A.cdeZdZdZdZddeefdZdZdZ dZ dZ d Z d Z d Zd Zd Zy)rzcClass 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&) _stdoutSTDOUT_stderrr,rH_bufsize _save_fds _real_fds _handlers_handle_stderr_handle_stdout)selfr&r'r,bufsizes r$__init__zWurlitzer.__init__s}* V <O Ps*C))#DDcV|jr|j|jd}|S)zTDecode data, if any Called before passing to stdout/stderr streams replace)r,decoderTdatas r$_decodezWurlitzer._decodes$ ==;;t}}i8D rWcr|jr+|jj|j|yyN)rKwritermrks r$rSzWurlitzer._handle_stdout) << LL  t||D1 2 rWcr|jr+|jj|j|yyro)rMrprmrks r$rRzWurlitzer._handle_stderrrqrWc>|j|jf|_y)zSetup handle for output, if anyN)rKrMhandlerTs r$ _setup_handlezWurlitzer._setup_handles||T\\2 rWcy)zCFinish handle, if anything should be done when it's all wrapped up.Nrus r$_finish_handlezWurlitzer._finish_handles rWcj|jr.tjrtjj|jr.tj rtj jt tjt ttjtyy)z&flush sys.stdout/err and low-level FDsN) rKrZr&flushrMr'rlibcfflushrrus r$_flushzWurlitzer._flushse <.flush_mains*!oo'&= rW)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 unregisterr3r_rErYr)drainingrpollerpipe_events selector_keyrfrrarlhandlerr]r flush_threadnamesrrTs 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)r~rvr3r]r _control_wrKrgappendrMr threadingThreaddaemonstartthreadrt)rTr]rrrrrrs` @@@@r$ __enter__zWurlitzer.__enter__s  +-779(!), <<##H-D LL "E$K <<##H-D LL "E$Kg  !''z: " 6 6 p &&i8 !  {{rWc|jtj|jd|jj tj |j|jjD]5\}}|j|}t||tj |7|jy)N) r~r3rprrrr_rPitemsrOr4ry)rTexc_type exc_value tracebackrarbrcs r$__exit__zWurlitzer.__exit__bs  %(  !"^^113 MD'nnT*G ' " HHW   rW)__name__ __module__ __qualname____doc__r_default_encodingrHrVrgrmrSrRrvryr~rrrxrWr$rrsT N""$ "7H,333  $^@rWrc#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,rUr)ioStringIOBytesIOPIPErLrseek) r&r'r,rU stdout_pipe stderr_pipePipeIOstdout_rstdout_wstderr_rstderr_wws r$rrts*!&%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,rU)rrZr&r'rs r$rrs SZZ(G LLrWct5t 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)rrrrxrWr$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) rYrrrZrrrr_extension_enabled)ipras r$load_ipython_extensionrs} 2x & ST, 3  % MMV W  II}&78II~~6rWctsy|jjdt|jjdtt day)zFUnload me as an IPython extension Use: %unload_ext wurlitzer NrrF)rrrrr)rs r$unload_ipython_extensionrsA II(9:II8rW)r+)6r __future__r __version____all__rr6rr3rBrrZrr8r contextlibr r r r functoolsr queuerrrrCDLLr|r%rrrin_dll ValueErrorrLrrYstdinrlowerr4rHrrrrLockrrrrrrrxrWr$rs&     %))0 v{{4#&L Z 5''h7J''h7J CIIz48BF'" ! !DWWtd->33l)$My~~  1$ *%0OLLd55__++D+> __++D+> 5!2!4 J5 5sHD268E2 D>=D>F8F?FFFFFF