pj\2UdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZmZdd lmZejZejZejZ ej!Z"ejZ#ejZ$ejZ%ej&e'Z(d&d'dZ)d&d(dZ*d&d'dZ+d&d(dZ,d&d'dZ-e.edrPeej/ej0zej1zeej/ej2zej1ze ej/ej0ej2zej1zziZ3niZ3d&d'dZ4e.edr eej5eej6e ej5ej6ziZ7niZ7d&d'dZ8d)dZ9erej:Z:de;d <d!ej<vrNej<d!Z=e=>d"r e=e?vre@d#e=e?e=ZAdSerejBd$kre9se:ZAdSej eCed%dure-ZAdSe.edre8ZAdSe)ZAdS)*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)_psycopggen PQGen[RV]filenointinterval float | Nonereturnr c t|}t5}||| ||}|s|t 4|||dd}||}|||#1swxYwYdS#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`. Ttimeoutrr N nextrregisterselectsend READY_NONE unregister StopIterationargs) rrrsselrlistreadyexrvs _/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/psycopg/waiting.py wait_selectorr,%s- II    (# LL # # # ( 8 44HHZ(((v&&&a HHUOO VQ''' ( ( ( ( ( ( ( ( ( ( ( !w0D s;CBB99B==CB=C C2C-'C2-C2 PQGenConn[RV]c t|\}}|sd}t5}||| ||}|s|t 4|||dd}||\}}|||#1swxYwYdS#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. NTrrr r) rrrr%r&r'r(r)r*s r+ wait_connr/GsBII  H    (# LL # # # ( 8 44HHZ(((v&&&a HHUOO  VQ''' ( ( ( ( ( ( ( ( ( ( ( !w0D s;$CBCCC C C C<C71C<7C<c Kt t}d fd } t|} |tz}|tz}|s|st jd| d |r|||t|r| ||t |: t |d {Vn*#t$rYnwxYw d {V|r|||r||n4#|r|||r||wwxYw| }R#t$$r&}t jt)|d }~wt*$r"}|jr |jdnd } | cYd }~Sd }~wwxYw) 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. staterrNonec:|zdSNsetr1evr(s r+wakeupzwait_async..wakeup~s  Tbad poll status: rNr1rrr2)rrrWAIT_RWAIT_We InternalErrorclear add_readerREADY_R add_writerREADY_Wrwaitr remove_reader remove_writerr OSErrorOperationalErrorstrr#r$) rrrloopr9r%readerwriterr)r*r8r(s @@r+ wait_asyncrOjs\ B   D ! II ZFZF ?& ?o&=!&=&=>>> HHJJJE 9888 9888 /'&rwwyy(;;;;;;;;;;'''))OOOOOOO/&&v.../&&v.../&&v.../&&v..../A1 4 *** R))) !w0D sfBF ;E>(C'&E' C41E3C44E/F 1E11F G%!F66 G%G G% G%c Kt t}d fd } t|\}} |tz}|tz}|s|st jd| d |r|||t|r| ||t |r: t |d {Vn*#t$rYnwxYw d {V|r|||r||n4#|r|||r||wwxYw| \}}U#t$$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 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. r1rrr2c4|dSr4r5r7s r+r9zwait_conn_async..wakeups r:Tr;rNr<)rrrr=r>r?r@rArBrCrDrErrFrrGrHr r#r$) rrrLr9rr%rMrNr)r*r8r(s @@r+wait_conn_asyncrRs9 B   D II  (ZFZF ?& ?o&=!&=&=>>> HHJJJE 9888 9888 /$&rwwyy(;;;;;;;;;;'''))OOOOOOO/&&v.../&&v.../&&v.../&&v..../IFA1 (4 !w0D sZBF>E(C*)E* C74E6C77E/F1E44F F=F82F=8F=c t|}d}|f} tj|tzr|n||tzr|n|||\}}}d} |rt} |r | t z} | s|th|| }}#t$r"} | j r | j dnd} | 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) rrr=r>rCrEr r!r#r$) rrrr%emptyfnlistrlwlxlr(r)r*s r+ wait_selectrZs  II f*/%f*/% JBB E  !  $$$A! $ !w0D sBB CB;5C;C EpollSelectorcv t|}||dkrd}tj5}t|}||| ||}|s|t3|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 NrgTr )rrepoll_epoll_evmasksrpollr r!EPOLLOUTrCEPOLLINrEmodifyr#r$) rrrr%r]evmaskfileevsr8r(r)r*s r+ wait_epollresw  II  x!||H \^^ -u#A&F NN66 * * * -**X..HHZ(((QZ](($#E'%W$EHHUOO'* VV,,, - - - - - - - - - - -$ !w0D s;,D CC??DD DD D8D3-D83D8 PollSelectorc t|}||dkrd}nt|dz}tj}t|}||| ||}|s|t3|dd}d}|tjzrt}|tj zr |tz}||}t|}| ||#t$r"} | jr | jdnd} | cYd} ~ Sd} ~ wwxYw)za Wait for a generator using poll where supported. Parameters are like for `wait()`. Nrg@@Tr )rrrr_ _poll_evmasksrr r!POLLOUTrCPOLLINrErbr#r$) rrrr%r_rcrdr8r(r)r*s r+ wait_pollrkHsP  II  x!||HH8f,--H{}}q! ff%%% (ii))G $$$ABEV^O# V]N" ! A"1%F KK ' ' ' ( !w0D sDD D=D82D=8D=boolctjd}|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_patchedrvns ((AN N""8,, t  N N N NNH" M M M M M M M M N 5s; A*A%%A*rrFPSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got win32SelectSelectorr4)rrrrrrrr )rr-rrrr )rrl)D__doc__ __future__rosrnrlogging selectorsasynciorrrrrr r?abcr r r r_enumsrr_cmodulerRr=Wr>RWWAIT_RWNONEr!rCrEREADY_RW getLogger__name__rsr,r/rOrRrZhasattr EPOLLONESHOTraEPOLLERRr`r^rerjrirhrkrvwait_c__annotations__environfname startswithglobals ImportErrorrFplatformgetattrrTr:r+rs#""""" AAAAAAAAAAAA%%%%%%////////////00000000   ' Z ' ' 8  8 $ $D     F:::::z77777zD 79o&&#fn4vF#fo5G$ >FO +fo =?NNN*****Z 79n%% /MM M#####L*  _F"*$$ J* +E   G $ $ WWYY(>(>k       799U DDD  #,'))2D2D2F2F) DDD'')5Et"L"LLL DDD WY'' DDD DDDr: