gi6UdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZdd lmZdd lmZejZejZejZ ej!Z"ejZ#ejZ$ejZ%ej&e'Z(ej)d krd+dZ*nd+dZ*d,d-dZ+d,d.dZ,d,d-dZ-d,d.dZ.d,d-dZ/e0edr8eej1ej2zeej1ej3ze ej1ej2zej3ziZ4niZ4d,d-d Z5e0ed!r0eej6eej7e ej6ej7ziZ8ej6ej7zZ9niZ8d,d-d"Z:d/d$Z;erej<ZvrNej>d'Z?e?@d(r e?eAvreBd)e?eAe?ZCdSerej)d kre;se.wakeups  r$Tbad poll status: rrrArrr)r.rrr/WAIT_RWAIT_Wr InternalErrorclear add_readerREADY_R add_writerREADY_Wrwaitr remove_reader remove_writerr2OSErrorrr4r5) r&rr(looprGr6readerwriterrr:rFr9s @@r wait_asyncrY~sE@AAA B   D  II ZFZF ?f ?o&=!&=&=>>> HHJJJE 9888 9888 /"27799h7777777777#D/&&v.../&&v.../&&v.../&&v..../A+ . EEE !;<<"D  s`BF (C65D56 DD5DD5/F51E&&F F? F F?,F:4F?:F?c K|tdt t}d fd } t|\}} |tz}|t z}|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}|j} | 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. :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. Nr+rArrrc4|dSr"rCrEs rrGzwait_conn_async..wakeups r$TrHrrI)r.rrr/rJrKrrLrMrNrOrPrQrrRrrSrTr2r4r5) r&r(rVrGrr6rWrXrr:rFr9s @@rwait_conn_asyncr\sA@AAA B   D II  (ZFZF ?f ?o&=!&=&=>>> HHJJJE 9888 9888 /$&rwwyy(;;;;;;;;;;'''))OOOOOOO/&&v.../&&v.../&&v.../&&v..../IFA1 (4  sZBF"E(C;:E; DEDE%/F"1FF"" F?,F:4F?:F?c|td t|}d}|f} tj|tzr|n||tzr|n|||\}}}|r#t |t jdd} |rt} |r | tz} | | }#t$r} t jd| d} ~ wt$r} | j } | 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. Nr+r#Trr)r.r/r1rJrKr rrrOrQr2rUr4r5) r&rr(r6emptyfnlistrlwlxlr9rr:s r wait_selectrcs> @AAA II f*/%f*/% JBB  E (((()CDDDE  ! A# & EEE !;<<"D  s*BB-- C,7C  C,C'!C,'C, EpollSelectorc|td t|}|dkrd}tj5}t|}||| ||x}s*t||tB|dd}d}|tj zrt}|tj zr |tz}||}t|}|||#1swxYwYdS#t$r} | j} | 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 Nr+rr%Tr )r.r/r1epoll_epoll_evmasksr0pollr r2r3EPOLLINrOEPOLLOUTrQmodifyr4r5) r&rr(r6rfevmaskfileevsrFr9rr:s r wait_epollrn-s @AAA II a<<H \^^ -u#A&F NN66 * * * -#(::h#7#77$V,,,HHZ(((QZ]&$#E'%W$EHHUOO'* VV,,, - - - - - - - - - - -$  s;*D(CDDD("D#D(( E2E:EE PollSelectorc|td t|}|dkrd}nt|dz}tj}t |}||| ||x}s|t3|dd}d}|tj zrt}|tj zr |tz}|s-|tzr#t|tjd||}t |}|||#t$$r} | j} | cYd} ~ Sd} ~ wwxYw)za Wait for a generator using poll where supported. Parameters are like for `wait()`. Nr+rg@@Tr r)r.r/rr1rh _poll_evmasksr0r2r3POLLINrOPOLLOUTrQPOLL_BADr rrrkr4r5) r&rr(r6rhrlrmrFr9rr:s r wait_pollrugs @AAA# II a<<HH8f,--H{}}q! ff%%% (#yy222G $$$ABEFM! FN" !  ER(] E (((()CDDDA"1%F KK ' ' ') (,  sD:E E*E%E*%E*boolctjdx}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.monkeyr1Tz+failed to detect gevent monkey-patching: %sNF)sysmodulesgetis_module_patchedrloggerwarning)mrs r_is_select_patchedrs KOOO , ,,qN N""8,, t  N N N NNH" M M M M M M M M N 5s; A*A%%A*rrRPSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got SelectSelector)rrrr)r%)r&r'rrr(r)rr )r&r<r(r)rr )rrv)E__doc__ __future__rrrxr1logging selectorsasynciorrrrrr rabcr r r r_enumsrr_cmodulerRrJWrKRWWAIT_RWNONEr3rOrQREADY_RW getLogger__name__r|platformr r;r?rYr\rchasattr EPOLLONESHOTrirjrgrnrrrsrqrtrurwait_c__annotations__environfname startswithglobals ImportErrorrRgetattrr#r$rrs #""""" AAAAAAAAAAAA%%%%%%////////////""""""    ' Z ' ' 8  8 $ $<7IIIII!!!!!HD:::::z:::::@%%%%%P 79o&&#fn4#fo5$v~5GNN N,,,,,^ 79n%% /M /0HHM+++++\(  _F"*$$ J* +E   G $ $ WWYY(>(>k       799U DDD  #,'))2D2D2F2F) DDD'')5Et"L"LLL DDD WY'' DDD DDDr$