U Ú²gy ã@s<dZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z dd l m Zdd lmZmZmZdd lmZdd lmZdd lmZddlmZddlmZerÔddlmZddlmZddl m!Z!e j"j#Z#e j"j$Z$Gdd„deeefƒZ%Gdd„de%defeeƒZ&Gdd„de%defeeƒZ'dS)z% psycopg client-side binding cursors é)Ú annotations)Ú TYPE_CHECKING)Úpartialé)Ú PostgresQueryÚPostgresClientQuery)Úpq)Úadapt)Úerrors)ÚConnectionTypeÚQueryÚParams)ÚRow)ÚCursor)ÚPrepare)Ú BaseCursor)Ú AsyncCursor)ÚAny)Ú Connection)ÚAsyncConnectionc@sTeZdZeZdddddœdd„Zddd œd d d d dœdd„Zdd d ddœdd„ZdS)ÚClientCursorMixinNr z Params | NoneÚstr)ÚqueryÚparamsÚreturncCs(t |¡|_| ||¡}|j |jj¡S)z£ Return the query and parameters merged. Parameters are adapted and merged to the query the same way that `!execute()` would do. )r Z TransformerZ_txZ_convert_queryrÚdecodeÚencoding)ÚselfrrÚpgq©rú9/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/client_cursor.pyÚmogrify$s  zClientCursorMixin.mogrifyF)Úforce_extendedÚbinaryrÚboolz bool | NoneÚNone)rr"r#rcCs„|dkr|j}n |rtnt}|tkr.t d¡‚||_|jjr\|jjj  t |j j |j dƒ¡n$|rr|j   |j d¡n|j  |j ¡dS)Nz0client-side cursors don't support binary results)ÚformatÚBINARYÚTEXTÚeZNotSupportedErrorZ_queryZ_connZ _pipelineZ command_queueÚappendrZ_pgconnZsend_query_paramsrZ send_query)rrr"r#Úfmtrrr Ú _execute_send0s ÿ ÿzClientCursorMixin._execute_sendztuple[Prepare, bytes])rÚpreparercCs tjdfS)Nó)rÚNO)rrr-rrr Ú _get_preparedPszClientCursorMixin._get_prepared)N)N)Ú__name__Ú __module__Ú __qualname__rZ _query_clsr!r,r0rrrr r!sû!ÿrc@seZdZdZdS)Ú ClientCursorÚpsycopgN©r1r2r3rrrr r4Vsr4zConnection[Any]c@seZdZdZdS)ÚAsyncClientCursorr5Nr6rrrr r7Zsr7zAsyncConnection[Any]N)(Ú__doc__Ú __future__rÚtypingrÚ functoolsrZ_queriesrrÚrr r r)Úabcr r r ZrowsrÚcursorrZ _preparingrZ _cursor_baserZ cursor_asyncrrÚ connectionrZconnection_asyncrZFormatr(r'rr4r7rrrr Ús2              5  ÿ