yjGUdZddlmZddlZddlZddlZddlZddlmZm Z dgZ ej Z iZ ded<Gdd ZddZejddZdS)u!Thread-scoped stdout/stderr silencing for background worker threads. ``contextlib.redirect_stdout``/``redirect_stderr`` reassign the *process-global* ``sys.stdout``/``sys.stderr``. When a daemon worker thread (e.g. the background memory/skill review) wraps its whole body in those context managers, every other thread in the process — including a gateway's asyncio event-loop thread driving a Telegram long-poll — sees ``sys.stdout``/``sys.stderr`` pointing at ``devnull`` for the full duration. Any bare ``print`` / ``sys.stderr.write`` from those other threads is silently lost during that window (see issue #55769 / #55925). This module installs a thin proxy as ``sys.stdout``/``sys.stderr`` that routes writes per-thread: threads registered as "silenced" go to a sink; every other thread passes through to the *original* stream. The proxy is installed once, idempotently, and is never uninstalled (uninstalling would race other threads mid-write), so the only observable effect for unregistered threads is one extra attribute lookup per write. ) annotationsN)IteratorTextIOthread_scoped_silencez!dict[str, '_ThreadRoutingStream'] _installedcXeZdZdZddZddZdd Zdd Zd ZdZ dZ ddZ dZ dZ dS)_ThreadRoutingStreamaA ``sys.stdout``/``sys.stderr`` stand-in that routes writes per-thread. Threads whose ident is in ``_silenced`` write to ``_sink``; all other threads write to ``_passthrough`` (the original stream captured at install time). Attribute access for anything other than the methods we override is delegated to the *current* target so things like ``.encoding`` / ``.fileno()`` behave like the underlying stream for the calling thread. passthroughrsinkreturnNonec`||_||_i|_tj|_dSN) _passthrough_sink _silenced threadingLock_lock)selfr r s @/home/ragecks/.hermes/hermes-agent/agent/thread_scoped_output.py__init__z_ThreadRoutingStream.__init__-s-' *,^%% c~|jtjddkr|jS|jSNr)rgetr get_identrrrs r_targetz_ThreadRoutingStream._target6s9 >  i133Q 7 7! ; ;:   ridentintc|j5|j|ddz|j|<ddddS#1swxYwYdSNr)rrr)rr s rsilencez_ThreadRoutingStream.silence<s Z E E$(N$6$6ua$@$@1$DDN5 ! E E E E E E E E E E E E E E E E E Es'<AAc|j5|j|ddz }|dkr ||j|<n|j|dddddS#1swxYwYdSr#)rrrpop)rr depths r unsilencez_ThreadRoutingStream.unsilence@s Z 0 0N&&ua0014Eqyy(-u%%""5$///  0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0sA A  A$'A$c ||S#t$r)t|trt |ndcYSwxYwr)rwrite Exception isinstancestrlen)rdatas rr+z_ThreadRoutingStream.writeIsc =<<>>''-- - = = = *4 5 5<3t9991 < < < =s&)0AAcr |S#t$rYdSwxYwr)rflushr,rs rr2z_ThreadRoutingStream.flushOsC <<>>'')) )   44 s %( 66cx|} ||S#t$rYdSwxYwr)r writelinesr,)rlinestargets rr4z_ThreadRoutingStream.writelinesUsJ $$U++ +   44 s + 99boolc t|S#t$rYdSwxYw)NF)r7risattyr,rs rr9z_ThreadRoutingStream.isatty\sK  --//00 0   55 s25 AAcN|Sr)rfilenors rr;z_ThreadRoutingStream.filenobs||~~$$&&&rcFt||Sr)getattrr)rnames r __getattr__z _ThreadRoutingStream.__getattr__est||~~t,,,rN)r rr rr r )r r)r r!r r )r r7)__name__ __module__ __qualname____doc__rrr%r)r+r2r4r9r;r?rrr r #s&&&&!!!! EEEE0000===   '''-----rr attrr.r rr '_ThreadRoutingStream'ctt5t|}tt|d}|||ur|cdddS||n|}t t jdd}t||}tt|||t|<|cdddS#1swxYwYdS)zGInstall (idempotently) a routing proxy as ``sys.`` and return it.Nwzutf-8)encoding) _install_lockrrr=sysopenosdevnullr setattr)rEr proxycurrentr s r_ensure_installedrRks   t$$#tT**  E!1!1        ")!4gg+ BJg666$[$77T5!!! 4                  s8B- AB--B14B1Iterator[None]c#Ktj}tdtjp tj}tdtjp tj}|||| dV| || |dS#| || |wxYw)a-Silence ``stdout``/``stderr`` for the *current thread only*. Other threads keep writing to the real streams. Use this around a worker thread's body instead of ``contextlib.redirect_stdout(devnull)`` when the process is multi-threaded and another thread must keep its console output. stdoutstderrN) rrrRrK __stdout__rU __stderr__rVr%r))r out_proxy err_proxys rrr}s   ! !E!(CN,HcjIII!(CN,HcjIII e e# E"""E""""" E"""E""""s B==,C))rEr.r rr rF)r rS)rC __future__r contextlibrMrKrtypingrr__all__rrJr__annotations__r rRcontextmanagerrrDrrras$#""""" ######## " #   13 2222E-E-E-E-E-E-E-E-P$ ######r