U d/e)@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)ACursorc@seZdZed\ZZZZdS)z Cursor.StatesN)__name__ __module__ __qualname__rangeNONERUNNINGFINISHED CURSOR_CLOSEDrrB/tmp/pip-unpacked-wheel-mds_o03w/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)rZ is_closedrrr__repr__'s zCursor.__repr__ccs |}|dkrdS|VqdSN)fetchone)rZonerrr__iter__.szCursor.__iter__cCs|Sr)rrrrr __enter__6szCursor.__enter__cCs |dSr))close)rexc_typeexc_valexc_tbrrr__exit__9szCursor.__exit__cCs:|j|jjkrdS|jpg}|jp$g}ddt||DS)Nc Ss$g|]\}}t||dddddqS)NT)r ).0nameZ type_coderrr Dsz&Cursor.description..)r$r%r_columns_typeszip)rcolumnstypesrrr description<s  zCursor.descriptioncCs|jS)zQ :return: the number of rows that the last .execute*() produced. ) _rowcountr,rrrrowcountIszCursor.rowcountcCs@|j|jj|_z|jj|Wntk r:YnXdS)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) rZ disconnectr%rr$rZcursorsremove ValueErrorr,rrrr.Ps   z Cursor.closeNc Csx||z&|\}}||f|dd|}Wn*tk r`}z t|W5d}~XYnX|||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)paramsZwith_column_typesN_check_cursor_closed _begin_query_prepare DriverErrorr_process_response _end_query)r operation parametersexecuteexecute_kwargsresponseorigrrrrJ`s    zCursor.executec Csz||z$|\}}||fd|i|}Wn*tk r^}z t|W5d}~XYnX|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 r@NT) executemanyrA)rrHZseq_of_parametersrJrKrLrMrrrrNzs   zCursor.executemanycCs4||jrt|jdS|js$dS|jdSdS)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.fetchonecCsv||dkr|j}|jr>|dkr.t|jStt|j|S|dkrT|j}g|_n|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)rOrrPlistrRr)rsizervrrr fetchmanys  zCursor.fetchmanycCs(||jrt|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. )rOrPrUrR)rrWrrrfetchalls  zCursor.fetchallcCsdSr)r)rZsizesrrr setinputsizesszCursor.setinputsizescCsdSr)r)rrVcolumnrrr setoutputsizeszCursor.setoutputsizecCs|jS)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)rP_max_row_buffer)rZstream_resultsZmax_row_bufferrrrset_stream_resultss zCursor.set_stream_resultscCs ||_dS)z} Specifies settings for cursor. :param settings: dictionary of query settings :return: None N) _settings)rsettingsrrr set_settingsszCursor.set_settingscCs ||_dS)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_checkszCursor.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)rr4 structuredatarrrset_external_tables zCursor.set_external_tablecCs ||_dS)z Specifies the query identifier for cursor. :param query_id: the query identifier. :return: None N) _query_id)rquery_idrrr set_query_idszCursor.set_query_idcCsbdd|jDpd}|jj}|jrF|jj}|jp6i|_|j|jd<|j||j|j d}||fS)NcSs g|]\}\}}|||dqS))r4rgrhr)r3r4rgrhrrrr5s z#Cursor._prepare..Zmax_block_size)raexternal_tablesrdrk) rfitemsrrJrPZ execute_iterr`r^rcrj)rrmrJrKrrrrDs   zCursor._prepareFcCs|r||_d}|rt|trBg|_|_|_t|tr>||_dS|jrVt|}|}n|\}}||_|rt |\|_|_|jst ||_n g|_|_||_dSr)) r< isinstanceintr6r7rRrPrQrr8len)rrLrNr]ZrowsrrrrF4s&   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. NrTFr) r%rr$r6r7rRr<rPr^r`rjrfrcr,rrrrTs zCursor._reset_statecCs|jj|_dSr))r%rr$r,rrrrCfszCursor._begin_querycCs|jj|_dSr))r%rr$r,rrrrGiszCursor._end_querycCs|j|jjkrtddS)Nzcursor already closed)r$r%rrr,rrrrBlszCursor._check_cursor_closedcCs|j|jjkrtddS)Nzno results to fetch)r$r%rr r,rrrrOpszCursor._check_query_started)N)N)N)F)$r rrobjectrr%rr(r+r-r2propertyr;r=r.rJrNr*rXrYrZr\r]r_rbrerirlrDrFrrCrGrBrO __classcell__rrr"rr sB           r N) collectionsr itertoolsrerrorsrrErrr r rrr rrrrs