o FDi /@srddlZddlmZddlmZddlmZmZddlm Z eddZ e e Z Gdd d ZGd d d eZdS) N) namedtuple)Optional)InterfaceErrorProgrammingError) CursorStatusColumnzAname type_code display_size internal_size precision scale null_okc@sZeZdZdZdJdefddZdefddZed d Z ed d Z edefd dZ ddZ ddZ ddZ  dKdefddZdLddZdKddZddZdeefdd Zd!d"Zd#d$Zd%d&Zd'd(ZdMd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zed3d4Zd5d6Z d7d8Z!d9d:Z"d;d<Z#d=d>Z$d?d@Z%dAdBZ&dCdDZ'dEdFZ(dNdHdIZ)dS)OCursorNFechocCs$||_|g|_||_d|_dS)N) _connection _reset_state_rows_echo _arraysize)self connectionr rG/var/www/Datamplify/venv/lib/python3.10/site-packages/asynch/cursors.py__init__s  zCursor.__init__returnc Cs<|jj}|j}d|d|jd|jdt|dd|d S)N) __class____name__statusr rid)rcls_namerrrr__repr__szCursor.__repr__cC|jS)zmThis read-only attribute return a reference to the Connection object on which the cursor was created.)r rrrrrszCursor.connectioncCr)zQ :return: the number of rows that the last .execute*() produced. ) _rowcountr rrrrowcount%szCursor.rowcountcCr)zReturn the status of the cursor. :return: the Cursor object status :rtype: str (CursorStatus StrEnum) )_stater rrrr,sz Cursor.statuscGdSz!Does nothing, required by DB API.Nrrargsrrr setinputsizes6zCursor.setinputsizescGr$r%rr&rrrsetoutputsizes9r)zCursor.setoutputsizescstj|_dSN)rclosedr#r rrrclose<s z Cursor.closequerycs~|||||\}}||f|dd|IdH}||IdH||jr executemanycolumns_with_typesrowsrrrr6Ts&   zCursor._process_responsecs|||||\}}||fd|i|IdH}|j|ddIdH||jr=t|td||j S)Nr'T)rHr0r1r:rrrrHms  zCursor.executemanycsN||jrz |jIdHWStyYdSw|js!dS|jdSNr)_check_query_startedrCr next Exceptionpopr rrrfetchone}s  zCursor.fetchonesizecs||dur |j}|dkrgS|jr6g}|j2z3dHW}|||dkr2t||kr2|Sq6|S|dkrB|j}g|_|S|jd|}|j|d|_|SrK)rLrrCr appendrG)rrQrvirrr fetchmanys. zCursor.fetchmanycs6||jrdd|j2IdHS|j}g|_|S)Ncsg|z3dHW}|q6Sr+r).0rowrrr sz#Cursor.fetchall..)rLrCr )rrSrrrfetchallszCursor.fetchallcCsHtj|_d|_d|_d|_d|_d|_d|_d|_ d|_ i|_ d|_ dS)zbReset the state of the cursor. Prepares a cursor object to handle another query. NFr) rreadyr#rArBr r!rC_max_row_buffer _settings _query_id_external_tables _types_checkr rrrr s zCursor._reset_statecCs2g}|jD]\}\}}||||dq|S)Nname structuredata)r`itemsrR)rtablesrcrdrerrr_make_external_tablesszCursor._make_external_tablesc Cs|dg}g}|j}|D])}g}|jD]}|j|j|d} ||j| fq||j||jdddq || |S)Nexternal_tables)type_expression clickhousererb) get type_compilercolumnsprocesstyperRrcdialect_optionsextendrh) rdialectexecution_optionsrirgrmtablerdctype_rrrmake_external_tabless    zCursor.make_external_tablescCs|r |j}||j|}ni}|pd}|jjj}|d|jp!i}|d|j|_|jr@|jjj }|d|j |_ |j |d<||_|||d|j |j d}||fS)Nsettingsstream_resultsmax_block_size types_check)ryrir|query_id) rtrxrsrhr r<rlr^rC execute_iterr]rar_)rr;rtrir<ryr=rrrr5s&     zCursor._preparecCs|Sr+rr rrr __aiter__szCursor.__aiter__cs |IdH}|s t|Sr+)rPStopAsyncIteration)ronerrr __anext__s zCursor.__anext__cs|IdHdSr+)r-)rexc_typeexc_valexc_tbrrr __aexit__szCursor.__aexit__cs|Sr+rr rrr __aenter__ szCursor.__aenter__cCs8|jtjkrdS|jp g}|jpg}ddt||DS)Nc Ss$g|]\}}t||dddddqS)NT)r)rVrc type_coderrrrXsz&Cursor.description..)r#rr\rArBrF)rrntypesrrr description s   zCursor.descriptioncCs|jtjkr td|dS)Nzno results to fetch from the )r#rr\rr rrrrLs zCursor._check_query_startedcCs|jjjr tddS)Nz_some records have not been fetched. fetch the remaining records before executing the next query)r is_query_executingrr rrrr3s zCursor._check_query_executingcCs |jtjkrtd|ddS)Nzthe z is already closed)r#rr,rr rrrr2$s zCursor._check_cursor_closedcC tj|_dSr+)rrunningr#r rrrr4( zCursor._begin_querycCrr+)rfinishedr#r rrrr7+rzCursor._end_querycCs||_||_dS)am Toggles results streaming from server. Driver will consume block-by-block of `max_row_buffer` size and yield row-by-row from each block. :param stream_results: enable or disable results streaming. :param max_row_buffer: specifies the maximum number of rows to buffer at a time. :return: None N)rCr])rrzmax_row_bufferrrrset_stream_results.s zCursor.set_stream_resultscC ||_dS)z} Specifies settings for cursor. :param settings: dictionary of query settings :return: None N)r^)rryrrr set_settings< zCursor.set_settingscCr)z Toggles type checking for sequence of INSERT parameters. Disabled by default. :param types_check: new types check value. :return: None N)ra)rr|rrrset_types_checkEs zCursor.set_types_checkcCs||f|j|<dS)a Adds external table to cursor context. If the same table is specified more than once the last one is used. :param name: name of external table :param structure: list of tuples (name, type) that defines table structure. Example [(x, 'Int32')]. :param data: sequence of rows of tuples or dicts for transmission. :return: None N)r`)rrcrdrerrrset_external_tableOs zCursor.set_external_tabler[cCr)z Specifies the query identifier for cursor. :param query_id: the query identifier. :return: None N)r_)rr}rrr set_query_id]rzCursor.set_query_id)NF)NN)Fr+)r[)*r __module__ __qualname__rEboolrstrrpropertyrr"rr(r*r-r<r6rHrPrr@rUrYr rhrxr5rrrrrrLr3r2r4r7rrrrrrrrrr sV            rcsVeZdZdeffdd Zdeedeeffdd Zdeeffdd Z Z S) DictCursorrcs8tIdH}|jr|rtt|j|SiStd)zFetch exactly one row from the last executed query. :raises AttributeError: columns mismatch :return: one row from the query :rtype: dict NInvalid columns.)superrPrAdictrFAttributeError)rrWrrrrPjs  zDictCursor.fetchonerQcs<t|IdH}jr|rfdd|DSgStd)a#Fetch no more than `size` rows from the last executed query. :param size Optional[int]: fetch upt to the `size` entries or self._arraysize if None :raises AttributeError: columns mismatch :return: the list of rows from the query :rtype: list[dict] Ncg|] }ttj|qSrrrFrArVitemr rrrXz(DictCursor.fetchmany..r)rrUrAr)rrQrJrr rrUxs  zDictCursor.fetchmanycs:tIdH}jr|rfdd|DSgStd)zFetch all resulting rows from the last executed query. :raises AttributeError: columns mismatch :return: the list of all possible rows from the query :rtype: list[dict] Ncrrrrr rrrXrz'DictCursor.fetchall..r)rrYrAr)rrJrr rrYs  zDictCursor.fetchall) rrrrrPrr@listrUrY __classcell__rrrrrisr)logging collectionsrtypingr asynch.errorsrrasynch.proto.models.enumsrr getLoggerrr8rrrrrrs     ^