Mh%dZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlmZmZddlmZd d gZe dddZGddZGdd ZdS)a patch_stdout ============ This implements a context manager that ensures that print statements within it won't destroy the user interface. The context manager will replace `sys.stdout` by something that draws the output above the current prompt, rather than overwriting the UI. Usage:: with patch_stdout(application): ... application.run() ... Multiple applications can run in the body of the context manager, one after the other. ) annotationsN)contextmanager) GeneratorTextIOcast)get_app_sessionrun_in_terminal)Output patch_stdout StdoutProxyFrawboolreturnGenerator[None, None, None]c#Kt|5}tj}tj}t t |t_t t |t_ dV|t_|t_n#|t_|t_wxYw ddddS#1swxYwYdS)a Replace `sys.stdout` by an :class:`_StdoutProxy` instance. Writing to this proxy will make sure that the text appears above the prompt, and that it doesn't destroy the output from the renderer. If no application is curring, the behavior should be identical to writing to `sys.stdout` directly. Warning: If a new event loop is installed using `asyncio.set_event_loop()`, then make sure that the context manager is applied after the event loop is changed. Printing to stdout will be scheduled in the event loop that's active when the context manager is created. :param raw: (`bool`) When True, vt100 terminal escape sequences are not removed/escaped. )rN)r sysstdoutstderrrr)rproxyoriginal_stdoutoriginal_stderrs [/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/prompt_toolkit/patch_stdout.pyr r (s$     )**&%(( &%((  ) EEE(CJ(CJJ)CJ(CJ ( ( ( (J ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s*AB3+B/B3B""B33B7:B7ceZdZdZdS)_Donez-Sentinel value for stopping the stdout proxy.N)__name__ __module__ __qualname____doc__rrrIs3333r!rceZdZdZ d&d'd Zd(d Zd)dZd*dZd+dZd*dZ d,dZ d-dZ d.dZ d*dZ d/dZd*dZed0d Zd1d!Zd2d"Zed3d#Zed3d$Zd%S)4r a File-like object, which prints everything written to it, output above the current application/prompt. This class is compatible with other file objects and can be used as a drop-in replacement for `sys.stdout` or can for instance be passed to `logging.StreamHandler`. The current application, above which we print, is determined by looking what application currently runs in the `AppSession` that is active during the creation of this instance. This class can be used as a context manager. In order to avoid having to repaint the prompt continuously for every little write, a short delay of `sleep_between_writes` seconds will be added between writes in order to bundle many smaller writes in a short timespan. 皙?Fsleep_between_writesfloatrrrNonec||_||_tj|_g|_t |_|jj|_ tj |_ | |_d|_dSNF)r$r threadingRLock_lock_bufferr app_sessionoutput_outputqueueQueue _flush_queue_start_write_thread _flush_threadclosed)selfr$rs r__init__zStdoutProxy.__init___sw %9!_&& "$ +,, $/6 7.write_and_flushsl L ( ( * * *x ) &&t,,,, ""4((( L   r!c*tddS)NF) in_executor)r )rhsrwrite_and_flush_in_loopz=StdoutProxy._write_and_flush..write_and_flush_in_loops O ? ? ? ? ? ?r!Nrr&)call_soon_threadsafe)r6r_rZrkrhs` ` @rrUzStdoutProxy._write_and_flushs ! ! ! ! ! ! ! @ @ @ @ @ @ < O       % %&= > > > > >r!datacd|vr^|dd\}}|j|dgz}|g|_d|}|j|dS|j|dS)a Note: print()-statements cause to multiple write calls. (write('line') and write(' ')). Of course we don't want to call `run_in_terminal` for every individual call, because that's too expensive, and as long as the newline hasn't been written, the text itself is again overwritten by the rendering of the input command line. Therefor, we have a little buffer which holds the text until a newline is written to stdout.  rrNN)rsplitr,rCr2rBrQ)r6rnbeforeafterto_writerZs r_writezStdoutProxy._writes 4<<!KKa00MFE|vtn4H!7DL778$$D   ! !$ ' ' ' ' ' L   % % % % %r!c|d|j}g|_|j|dS)NrN)rCr,r2rBrgs r_flushzStdoutProxy._flushs9wwt|$$  d#####r!intc|j5||dddn #1swxYwYt|Sr9)r+rulen)r6rns rrezStdoutProxy.writes Z   KK                  4yys *..cn|j5|ddddS#1swxYwYdS)z( Flush buffered output. N)r+rwr:s rrfzStdoutProxy.flush s}Z   KKMMM                  s *.. TextIO | Nonec2|jjp tjSr9)r/rr __stdout__r:s rrzStdoutProxy.original_stdouts|"4cn4r!c4|jSr9)r/filenor:s rrzStdoutProxy.filenos|""$$$r!cJ|jj}|dS|Sr()r/risatty)r6rs rrzStdoutProxy.isattys$$ >5}}r!c4|jSr9)r/encodingr:s rrzStdoutProxy.encoding#s|$$&&&r!cdS)Nstrictr r:s rerrorszStdoutProxy.errors'sxr!N)r#F)r$r%rrrr&)rr )r<r=rr&rl)rrD)rr\)r_r\rZr`rr&)rnr`rr&)rnr`rrx)rr|)rrx)rr)rr`)rrrrr7r;r@r?r3rJrTrUrurwrerfpropertyrrrrrr r!rr r Ms&'*6    #6#6#6#6J    "?"?"?"?H&&&&.$$$$  555X5 %%%%'''X'Xr!)F)rrrr)r __future__rasyncior0rr)rV contextlibrtypingrrr applicationr r r.r __all__r rr r r!rrsF(#"""""  %%%%%%**********99999999  )))))@44444444\\\\\\\\\\r!