o EDiH! @sddlZddlZddlZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZddlmZejdkZedjZdZdd dd d Zd d dddZedkrQded<ddeDZded<ded<dddddZdedefd d!ZdMd"ed#ed$ed%eefd&d'Zd(ed$efd)d*Z d(ed$efd+d,Z!d-efd.d/Z"d0ed1efd2d3Z#dNd4ed5eed1efd6d7Z$d4ed8ed(e%d1efd9d:Z&d;d<Z'd=eeeefd1efd>d?Z(d=eeeeffd@dAZ)dOd#edCefdDdEZ*GdFdGdGeZ+GdHdIdIZ,dJeed1ee egeffdKdLZ-dS)PN)SequenceMutableSequenceDictOptionalUnion GeneratorCallable)ProgrammingErrorStreamClosedError DataError)Closablebigibhq)r &O) @rlrcCsi|]\}}||qSr).0kvrrY/var/www/Datamplify/venv/lib/python3.10/site-packages/clickhouse_connect/driver/common.py sr$frdz[s]z[ms]z[us]z[ns])rrsizesignedcCs2zt|}Wn tyYdSw|r|S|S)z Determines the Python array.array code for the requested byte size :param size: byte size :param signed: whether int types should be signed or unsigned :return: Python array.array code N) array_mapKeyErrorupper)r)r*coderrr# array_types   r/r.columndestcol_namec Csxztdt||}||j|7}WdStttjfy;}zd}|r,d|d}td|dd|d}~ww) a+ Write a column of native Python data matching the array.array code :param code: Python array.array code matching the column data type :param column: Column of native Python values :param dest: Destination byte buffer :param col_name: Optional column name for error tracking <z for source column ``zUnable to create Python arrayz3. This is usually caused by trying to insert None z4values into a ClickHouse column that is not NullableN)structStructlenpack TypeError OverflowErrorerrorr )r.r0r1r2buffexcol_msgrrr# write_array)s  r@valuecCs||dddS)z Write a single UInt64 value to a binary write buffer :param value: UInt64 value to write :param dest: Destination byte buffer rlittleN)extendto_bytes)rAr1rrr# write_uint64<srEcCs8 |d@}|dL}|dkr||dS|d|Bq)z Write a LEB128 encoded integer to a target binary buffer :param value: Integer value (positive only) :param dest: Target buffer TrNr)append)rAr1rrrr# write_leb128Es rIpreccCsH|dks|dkrtd|d|dkrdS|dkrdS|d kr"d Sd S) z Determine the bit size of a ClickHouse or Python Decimal needed to store a value of the requested precision :param prec: Precision of the Decimal in total number of base 10 digits :return: Required bit size rrzInvalid precision z for ClickHouse Decimal type rr'rr)ArithmeticError)rJrrr# decimal_sizeTsrOxreturncCs$|dr|dr|ddS|S)Nr5r) startswithendswith)rPrrr#unescape_identifieres rUsourceupdatecCs"|r|ni}|r|||SN)copyrW)rVrWrYrrr# dict_copyks rZkeycCs|dur|||<|SrXr)rVr[rArrr#dict_addrsr\ccsdEdHdS)Nrrrrrr# empty_genxsr]valcCs|sdSt|S)Nr)intr^rrr# coerce_int|sracCs&|sdS|dupt|to|dvS)NFT)true1yyes) isinstancestrlowerr`rrr# coerce_boolsriTnullablecCs,|r tdd|DdSt|r|dSdS)Ncss|] }|dur|VqdSrXr)r rPrrr# szfirst_value..r)nextr8)r0rjrrr# first_values rmc@sTeZdZdZdZddedeefddZdd Z d d Z d d Z ddZ ddZ dS) SliceViewa Provides a view into a sequence rather than copying. Borrows liberally from https://gist.github.com/mathieucaroff/0cf094325fb5294fb54c6a577f05a2c1 Also see the discussion on SO: https://stackoverflow.com/questions/3485475/can-i-create-a-view-on-a-python-list )_source_rangeNrV source_slicecCsXt|tr|j|_|j||_dS||_|dur!tt||_dStt|||_dSrX)rfrnrorpranger8)selfrVrqrrr#__init__s zSliceView.__init__cCs t|jSrX)r8rprsrrr#__len__s zSliceView.__len__cCs&t|tr t|j|S|j|j|SrX)rfslicernrorp)rsrrrr# __getitem__s  zSliceView.__getitem__cCs"|j}t|jt|j|j|jSrX)rprgrorwstartstopsteprsrrrr#__str__szSliceView.__str__cCs&|j}d|jt|j|j|jdS)Nz SliceView())rprorwryrzr{r|rrr#__repr__s zSliceView.__repr__cCsF||urdSt|t|krdSt||D] \}}||kr dSqdS)NTF)r8zip)rsotherr"wrrr#__eq__szSliceView.__eq__rX)__name__ __module__ __qualname____doc__slotsrrrwrtrvrxr~rrrrrr#rns  rnc@sFeZdZdZdZdedefddZddZd d Z d d Z d dZ dS) StreamContextz Wraps a generator and its "source" in a Context. This ensures that the source will be "closed" even if the generator is not fully consumed or there is an exception during consumption rVgen _in_contextrVrcCs||_||_d|_dSNFr)rsrVrrrr#rts zStreamContext.__init__cCs|SrXrrurrr#__iter__szStreamContext.__iter__cCs|jstdt|jS)Nz&Stream should be used within a context)rr rlrrurrr#__next__s zStreamContext.__next__cCs|jstd|_|S)NT)rr rrurrr# __enter__szStreamContext.__enter__cCsd|_|jd|_dSr)rrVcloser)rsexc_typeexc_valexc_tbrrr#__exit__s  zStreamContext.__exit__N) rrrr __slots__r rrtrrrrrrrr#rs rmethodcsdtdtfdddtdtfdddtdtfdd|sdS|}|d kr+dS|d kr1S|d kr7S|d krBfd dS|dkrHS|dkrSfddSgd}td|d|)NsrQcSs^|sdSgd}}|D]}|s|dkrd}q |r$||d}q ||q d|S)Nr4F_T)isspacerHr-join)routupchrrr# _to_camels   z$get_rename_method.._to_camelcSs|sdSgd}}|D]B}|r|dkr|dd}q |rF|dkr2|d||n|dkr>||n||d}q ||d}q |r_|ddkr_d|ddSd|S)Nr4rrrr)rrHisupperrhr)rrprevrrrr#_to_underscores&     ,z)get_rename_method.._to_underscorecSs&|d}|dkr||ddS|S)N.rr)rfind)rrrrr#_remove_prefixs z)get_rename_method.._remove_prefixNONE REMOVE_PREFIX TO_CAMELCASETO_CAMELCASE_WITHOUT_PREFIXc |SrXrr)rrrr# z#get_rename_method.. TO_UNDERSCORETO_UNDERSCORE_WITHOUT_PREFIXcrrXrr)rrrr#rr)none remove_prefix to_camelcaseto_camelcase_without_prefix to_underscoreto_underscore_without_prefixzInvalid option 'z'. Expected one of )rgstripr- ValueError)rname valid_optionsr)rrrr#get_rename_methods( rrX)NN)T).arrayr6systypingrrrrrrr$clickhouse_connect.driver.exceptionsr r r clickhouse_connect.driver.typesr byteorder must_swapitemsizeint_sizelow_card_versionr+ decimal_precitems array_sizes np_date_typesr_boolr/rgr@rErIrOrUrZanyr\r]rarirmrnrrrrrr#s@$     .(