3 Z_]@s6dZddlZddlmZddlZddZd ddZdS) zA wait callback to allow psycopg2 cooperation with eventlet. Use `make_psycopg_green()` to enable eventlet support in Psycopg. N) extensionscCs&ttdstdtjtjtdS)z?Configure Psycopg to be used with eventlet in non-blocking way.set_wait_callbackzAsupport for coroutines not available in this Psycopg version (%s)N)hasattrr ImportErrorpsycopg2 __version__reventlet_wait_callbackr r E/tmp/pip-build-yz_zf6az/eventlet/eventlet/support/psycopg2_patcher.pymake_psycopg_green!s   r cCsnxh|j}|tjkrPq|tjkr8tjj|jddq|tjkrXtjj|jddqt j d|qWdS)z>A wait callback useful to allow eventlet to work with Psycopg.T)read)writezBad result from poll: %rN) pollrZPOLL_OKZ POLL_READeventletZhubsZ trampolinefilenoZ POLL_WRITErZOperationalError)conntimeoutstater r r r+s   r)r)__doc__rrZ eventlet.hubsrr rr r r r s