o Tj@sdZddlmZddlZddlZddlmZddlm Z m Z m Z ddl Z ejr-ddlmZiddd d d d d dddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5 ZGd6d7d7ZGd8d9d9ZdFd>d?ZdGdDdEZdS)HzE Utility functions and objects for implementing the interchange API. ) annotationsN)lib) ArrowDtypeCategoricalDtypeDatetimeTZDtype)DtypeObjnullnboolbuint8Cuint16Suint32Iuint64Lint8cint16int32iint64l halffloatefloatfdoublegstringu large_stringUbinaryzz time32[s]ttsttmttuttntdDtdmztss:ztsm:ztsu:ztsn:tDstDmtDutDn) z time32[ms]z time64[us]z time64[ns]z date32[day]z date64[ms]z timestamp[s]z timestamp[ms]z timestamp[us]z timestamp[ns]z duration[s]z duration[ms]z duration[us]z duration[ns]c@s\eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdS) ArrowCTypesz Enum for Apache Arrow C type format strings. The Arrow C data interface: https://arrow.apache.org/docs/format/CDataInterface.html#data-type-description-format-strings r r rr srrrrrrrr r"r$r+r,zts{resolution}:{tz}ztt{resolution}N)__name__ __module__ __qualname____doc__NULLBOOLINT8UINT8INT16UINT16INT32UINT32INT64UINT64FLOAT16FLOAT32FLOAT64STRING LARGE_STRINGDATE32DATE64 TIMESTAMPTIMErJrJ`/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/core/interchange/utils.pyr1=s*r1c@s eZdZdZdZdZdZdZdS) Endiannessz.Enum indicating the byte-order of a data-type.<>=|N)r3r4r5r6LITTLEBIGNATIVENArJrJrJrKrL_s rLdtyperreturnstrcCsHt|trtjS|tdkrtjSt|trSddl}|j }|j |r.d|j d|j S|j |rE|jdurEd|jdd|jStt|d}|durS|Stt|jd}|durb|St|tjrktjSt|drt|dd}tjj|d d St|trtjj|jd|jd St|tjrtj St!d |d ) a  Represent pandas `dtype` as a format string in Apache Arrow C notation. Parameters ---------- dtype : np.dtype Datatype of pandas DataFrame to represent. Returns ------- str Format string in Apache Arrow C notation of the given `dtype`. OrNzd:,ts:M) resolutiontzzConversion of z- to Arrow C format string is not implemented.)" isinstancerr1r?nprUrDrpyarrow pyarrow_dtypetypes is_decimal precisionscale is_timestampr_unitPYARROW_CTYPESgetrWgetattrnameupperpd StringDtyper is_np_dtype datetime_datarHformatr BooleanDtyper8NotImplementedError)rUpapa_type format_strr^rJrJrKdtype_to_arrow_c_fmths:        ryseries pd.Series allow_copypd.Series | NonecCsXt|jtjs dS|jj}t|jdkrdS|std| }tj ||j|j |j dS)a Rechunk a multi-chunk pyarrow array into a single-chunk array, if necessary. - Returns `None` if the input series is not backed by a multi-chunk pyarrow array (and so doesn't need rechunking) - Returns a single-chunk-backed-Series if the input is backed by a multi-chunk pyarrow array and `allow_copy` is `True`. - Raises a `RuntimeError` if `allow_copy` is `False` and input is a based by a multi-chunk pyarrow array. NzFound multi-chunk pyarrow array, but `allow_copy` is False. Please rechunk the array before calling this function, or set `allow_copy=True`.)rUrmindex) r`rUrorarray _pa_arraylenchunks RuntimeErrorcombine_chunksSeriesrmr)rzr| chunked_arrayarrrJrJrK maybe_rechunks r)rUrrVrW)rzr{r|r rVr})r6 __future__rtypingnumpyra pandas._libsrpandas.core.dtypes.dtypesrrrpandasro TYPE_CHECKINGpandas._typingrrjr1rLryrrJrJrJrKs|        "" 8