o "4g\2@s^UdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZmZdd lmZejZejZejZ ej!Z"ejZ#ejZ$ejZ%e&e'Z(d.d/ddZ)d.d0ddZ*d.d/ddZ+d.d0ddZ,d.d/ddZ-e.edreej/ej0Bej1Beej/ej2Bej1Be ej/ej0ej2Bej1BBiZ3niZ3d.d/dd Z4e.ed!reej5eej6e ej5ej6BiZ7niZ7d.d/d"d#Z8d1d%d&Z9erej:Z:d'e;d(<d)ejd*re=e?vre@d+e=e?e=ZAdSerejBd,kre9se:ZAdSej eCed-dur!e-ZAdSe.ed!r+e8ZAdSe)ZAdS)2z 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 Csz?t|}t-}||| |j|d}|s|tq|||dd}||}|||q1s8wYWdSty]}z|jrO|jdnd}|WYd}~Sd}~ww)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`. TtimeoutrrN nextrregisterselectsend READY_NONE unregister StopIterationargs) rrrsselrlistreadyexrvr)X/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/waiting.py wait_selector%s(        r+ PQGenConn[RV]c CszGt|\}}|s d}t/}||| |j|d}|s$|tq|||dd}||\}}|||q1s@wYWdStye}z|jrW|jdnd}|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 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. NTrrrr) rrrr#r$r%r&r'r(r)r)r* wait_connGs,        r-c srtt}d fdd }zyt|} |t@}|t@}|s)|s)td|d |r8|||t |rA| ||t z1|d ur\z t |Id HWnty[Ynw Id HW|rk|||rr||n|rz|||r||ww|}qty}ztt|d }~wty}z|jr|jd nd } | WYd }~Sd }~ww) 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. staterrNonecs|OdSNsetr.evr&r)r*wakeup~s zwait_async..wakeupTbad poll status: rNr.rrr/)rrrWAIT_RWAIT_We InternalErrorclear add_readerREADY_R add_writerREADY_Wrwaitr remove_reader remove_writerrOSErrorOperationalErrorstrr!r") rrrloopr6r#readerwriterr'r(r)r4r* wait_asyncjs\      rKc sRtt}d fdd }z{t|\}} |t@}|t@}|s+|s+td|d |r:|||t |rC| ||t z/|r\z t |Id HWnty[Ynw Id HW|rk|||rr||n|rz|||r||ww|\}}qty}z|jr|jd nd } | 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 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. r.rrr/cs|dSr0r1r3r4r)r*r6s zwait_conn_async..wakeupTr7rNr8)rrrr9r:r;r<r=r>r?r@rArrBrrCrDrr!r") rrrHr6rr#rIrJr'r(r)r4r*wait_conn_asyncsV      rLc Csz;t|}d}|f} t|t@r|n||t@r|n|||\}}}d} |r(t} |r.| tO} | s6|tq || }q tyY} z| j rK| j dnd} | 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) rrr9r:r?rArrr!r") rrrr#emptyfnlistrlwlxlr&r'r(r)r)r* wait_selects4   rR EpollSelectorc Csz`t|}|dus |dkrd}tC}t|}||| ||}|s,|tq|dd}d}|tj@rs|       " # = = "  -   &