U ڲg0@sNdZddlmZddlZddlmZddlmZddlmZ ddl m Z m Z m Z mZdd lmZmZmZdd lmZmZdd lmZdd lmZdd lmZejjZejjZejjZ ejj!Z"ejj#Z$ejj%Z&ej'j(Z(ej'j)Z)ej'j*Z*ej'j+Z+ej'j,Z,ej-Z.ej/Z0ej1Z2ej-Z3ej/Z4ej1Z5e6e7Z8ddddddddZ9ddddddddZ:dddddZ;dd dd!d"Zdd(d)d*d+d,Z?dd-dd.d/Z@dd0dd1d2ZAdd3dd4d5ZBdAdd7d8d d9d:d;ZCddd?d@ZDer.ejEZEejFZFejGZGejHZHejIZIejJZJejKZKne9ZEe:ZFe;ZGeZJe?ZKdS)Ba Generators implementing communication protocols with the libpq Certain operations (connection, querying) are an interleave of libpq calls and waiting for the socket to be ready. This module contains the code to execute the operations, yielding a polling state whenever there is to wait. The functions in the `waiting` module are the ones who wait more or less cooperatively for the socket to be ready and make these generators continue. These generators yield `Wait` objects whenever an operation would block. These generators assume the connection fileno will not change. In case of the connection function, where the fileno may change, the generators yield pairs (fileno, `Wait`). The generator can be restarted sending the appropriate `Ready` state when the file descriptor is ready. If a None value is sent, it means that the wait function timed out without any file descriptor becoming ready; in this case the generator should probably yield the same value again in order to wait more. ) annotationsN) monotonic)pq)errors)BufferPipelineCommandPQGen PQGenConn) PGcancelConnPGconnPGresult)WaitReady)Deque)_psycopg)conninfo_encoding)timeoutstrfloatzPQGenConn[PGconn])conninforreturnccs|rt|nd}tj|}|jtkrNt|}tj d| ||d| }|t ksf|t kr|t krrtnt}|j|fV}|rt|krtd|rvqqvq"|tkrqq"|tkrt|}tj d| |t|dq"tjd|t|dq"d|_|S)zE Generator to create a database connection without blocking. rzconnection is bad: )pgconnzconnection timeout expiredzconnection failed: unexpected poll status: r)rrr Z connect_startencodestatusBADreOperationalErrorget_error_messageZ connect_poll POLL_READING POLL_WRITINGWAIT_RWAIT_WsocketZConnectionTimeoutPOLL_OK POLL_FAILEDZ finish_pgconn InternalErrorZ nonblocking)rrdeadlineconnencodingrwaitreadyr.6/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/generators.py_connect=s<   r0r zPQGenConn[None]) cancel_connrrccs|rt|nd}|r*t|kr*td|}|tkr>qq|tkrT|jtfVq|tkrj|jt fVq|t krt d| qt d|qdS)Nrzcancellation timeout expiredzcancellation failed: r)rrZCancellationTimeoutpollr&r!r%r#r"r$r'rr r()r1rr)rr.r.r/_cancelhs  r3r zPQGen[list[PGresult]])rrccs t|EdHt|EdH}|S)aG Generator sending a query and returning results without blocking. The query must have already been sent using `pgconn.send_query()` or similar. Flush the query and then return the result using nonblocking functions. Return the list of results returned by the database (whether success or error). N)_send _fetch_many)rrvr.r.r/_execute|s r7z PQGen[None]ccs6|}|dkrq2tV}|rq q|t@r|qdS)aZ Generator to send a query to the server without blocking. The query must have already been sent using `pgconn.send_query()` or similar. Flush the query and then return the result using nonblocking functions. After this generator has finished you may want to cycle using `fetch()` to retrieve the results available. rN)flushWAIT_RWREADY_R consume_input)rfr-r.r.r/r4s r4ccsfg}t|EdH}|sqb|||j}|tksb|tksb|tkrBqb|tkrt|dksbt|qbq|S)a  Generator retrieving results from the database without blocking. The query must have already been sent to the server, so pgconn.flush() has already returned 0. Return the list of results returned by the database (whether success or error). Nr) _fetchappendrCOPY_INCOPY_OUT COPY_BOTH PIPELINE_SYNClenAssertionError)rresultsresrr.r.r/r5s  r5zPQGen[PGresult | None]ccsH|r8tV}|rqq||s(q8tV}|r(qq(qt||S)z Generator retrieving a single result from the database without blocking. The query must have already been sent to the server, so pgconn.flush() has already returned 0. Return a result from the database (whether success or error). )is_busyr#r;_consume_notifies get_result)rr-r.r.r/r=s r=zDeque[PipelineCommand]zPQGen[list[list[PGresult]]])rcommandsrccsg}tV}|rqq|t@r|t|g}|s|}|dkr\|sLq||g}q.|j}|tkr|rrt ||gq.|t ks|t ks|t krt dq.||q.|t@r||sq|q|S)zGenerator to send queries from a connection in pipeline mode while also receiving results. Return a list results, including single PIPELINE_SYNC elements. Nz$COPY cannot be used in pipeline mode)r9r:r;rHrGrIr>rrBrDr?r@rArZNotSupportedErrorREADY_Wr8popleft)rrJrEr-rFrrr.r.r/_pipeline_communicates<   rNNonecCs$|}|sq |jr||qdSN)notifiesZnotify_handler)rnr.r.r/rHs rHzPQGen[list[pq.PGnotify]]ccs2tV|g}|}|r.||qq.q|SrP)r#r;rQr>)rnsrRr.r.r/rQ$s rQzPQGen[memoryview | PGresult]ccs|d\}}|dkrq0tV}|rq&q|q|dkr<|St|EdH}t|dkr`td|d}|jtkrtj ||j d|S)Nrrz)you cannot mix COPY with other operationsr+) Z get_copy_datar#r;r5rCrZProgrammingErrorr COMMAND_OKerror_from_result _encoding)rnbytesdatar-rEresultr.r.r/ copy_from3s     r[Trbool)rbufferr8rccsH||dkrtV}|rqqq|rDtV}|r"q0q"|}|dkr"qDq"dS)Nr)Z put_copy_datar$r8)rr]r8r-r<r.r.r/copy_toPsr^z bytes | NonezPQGen[PGresult])rerrorrccsn||dkrtV}|rqqqtV}|rq,q|}|dkrq@qt|EdH\}|jtkrjtj||jd|S)NrrT) Z put_copy_endr$r8r5rrUrrVrW)rr_r-r<rZr.r.r/copy_endls r`)T)L__doc__ __future__rloggingtimerrrrabcrrr r Zpq.abcr r r Zwaitingrr_compatrZ_cmodulerZ _encodingsrZ ConnStatusOKrZ PollingStatusr&ZREADINGr!ZWRITINGr"ZFAILEDr'Z ExecStatusrUr@r?rArBRr#Wr$ZRWr9r:rKZREADY_RW getLogger__name__loggerr0r3r7r4r5r=rNrHrQr[r^r`connectcancelexecutesendZ fetch_manyfetchZpipeline_communicater.r.r.r/sp        + 6