qj\2UdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZmZdd lmZej6Zej:Zej>Z ejBZ"ej6Z#ej:Z$ej>Z%ejLe'Z(ddd Z)ddd Z*ddd Z+dddZ,dddZ-e.edreej^ej`zejbzeej^ejdzejbze ej^ej`ejdzejbzziZ3niZ3dddZ4e.edr4eejjeejle ejjejlziZ7niZ7dddZ8ddZ9er 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)get_event_loopwait_forEvent TimeoutError)DefaultSelector)errors)RVPQGen PQGenConnWaitFunc)WaitReady)_psycopgc t|}t5}|j|| |j|}|s|j t +|j ||dd}|j |}|j||g#1swYyxYw#t$r)}|jr|jdnd}|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 StopIterationargs) genfilenointervalsselrlistreadyexrvs P/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/psycopg/waiting.py wait_selectorr(%s I  # LL # 8 4HHZ(v&a HHUO VQ'  !wwD s5BA;BBBB C'C C Cc t|\}}|sd}t5}|j|| |j|}|s|j t +|j ||dd}|j |\}}|j||j#1swYyxYw#t$r)}|jr|jdnd}|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 # 8 4HHZ(v&a HHUO  VQ'  !wwD s5B(A>BB%!B(%B(( C1CCCcn 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,r|j,dnd} | 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_asyncrKjs B  D ! IZFZF&oo(9!&=>> HHJE88 /'&rwwy(;;;'')OO&&v.&&v.A1<'$&&v.&&v. *  R)) !wwD sF5A>E D*#DDDD*D(D*9ED D%"D*$D%%D**)EE F2E== F2 F-'F2(F5-F22F5c& 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&r|j&dnd} | 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_asyncrNsm B  D I ZFZF&oo(9!&=>> HHJE88 /&rwwy(;;;'')OO&&v.&&v.IFA1<'$&&v.&&v. !wwD 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!$ !wwD sBB B6 B1+B61B6 EpollSelectorc> t|}||dkrd}tj5}t|}|j || |j |}|s|j t*|dd}d}|tjzrt}|tjzr |tz}|j |}t|}|j||#1swYyxYw#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 Nrgr )rrepoll_epoll_evmasksrpollrrEPOLLOUTr?EPOLLINrAmodifyrr) rrr r!rYevmaskfileevsr2r$r%r&s r' wait_epollras  I  x!|H \\^u#A&F NN66 ***X.HHZ(QZ]((#E'W$EHHUO'* VV,^$ !wwD s5(C*B4CC'#C*'C** D3DDD PollSelectorcB t|}||dkrd}nt|dz}tj}t|}|j || |j|}|s|j t*|dd}d}|tjzrt}|tjzr |tz}|j |}t|}|j||#t$r)} | jr| jdnd} | 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_pollrhHs   I  x!|H8f,-H{{}q! ff%ii)G$ABEV^^O#V]]N" A"1%F KK ' !wwD sC*C,, D5DDDctjjd}|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_patchedrrnsc (A 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'rs4# AA%//0   '' ZZ '' '' 88   8 $D F:z7zD 9o&##fnn4vF##foo5G$$ >>FOO +foo =?NN*Z 9n% /M M#L* __F"**$ JJ* +E   G $WY(> 9    9U D #,,')2D2F D')5Et"LL D Y' D Dr4