o 6 f1 @sUdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZmZdd lmZejZejZejZ ejZ!ejZ"ejZ#e$e%Z&d'd eed e'd ee(defddZ)d'd eed ee(defddZ* d'd eed e'd ee(defddZ+d'd eed ee(defddZ,d'd eed e'd ee(defddZ-e.edreej/ej0Bej1Beej/ej2Bej1Be ej/ej0ej2Bej1BBiZ3niZ3d'd eed e'd ee(defddZ4e.edr eej5eej6e ej5ej6BiZ7niZ7d'd eed e'd ee(defddZ8de9fdd Z:er-ej;Z;eee>?d#rHe>e@vrOeAd$e>e@e>ZBdSergejCd%krge:sge;ZBdSej eDed&durue-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 Csz=t|}t+} |||d}|s|j|d}|r|||dd}||@s-J||}q 1s6wYWdSty[}z|jrM|jdnd}|WYd}~Sd}~ww)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) rrrsselrlistreadyexrvr#H/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/waiting.py wait_selector$s(       r%c Csz@t|\}}|s d}t(} ||||j|d}|||s(td|dd}||\}}q1s9wYWdSty^}z|j rP|j dnd}|WYd}~Sd}~ww)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) rrrrrrr r!r"r#r#r$ wait_connFs(       r)c sXtt}dtddffdd }zzt|} |t@}|t@}|s-|s-td|d|r<| ||t |rE| ||t z2|durR IdHnz t |IdHWn tygYnwW|rp|||rw||n|r|||r||ww|}qty}z|jr|jdnd} | WYd}~Sd}~ww) 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. staterNcs|OdSNsetr*evr r#r$wakeup{s zwait_async..wakeupTbad poll status: r)rrrrWAIT_RWAIT_Wr' InternalErrorclear add_readerREADY_R add_writerREADY_Wwaitrr remove_reader remove_writerrrr) rrrloopr1rreaderwriterr!r"r#r/r$ wait_asyncgsV      rAc sNtt}dtddffdd }zjt|\}}|sd} |t@}|t@}|s3|s3td|d|rB| ||t |rK| ||t zt |IdHW|r^|||re||n|rm|||ru||ww|\}}qtytd ty}z|jr|jdnd} | WYd}~Sd}~ww) 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*rNcs|dSr+r,r.r/r#r$r1s zwait_conn_async..wakeupTr2rr&)rrrrr3r4r'r5r6r7r8r9r:rr;r<r=rrr(rr) rrr>r1rrr?r@r!r"r#r/r$wait_conn_asyncsP       rBc Csz6t|}d}|f} t|t@r|n||t@r|n|||\}}}d} |r(t} |r.| tO} | s1q || }q tyT} z| jrF| jdnd} | WYd} ~ Sd} ~ ww)z Wait for a generator using select where supported. BUG: on Linux, can't select on FD >= 1024. On Windows it's fine. r#TrN) rrr3r4r8r:rrr) rrrremptyfnlistrlwlxlr r!r"r#r#r$ wait_selects2  rH EpollSelectorc Cszet|}|dus |dkrd}nt|d}tA}t|}||| d}|s1||}|r*|dd}d}|tj@rAt}|tj @rK|t O}| |}t|}| ||q&1s^wYWdSt y} z| jru| jdnd} | WYd} ~ Sd} ~ ww)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_evmasksrpollEPOLLOUTr8EPOLLINr:rmodifyrr) rrrrrLevmaskfileevsr0r r!r"r#r#r$ wait_epoll s:         rT PollSelectorc CszYt|}|dus |dkrd}nt|d}t}t|}||| d}|s0||}|r)|dd}d}|tj@r@t}|tj@rJ|t O}| |}t|}| ||q%t yw} z| j ri| j dnd} | WYd} ~ Sd} ~ ww)za Wait for a generator using poll where supported. Parameters are like for `wait()`. NrrJTr)rrKrrN _poll_evmasksrPOLLOUTr8POLLINr:rrQrr) rrrrrNrRrSr0r r!r"r#r#r$ wait_pollBs8        rYc Cs^tjd}|r-z |drWdSWdSty,}z td|WYd}~dSd}~wwdS)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!r#r#r$_is_select_patchedgs  rbr;PSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got win32SelectSelectorr+)E__doc__osrZrlogging selectorstypingrasynciorrrrrr r'abcr r r r _enumsrr_cmodulerRr3Wr4RWWAIT_RWr8r:READY_RW getLogger__name__r_rKfloatr%r)rArBrHhasattr EPOLLONESHOTrPEPOLLERRrOrMrTrXrWrVrYboolrbwait_c__annotations__environfname startswithglobals ImportErrorr;platformgetattrr#r#r#r$s      $ "" :$< !$ / $%