eh2dZddlmZddlZddlZddlmZddlm Z m Z m Z ddl Z ejrddlmZiddd d d d d dddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5 ZGd6d7ZGd8d9ZdzE 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\eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZy) 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 rrsrrrrrrrr!r#r%r,r-zts{resolution}:{tz}ztt{resolution}N)__name__ __module__ __qualname____doc__NULLBOOLINT8UINT8INT16UINT16INT32UINT32INT64UINT64FLOAT16FLOAT32FLOAT64STRING LARGE_STRINGDATE32DATE64 TIMESTAMPTIME[/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/interchange/utils.pyr3r3=sj D D D E E F E F E FGGG FL F F &I DrMr3c eZdZdZdZdZdZdZy) Endiannessz.Enum indicating the byte-order of a data-type.<>=|N)r5r6r7r8LITTLEBIGNATIVENArLrMrNrPrP_s8 F C F BrMrPct|trtjS|t j dk(rtj St|trddl}|j}|jj|rd|jd|jS|jj|r+|jd|j dd|jSt"j%t'|d}||St)t|j*j-d}||St/j0|drStAd |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.)! isinstancerr3rAnpdtyperFrpyarrow pyarrow_dtypetypes is_decimal precisionscale is_timestampraunitPYARROW_CTYPESgetstrgetattrnameupperr is_np_dtype datetime_datarJformatrpd BooleanDtyper:NotImplementedError)rdpapa_type format_strr`s rNdtype_to_arrow_c_fmtr|hs%)*   "((3- !!! E: &%% 88  w '))*!GMM?; ; XX " "7 + 0F Q('**6 6#''G d;  ! ejj&6&6&8$?J uc"%%e,Q/2 $$++zb+II E? +$$++uzz!}+RR E2?? +  LM rMcRt|jtjsy|jj }t |jdk(ry|s td|j}tj||j|j|jS)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`.)rdrqindex) rbrdrvrarray _pa_arraylenchunks RuntimeErrorcombine_chunksSeriesrqr)series allow_copy chunked_arrayarrs rN maybe_rechunkrs fllBMM 2LL**M =   A%  !   & & (C 99S 6;;fll SSrM)rdrreturnro)rz pd.Seriesrr rzpd.Series | None)r8 __future__rtypingnumpyrc pandas._libsrpandas.core.dtypes.dtypesrrrpandasrv TYPE_CHECKINGpandas._typingrrmr3rPr|rrLrMrNrsR#   '  C C S c   c   c  C S S S S c cC  c!"#$=DD  0fTrM