o EDi)@sVddlmZddlmZddlmZddlmZmZm Z eddZ Gd d d e Z d S) ) namedtuple)islice)Error)InterfaceErrorOperationalErrorProgrammingErrorColumnzAname type_code display_size internal_size precision scale null_okcseZdZGdddeZeZfddZddZddZd d Z d d Z e d dZ e ddZ ddZd=ddZddZddZd=ddZddZddZd=d d!Ze d"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd>d1d2Zd3d4Zd5d6Zd7d8Z d9d:Z!d;d<Z"Z#S)?Cursorc@seZdZed\ZZZZdS)z Cursor.StatesN)__name__ __module__ __qualname__rangeNONERUNNINGFINISHED CURSOR_CLOSEDrrW/var/www/Datamplify/venv/lib/python3.10/site-packages/clickhouse_driver/dbapi/cursor.pyStatessrcs2||_||_|d|_d|_tt|dS)Nr)_client _connection _reset_state arraysize_columns_with_typessuperr __init__)selfclient connection __class__rrrs zCursor.__init__cCs|j|jjk}dt||S)Nz()_state_statesrformatid)r is_closedrrr__repr__'szCursor.__repr__ccs |}|dur dS|VqN)fetchone)ronerrr__iter__.szCursor.__iter__cCs|Sr*rrrrr __enter__6szCursor.__enter__cCs |dSr*)close)rexc_typeexc_valexc_tbrrr__exit__9s zCursor.__exit__cCs:|j|jjkr dS|jp g}|jpg}ddt||DS)Nc Ss$g|]\}}t||dddddqS)NT)r ).0name type_coderrr Dsz&Cursor.description..)r$r%r_columns_typeszip)rcolumnstypesrrr description<s  zCursor.descriptioncC|jS)zQ :return: the number of rows that the last .execute*() produced. ) _rowcountr.rrrrowcountIszCursor.rowcountcCs>|j|jj|_z |jj|WdStyYdSw)z Close the cursor now. The cursor will be unusable from this point forward; an :data:`~clickhouse_driver.dbapi.Error` (or subclass) exception will be raised if any operation is attempted with the cursor. N) r disconnectr%rr$rcursorsremove ValueErrorr.rrrr0Ps   z Cursor.closeNc Csn||z|\}}||f|dd|}Wnty+}zt|d}~ww|||dS)a Prepare and execute a database operation (query or command). :param operation: query or command to execute. :param parameters: sequence or mapping that will be bound to variables in the operation. :return: None T)paramswith_column_typesN_check_cursor_closed _begin_query_prepare DriverErrorr_process_response _end_query)r operation parametersexecuteexecute_kwargsresponseorigrrrrQ`s"     zCursor.executec Csp||z|\}}||fd|i|}Wnty*}zt|d}~ww|j|dd|dS)aE Prepare a database operation (query or command) and then execute it against all parameter sequences found in the sequence `seq_of_parameters`. :param operation: query or command to execute. :param seq_of_parameters: sequences or mappings for execution. :return: None rFNT) executemanyrH)rrOseq_of_parametersrQrRrSrTrrrrUzs"    zCursor.executemanycCs0||jr t|jdS|jsdS|jdS)z Fetch the next row of a query result set, returning a single sequence, or None when no more data is available. :return: the next row of a query result set or None. Nr)_check_query_started_stream_resultsnext_rowspopr.rrrr+s   zCursor.fetchonecCsx||dur |j}|jr|dkrt|jStt|j|S|dkr+|j}g|_|S|jd|}|j|d|_|S)a. Fetch the next set of rows of a query result, returning a sequence of sequences (e.g. a list of tuples). An empty sequence is returned when no more rows are available. :param size: amount of rows to return. :return: list of fetched rows or empty list. Nr)rWrrXlistrZr)rsizervrrr fetchmanys  zCursor.fetchmanycCs(||jr t|jS|j}g|_|S)z Fetch all (remaining) rows of a query result, returning them as a sequence of sequences (e.g. a list of tuples). :return: list of fetched rows. )rWrXr]rZ)rr_rrrfetchalls  zCursor.fetchallcCdSr*r)rsizesrrr setinputsizeszCursor.setinputsizescCrbr*r)rr^columnrrr setoutputsizerezCursor.setoutputsizecCr?)z :return: list of column names with corresponding types of the last .execute*(). E.g. [('x', 'UInt64')]. )rr.rrrcolumns_with_typesszCursor.columns_with_typescCs||_||_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)rX_max_row_buffer)rstream_resultsmax_row_bufferrrrset_stream_resultss zCursor.set_stream_resultscC ||_dS)z} Specifies settings for cursor. :param settings: dictionary of query settings :return: None N) _settings)rsettingsrrr set_settings zCursor.set_settingscCrm)z Toggles type checking for sequence of INSERT parameters. Disabled by default. :param types_check: new types check value. :return: None N) _types_check)r types_checkrrrset_types_checks 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)_external_tables)rr6 structuredatarrrset_external_tables zCursor.set_external_tablecCrm)z Specifies the query identifier for cursor. :param query_id: the query identifier. :return: None N) _query_id)rquery_idrrr set_query_idrqzCursor.set_query_idcCsbdd|jDp d}|jj}|jr#|jj}|jpi|_|j|jd<|j||j|j d}||fS)NcSs g|] \}\}}|||dqS))r6rvrwr)r5r6rvrwrrrr8s  z#Cursor._prepare..max_block_size)roexternal_tablesrsrz) ruitemsrrQrX execute_iterrnrirrry)rr}rQrRrrrrKs   zCursor._prepareFcCs|r||_d}|rt|tr!g|_|_|_t|tr||_dS|jr+t|}|}n|\}}||_|rEt |\|_|_|jsDt ||_ng|_|_||_dSr*) r@ isinstanceintr9r:rZrXrYrr;len)rrSrUrhrowsrrrrM4s(    zCursor._process_responsecCsJ|jj|_d|_d|_d|_d|_d|_d|_d|_ d|_ i|_ d|_ dS)zE Resets query state and get ready for another query. Nr\Fr) r%rr$r9r:rZr@rXrirnryrurrr.rrrrTs  zCursor._reset_statecC|jj|_dSr*)r%rr$r.rrrrJfzCursor._begin_querycCrr*)r%rr$r.rrrrNirzCursor._end_querycC|j|jjkr tddS)Nzcursor already closed)r$r%rrr.rrrrIlzCursor._check_cursor_closedcCr)Nzno results to fetch)r$r%rr r.rrrrWprzCursor._check_query_startedr*)F)$r rrobjectrr%rr)r-r/r4propertyr>rAr0rQrUr+r`rardrgrhrlrprtrxr{rKrMrrJrNrIrW __classcell__rrr"rr sD            r N) collectionsr itertoolsrerrorsrrLrrr r rr rrrrs