*Jf1 UdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZmZdd lmZejZejZejZ ejZ!ejZ"ejZ#ej$e%Z&dd eed e'd ee(defdZ)dd eed ee(defdZ* dd eed e'd ee(defdZ+dd eed ee(defdZ,dd eed e'd ee(defdZ-e.edrPeej/ej0zej1zeej/ej2zej1ze ej/ej0ej2zej1zziZ3niZ3dd eed e'd ee(defdZ4e.edr eej5eej6e ej5ej6ziZ7niZ7dd eed e'd ee(defdZ8de9fdZ:erej;Z;eee>?dr e>e@vreAde>e@e>ZBdSerejCdkre:se;ZBdSej eDeddure-ZBdSe.edre8ZBdSe)ZBdS) z Code concerned with waiting in different contexts (blocking, async, etc). These functions are designed to consume the generators returned by the `generators` module function and to return their final value. N)Optional)get_event_loopwait_forEvent TimeoutError)DefaultSelector)errors)RVPQGen PQGenConnWaitFunc)WaitReady)_psycopggenfilenotimeoutreturnc t|}t5} |||d}|s||}||||dd}||zsJ||}r#1swxYwYdS#t $r"}|jr |jdnd}|cYd}~Sd}~wwxYw)a+ Wait for a generator using the best strategy available. :param gen: a generator performing database operations and yielding `Ready` values when it would block. :param fileno: the file descriptor to wait on. :param timeout: timeout (in seconds) to check for other interrupt, e.g. to allow Ctrl-C. :type timeout: float :return: whatever `!gen` returns on completion. Consume `!gen`, scheduling `fileno` for completion when it is reported to block. Once ready again send the ready state back to `!gen`. TNrrr )nextrregisterselect unregistersend StopIterationargs) rrrsselrlistreadyexrvs R/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/waiting.py wait_selectorr&$s# II    $# $ VQ'''8JJwJ77E 8v&&&$Qx{5y   HHUOO $ $ $ $ $ $ $ $ $ $ $ !w0D s;B A4BBB BB C *CC C c t|\}}|sd}t5} |||||}|||st jd|dd}||\}}~#1swxYwYdS#t$r"}|j r |j dnd}|cYd}~Sd}~wwxYw)a Wait for a connection generator using the best strategy available. :param gen: a generator performing database operations and yielding (fd, `Ready`) pairs when it would block. :param timeout: timeout (in seconds) to check for other interrupt, e.g. to allow Ctrl-C. If zero or None, wait indefinitely. :type timeout: float :return: whatever `!gen` returns on completion. Behave like in `wait()`, but take the fileno to wait from the generator itself, which might change during processing. NTrconnection timeout expiredrr ) rrrrreConnectionTimeoutrrr) rrrrr r!r"r#r$s r% wait_connr+Fs,II  G    ,# , VQ''' 7 33v&&&L-.JKKK$Qx{HHUOO  , , , , , , , , , , , !w0D s;$B3BB&&B**B3-B*.B33 C=CCCc Kt t}dtddf fd } t|} |tz}|t z}|s|st jd| d |r| ||t|r| ||t | d{Vn: t |d{Vn#t$rYnwxYw|r|||r||n4#|r|||r||wwxYw| }S#t&$r"}|jr |jdnd} | cYd}~Sd}~wwxYw)aW Coroutine waiting for a generator to complete. :param gen: a generator performing database operations and yielding `Ready` values when it would block. :param fileno: the file descriptor to wait on. :return: whatever `!gen` returns on completion. Behave like in `wait()`, but exposing an `asyncio` interface. staterNc:|zdSNsetr-evr"s r%wakeupzwait_async..wakeup{s  Tbad poll status: r)rrrrWAIT_RWAIT_Wr) InternalErrorclear add_readerREADY_R add_writerREADY_Wwaitrr remove_reader remove_writerrrr) rrrloopr4rreaderwriterr#r$r3r"s @@r% wait_asyncrEgs> B   De  II ZFZF ?& ?o&=!&=&=>>> HHJJJE 9888 9888 /?''))OOOOOOOO&rwwyy'::::::::::'/&&v.../&&v.../&&v.../&&v..../A1 4 !w0D sZBFE "(D E DE DE /F 1E;;F GF<6G<Gcx Kt t}dtddf fd } t|\}}|sd} |tz}|t z}|s|st jd| d |r| ||t|r| ||t t |d{V|r|||r||n4#|r|||r||wwxYw| \}}'#t$$rt jdt($r"}|jr |jdnd} | cYd}~Sd}~wwxYw) a Coroutine waiting for a connection generator to complete. :param gen: a generator performing database operations and yielding (fd, `Ready`) pairs when it would block. :param timeout: timeout (in seconds) to check for other interrupt, e.g. to allow Ctrl-C. If zero or None, wait indefinitely. :return: whatever `!gen` returns on completion. Behave like in `wait()`, but take the fileno to wait from the generator itself, which might change during processing. r-rNc4|dSr/r0r2s r%r4zwait_conn_async..wakeups r5Tr6rr()rrrrr7r8r)r9r:r;r<r=r>rr?r@rArrr*rr) rrrBr4rrrCrDr#r$r3r"s @@r%wait_conn_asyncrHs0 B   De II  G (ZFZF ?& ?o&=!&=&=>>> HHJJJE 9888 9888 /rwwyy'222222222/&&v.../&&v.../&&v.../&&v..../IFA% (( @@@!">??? !w0D s6BE0 (D"3/E0"1EE00'F9F4.F94F9cR t|}d}|f} tj|tzr|n||tzr|n|||\}}}d} |rt} |r | t z} | sN|| }c#t$r"} | jr | jdnd} | cYd} ~ Sd} ~ wwxYw)z Wait for a generator using select where supported. BUG: on Linux, can't select on FD >= 1024. On Windows it's fine. TrN) rrr7r8r<r>rrr) rrrremptyfnlistrlwlxlr"r#r$s r% wait_selectrPs  II f*/%f*/% JBB E  !  A " !w0D sA8A:: B&B!B&!B& EpollSelectorcn t|}||dkrd}nt|dz}tj5}t|}||| d}|s||}||dd}d}|tjzrt}|tj zr |tz}| |}t|}| ||#1swxYwYdS#t$r"} | jr | jdnd} | cYd} ~ Sd} ~ wwxYw)a$ Wait for a generator using epoll where supported. Parameters are like for `wait()`. If it is detected that the best selector strategy is `epoll` then this function will be used instead of `wait`. See also: https://linux.die.net/man/2/epoll_ctl BUG: if the connection FD is closed, `epoll.poll()` hangs. Same for EpollSelector. For this reason, wait_poll() is currently preferable. To reproduce the bug: export PSYCOPG_WAIT_FUNC=wait_epoll pytest tests/test_concurrency.py::test_concurrent_close Nr@@Tr )rintrepoll_epoll_evmasksrpollEPOLLOUTr<EPOLLINr>rmodifyrr) rrrrrUevmaskfileevsr3r"r#r$s r% wait_epollr] s  II ?gkkGG'F*++G \^^ -u#A&F NN66 * * * -!2#jj11G"2QZ](($#E'%W$EHHUOO'* VV,,, - - - - - - - - - - -$ !w0D s<?DB:C;;C??DC?D D4D/)D4/D4 PollSelectorcR t|}||dkrd}nt|dz}tj}t|}||| d}|s||}||dd}d}|tjzrt}|tjzr |tz}| |}t|}| ||#t$r"} | j r | j dnd} | cYd} ~ Sd} ~ wwxYw)za Wait for a generator using poll where supported. Parameters are like for `wait()`. NrrSTr )rrTrrW _poll_evmasksrPOLLOUTr<POLLINr>rrZrr) rrrrrWr[r\r3r"r#r$s r% wait_pollrcBsI  II ?gkkGG'F*++G{}}q! ff%%% (G -))G,, -ABEV^O# V]N" ! A"1%F KK ' ' ' ( !w0D sC8C:: D&D!D&!D&ctjd}|rK |drdSn2#t$r%}t d|Yd}~nd}~wwxYwdS)z Detect if some greenlet library has patched the select library. If this is the case, avoid to use the wait_c function as it doesn't behave in a collaborative way. Currently supported: gevent. z gevent.monkeyrTz+failed to detect gevent monkey-patching: %sNF)sysmodulesgetis_module_patched Exceptionloggerwarning)mr#s r%_is_select_patchedrmgs ((AN N""8,, t  N N N NNH" M M M M M M M M N 5s; A*A%%A*r?PSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got win32SelectSelectorr/)E__doc__osrerlogging selectorstypingrasynciorrrrrr r)abcr r r r_enumsrr_cmodulerRr7Wr8RWWAIT_RWr<r>READY_RW getLogger__name__rjrTfloatr&r+rErHrPhasattr EPOLLONESHOTrYEPOLLERRrXrVr]rbrar`rcboolrmwait_c__annotations__environfname startswithglobals ImportErrorr?platformgetattrrJr5r%rs AAAAAAAAAAAA%%%%%%////////////00000000   ' ' ' 8  8 $ $uRy#SUD9R=8E?bD=A77 r77*25/77777t66y}6x6RT6666xU2YhuoQSB 79o&&#fn4vF#fo5G$ >FO +fo =?NNN,,E"I,s,Xe_,PR,,,,^ 79n%% /MM M""59"c"HUO"r""""JD*  _F"*$$ J* +E   G $ $ WWYY(>(>k       799U DDD  #,'))2D2D2F2F) DDD'')5Et"L"LLL DDD WY'' DDD DDDr5