U ڲg\2@sUdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZmZdd lmZejZejZejZ ej!Z"ejZ#ejZ$ejZ%e&e'Z(d-d d d ddddZ)d.dd ddddZ*d/d d d ddddZ+d0dd ddddZ,d1d d d ddddZ-e.edreej/ej0Bej1Beej/ej2Bej1Be ej/ej0ej2Bej1BBiZ3niZ3d2d d d ddddZ4e.edreej5eej6e ej5ej6BiZ7niZ7d3d d d ddd d!Z8d"d#d$d%Z9erej:Z:d&e;d'<d(ejd)r8e=e?krFe@d*e=e?e=ZAnPerrejBd+krre9sre:ZAn0ej eCed,dkre-ZAne.edre8ZAne)ZAdS)4z 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)_psycopgz PQGen[RV]intz float | Noner )genfilenointervalreturnc Csztt|}t\}||||j|d}|s:|tq|||dd}||}|||qW5QRXWn>tk r}z |jr|jdnd}|WYSd}~XYnXdS)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`. timeoutrrN nextrregisterselectsend READY_NONE unregister StopIterationargs) rrrsselrlistreadyexrvr'3/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/waiting.py wait_selector%s      r)z PQGenConn[RV])rrrc Cszt|\}}|sd}t`}||||j|d}|sF|tq*|||dd}||\}}|||q*W5QRXWn>tk r}z |jr|jdnd}|WYSd}~XYnXdS)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. Nrrrr) rrrr!r"r#r$r%r&r'r'r( wait_connGs"       r*c sttt}dddfdd }zt|}|t@}|t@}|sT|sTtd|d|rr|||t |r| ||t zJ|dk rzt|IdHWqtk rYqXnIdHW5|r| ||r| |X|}q,Wnrtk r.}ztt|W5d}~XYnBtk rn}z"|jrR|jdnd} | WYSd}~XYnXdS) 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. rNonestatercs|OdSNsetr-evr$r'r(wakeup~szwait_async..wakeupbad poll status: rN)rrrWAIT_RWAIT_We InternalErrorclear add_readerREADY_R add_writerREADY_W remove_reader remove_writerrwaitrrOSErrorZOperationalErrorstrrr ) rrrloopr4r!readerwriterr%r&r'r2r( wait_asyncjs@  rGc sHtt}dddfdd }zt|\}}|t@}|t@}|sX|sXtd|d|rv|||t |r| ||t zF|rzt|IdHWqtk rYqXnIdHW5|r| ||r| |X|\}}q0WnBtk rB}z"|jr&|jdnd} | WYSd}~XYnXdS) 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. rr+r,cs|dSr.r/r1r2r'r(r4szwait_conn_async..wakeupr5rN)rrrr6r7r8r9r:r;r<r=r>r?r@rrArrrr ) rrrDr4rr!rErFr%r&r'r2r(wait_conn_asyncs<   rHc Cszxt|}d}|f}t|t@r$|n||t@r2|n|||\}}}d} |rNt} |rZ| tO} | sj|tq|| }qWn>tk r} z | j r| j dnd} | WYSd} ~ XYnXdS)z Wait for a generator using select where supported. BUG: on Linux, can't select on FD >= 1024. On Windows it's fine. r'rN) rrr6r7r<r>rrrr ) rrrr!emptyZfnlistZrlZwlZxlr$r%r&r'r'r( wait_selects,  rJ EpollSelectorc Cszt|}|dks|dkrd}t}t|}|||||}|sV|tq<|dd}d}|tj@rvt }|tj @r|t O}||}t|}| ||qt k r} z | jr| jdnd} | WYSd} ~ XYnXdS)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<EPOLLINr>modifyrr ) rrrr!rLevmaskfileevsr3r$r%r&r'r'r( wait_epolls.        rT PollSelectorc Cszt|}|dks|dkr d}n t|d}t}t|}|||||}|sb|tqH|dd}d}|tj@rt }|tj @r|t O}||}t|}| ||qHWn>t k r} z | jr| jdnd} | WYSd} ~ XYnXdS)za Wait for a generator using poll where supported. Parameters are like for `wait()`. Nrg@@r)rrrrN _poll_evmasksrrrPOLLOUTr<POLLINr>rQrr ) rrrr!rNrRrSr3r$r%r&r'r'r( wait_pollHs0        rYbool)rc CsXtjd}|rTz|dr"WdSWn.tk rR}ztd|W5d}~XYnXdS)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)sysmodulesgetZis_module_patched Exceptionloggerwarning)mr%r'r'r(_is_select_patchedns   rbr rAZPSYCOPG_WAIT_FUNCZwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got win32SelectSelector)N)N)N)N)N)N)N)D__doc__ __future__rosr[rlogging selectorsZasynciorrrrrr r8abcr r r r Z_enumsrrZ_cmodulerRr6Wr7ZRWZWAIT_RWNONErr<r>ZREADY_RW getLogger__name__r_r)r*rGrHrJhasattr EPOLLONESHOTrPEPOLLERRrOrMrTrXrWrVrYrbZwait_c__annotations__environfname startswithglobals ImportErrorrAplatformgetattrr'r'r'r(s     "#==" -  &