a Of@sdZddlmZddlZddlZddlmZddlm Z m Z m Z ddl Z ejrZddlmZddd d d d d d ddddddddddddddddddd d!d"d#d$ZGd%d&d&ZGd'd(d(Zd)d*d+d,d-Zd.d/d0d1d2d3ZdS)4zE Utility functions and objects for implementing the interchange API. ) annotationsN)lib) ArrowDtypeCategoricalDtypeDatetimeTZDtype)DtypeObjnbCSILcilefguUzZttsZttmZttuZttntdDtdmztss:ztsm:ztsu:ztsn:ZtDsZtDmZtDuZtDn)nullboolZuint8Zuint16Zuint32Zuint64Zint8Zint16Zint32Zint64Z halffloatfloatdoublestringZ large_stringbinaryz time32[s]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 rr rr sr rr rr rrrrrrrzts{resolution}:{tz}ztt{resolution}N)__name__ __module__ __qualname____doc__ZNULLBOOLZINT8ZUINT8ZINT16ZUINT16ZINT32ZUINT32INT64ZUINT64ZFLOAT16ZFLOAT32ZFLOAT64STRINGZ LARGE_STRINGZDATE32ZDATE64 TIMESTAMPZTIMEr)r)V/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/core/interchange/utils.pyr=s(rc@s eZdZdZdZdZdZdZdS) Endiannessz.Enum indicating the byte-order of a data-type.<>=|N)r!r"r#r$ZLITTLEZBIGZNATIVEZNAr)r)r)r*r+_s r+rstr)dtypereturncCs>t|trtjS|tdkr$tjSt|trddl}|j }|j |r\d|j d|j S|j |r|jdurd|jdd|jStt|d}|dur|Stt|jd}|dur|St|drt|dd}tjj|d d St|trtjj|jd|jd St|tjr*tjSt d |d dS) 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.)! isinstancerrr&npr1r'rZpyarrowZ pyarrow_dtypetypesZ is_decimal precisionscaleZ is_timestampr:unitPYARROW_CTYPESgetr0getattrnameupperrZ is_np_dtypeZ datetime_datar(formatrpdZ BooleanDtyper%NotImplementedError)r1paZpa_type format_strr9r)r)r*dtype_to_arrow_c_fmths6      rKz pd.Seriesrzpd.Series | None)series allow_copyr2cCsXt|jtjsdS|jj}t|jdkr,dS|s8td| }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`.)r1rDindex) r;r1rGrarrayZ _pa_arraylenchunks RuntimeErrorZcombine_chunksZSeriesrDrO)rLrMZ chunked_arrayZarrr)r)r* maybe_rechunks rT)r$ __future__rtypingnumpyr<Z pandas._libsrZpandas.core.dtypes.dtypesrrrZpandasrG TYPE_CHECKINGZpandas._typingrrArr+rKrTr)r)r)r*sV   "" 3