iw1UdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZdd lmZdd lmZej6Zej:Zej>Z ejBZ"ej6Z#ej:Z$ej>Z%ejLe'Z(ddd Z)ddd Z*dddZ+dddZ,dddZ-e.edreej^ej`zejbzeej^ejdzejbze ej^ej`ejdzejbzziZ3niZ3dddZ4e.edr4eejjeejle ejjejlziZ7niZ7dddZ8d dZ9er ejtZ:de;d<dejxvr?ejxdZ=e=j}dr e=e?vr e@de=e?e=ZAyerejdk7r e9se:ZAyejeCeddure-ZAye.edre8ZAye)ZAy)!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. ) annotationsN)Event TimeoutErrorget_event_loopwait_for)DefaultSelector)errors)RVPQGen PQGenConnWaitFunc)Ready)Wait)_psycopgc t|}t5}|j|| |j|x}s|j t +|j ||dd}|j |}|j||g#1swYyxYw#t$r}|j}|cYd}~Sd}~wwxYw)a8 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 interval: interval (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. Consume `!gen`, scheduling `fileno` for completion when it is reported to block. Once ready again send the ready state back to `!gen`. timeoutrr N nextrregisterselectsend READY_NONE unregister StopIterationvalue) genfilenointervalsselrlistreadyexrvs o/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/venv/lib/python3.12/site-packages/psycopg/waiting.py wait_selectorr(&s I   (# LL #!$H!===HHZ(v&a HHUO VQ' ( (  s5BA;BBBB B?' B:4B?:B?c t|\}}|sd}t5}|j|| |j|x}s|j t +|j ||dd}|j |\}}|j||j#1swYyxYw#t$r}|j}|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 interval: interval (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. Nrrr r) rr rr!r"r#r$r%r&s r' wait_connr*GsI H   (# LL #!$H!===HHZ(v&a HHUO  VQ' ( (  s5B(A>BB%!B(%B(( C 1 C>C C cL Kt t}d fd } t|} |tz}|tz}|s|st j d| jd |r|j||t|r|j||t |$ t j|d{n jd{|r|j||r|j!| |j# }7V#t$rYFwxYw7N#|r|j||r|j!|wwxYw#t$$r#}t j&t)|d}~wt*$r}|j,} | cYd}~Sd}~wwxYww)a 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. :param interval: interval (in seconds) to check for other interrupt, e.g. to allow Ctrl-C. If None, wait indefinitely. :return: whatever `!gen` returns on completion. Behave like in `wait()`, but exposing an `asyncio` interface. c0|zjyNsetstateevr$s r'wakeupzwait_async..wakeup}s  bad poll status: rNr1rreturnNone)rrrWAIT_RWAIT_We InternalErrorclear add_readerREADY_R add_writerREADY_Wrwaitr remove_reader remove_writerrOSErrorOperationalErrorstrrr) rrr loopr3r!readerwriterr%r&r2r$s @@r' wait_asyncrKisz B  D ! IZFZFfoo(9!&=>> HHJE88 /'&rwwy(;;;'')OO&&v.&&v.A1<'$&&v.&&v. *  R))  sF$A>E D*#DDDD*D(D*9ED D%"D*$D%%D**)EE F!E== F! FF!F$F!!F$c Kt t}d fd } t|\}} |tz}|tz}|s|st j d| jd |r|j||t|r|j||t |r$ t j|d{n jd{|r|j||r|j!| |j# \}}7Y#t$rYIwxYw7Q#|r|j||r|j!|wwxYw#t$$r}|j&} | cYd}~Sd}~wwxYww)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 interval: interval (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. c*|jyr-r.r0s r'r3zwait_conn_async..wakeups r4r5rNr6)rrrr9r:r;r<r=r>r?r@rArrBrrCrDrrr) rr rHr3rr!rIrJr%r&r2r$s @@r'wait_conn_asyncrNsa B  D I ZFZFfoo(9!&=>> HHJE88 /&rwwy(;;;'')OO&&v.&&v.IFA1<'$&&v.&&v.  sFBE#D0&DDDD0D.D0!= 1024. On Windows it's fine. rN) rrr9r:r?rArrrr) rrr r!emptyfnlistrlwlxlr$r%r&s r' wait_selectrVs  If*%f*% JBB E $A!$  sBB B% B B% B% EpollSelectorc t|}||dkrd}tj5}t|}|j || |j |x}s|j t*|dd}d}|tjzrt}|tjzr |tz}|j |}t|}|j||#1swYyxYw#t$r} | j} | 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 Nrgr )rrepoll_epoll_evmasksrpollrrEPOLLOUTr?EPOLLINrAmodifyrr) rrr r!rYevmaskfileevsr2r$r%r&s r' wait_epollras  I  x!|H \\^ -u#A&F NN66 *#(::h#777HHZ(QZ]((#E'W$EHHUO'* VV, - -"  s5(C*B4CC'#C*'C** D 3 DD D  PollSelectorc  t|}||dkrd}nt|dz}tj}t|}|j || |j|x}s|j t*|dd}d}|tjzrt}|tjzr |tz}|j |}t|}|j||#t$r} | j} | cYd} ~ Sd} ~ wwxYw)za Wait for a generator using poll where supported. Parameters are like for `wait()`. Nrg@@r )rintrr[ _poll_evmasksrrrPOLLOUTr?POLLINrAr^rr) rrr r!r[r_r`r2r$r%r&s r' wait_pollrhFs  I  x!|H8f,-H{{}q! ff%#yy22G2$ABEV^^O#V]]N" A"1%F KK '  sC*C,, D 5 DD D ctjjdx}r |jdry yy#t$r }t j d|Yd}~yd}~wwxYw)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_patchedrrksf KKOOO ,,q, N""8,-  N NNH" M M  Ns8 A!AA!rrBPSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got win32SelectSelectorr-)rz PQGen[RV]rrdr float | Noner7r )rz PQGenConn[RV]r rwr7r )r7bool)D__doc__ __future__rosrjrlogging selectorsasynciorrrrrr r;abcr r r r_enumsrr_cmodulerRr9Wr:RWWAIT_RWNONErr?rAREADY_RW getLogger__name__ror(r*rKrNrVhasattr EPOLLONESHOTr]EPOLLERRr\rZrargrfrerhrrwait_c__annotations__environfname startswithglobals ImportErrorrBplatformgetattrrPr4r'rs7# AA%//"    '' ZZ '' '' 88   8 $BD:z7zD 9o&##fnn4vF##foo5G$$ >>FOO +foo =?NN)X 9n% /M M"J( __F"**$ JJ* +E   G $WY(> 9    9U D #,,')2D2F D')5Et"LL D Y' D Dr4