zhtddlmZddlZddlmZddlmZddlmZddlmZer&ddl Z ddl Z ddl mZddlmZdd lmZdd lmZdd lmZGd d ej,Z ddZGddZGddZy)) annotationsN) TYPE_CHECKING)Any)NoReturn) parse_version)SelfInterchangeSeries)DType) DataFrameLike)Versionc(eZdZdZdZdZdZdZdZdZ y) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICAL]/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/narwhals/_interchange/dataframe.pyrrs% C D E D FHKr!rc|j}|dtjk(rm|ddk(r|jS|ddk(r|j S|ddk(r|j S|ddk(r|j Sd}t||dtjk(rm|ddk(r|jS|ddk(r|jS|ddk(r|jS|ddk(r|jSd}t||dtjk(r=|ddk(r|jS|ddk(r|jSd }t||dtj k(r|j#S|dtj$k(r|j'S|dtj(k(r|j+S|dtj,k(r|j/Sd |}t|) Nrr@ zInvalid bit width for INTzInvalid bit width for UINTzInvalid bit width for FLOATzInvalid dtype, got: )dtypesrrInt64Int32Int16Int8AssertionErrorrUInt64UInt32UInt16UInt8rFloat64Float32rBooleanrStringrDatetimer Categorical)interchange_dtypeversionr(msgs r"'map_interchange_dtype_to_narwhals_dtyper; s^^Fy}}, Q 2 %<<> ! Q 2 %<<> ! Q 2 %<<> ! Q 1 $;;= )S!!y~~- Q 2 %==? " Q 2 %==? " Q 2 %==? " Q 1 $<<> !*S!!y. Q 2 %>># # Q 2 %>># #+S!!y~~-~~y///}}y111  y444!!## !2 3 4C  r!ceZdZddZddZy)WrapInterchangeFramec||_yN_interchange_frame)selfinterchange_frames r"__init__zWrapInterchangeFrame.__init__Os "3r!c|jSr?r@rBs r" __dataframe__z"WrapInterchangeFrame.__dataframe__Rs&&&r!N)rCInterchangeFramereturnNone)rIrH)rrrrDrGr r!r"r=r=Ns 4'r!r=ceZdZddZddZddZddZddZddZddZ e ddZ e dd Z dd Z dd Z dd Zy )rHc<|j|_||_yr?)rGrA_version)rBdfr9s r"rDzInterchangeFrame.__init__Ws"$"2"2"4 r!cN|jt|j|S)Nr9) __class__r=rA)rBr9s r" _with_versionzInterchangeFrame._with_version[s(~~ !8!8 97  r!c|Sr?r rFs r"__narwhals_dataframe__z'InterchangeFrame.__narwhals_dataframe__`s r!cd}t|)NzCannot access native namespace for metadata-only dataframes with unknown backend.If you would like to see this kind of object supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.NotImplementedError)rBr:s r"__native_namespace__z%InterchangeFrame.__native_namespace__cs Y "#&&r!chddlm}||jj||jS)Nrr rP)narwhals._interchange.seriesr rAget_column_by_namerM)rBnamer s r" get_columnzInterchangeFrame.get_columnks,B  # # 6 6t =1.5.0 to be installed, found )pandasrapi interchangefrom_dataframerA __version__rW)rBpdr:s r" to_pandaszInterchangeFrame.to_pandasrsZ   )66%%44T5L5LM M99;8HJ &c* *r!c2ddlm}||jS)Nr)rc)"pyarrow.interchange.from_dataframercrA)rBrcs r"to_arrowzInterchangeFrame.to_arrow~s d5566r!c |jjDcic]<}|t|jj|j|j >c}Scc}wr?)rA column_namesr;r[dtyperM)rB column_names r"schemazInterchangeFrame.schemasc $66CCE   @''::;GMM     sAA!cHt|jjSr?)listrArkrFs r"columnszInterchangeFrame.columnssD++88:;;r!c$d|d}t|)Nz Attribute aG is not supported for metadata-only dataframes. Hint: you probably called `nw.from_native` on an object which isn't fully supported by Narwhals, yet implements `__dataframe__`. If you would like to see this kind of object supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.rV)rBattrr:s r" __getattr__zInterchangeFrame.__getattr__s'B B "#&&r!c|jjt|}t|ds d}t ||j |j |jS)N_dfzExpected interchange object to implement `_df` property to allow for recovering original object. See https://github.com/data-apis/dataframe-api/issues/360.rP)rAselect_columns_by_namerphasattrrWrQrvrM)rBrkframer:s r" simple_selectzInterchangeFrame.simple_selectsZ''>>tL?QRue$M &c* *~~eii~??r!cd}t|)Nz`select`-ing not by name is not supported for interchange-only level. If you would like to see this kind of object better supported in Narwhals, please open a feature request at https://github.com/narwhals-dev/narwhals/issues.rV)rBexprsr:s r"selectzInterchangeFrame.selects  B "#&&r!N)rNr r9r rIrJ)r9r rIr)rIr)rIr)r\strrIr )rIz pd.DataFrame)rIzpa.Table)rIzdict[str, DType])rIz list[str])rsr~rIr)rkr~rIr)r|r~rIr)rrrrDrRrTrXr]rfripropertyrnrqrtrzr}r r!r"rHrHVso  '  +7  <<'@ ' '  'r!rH)r8ztuple[DtypeKind, int, Any, Any]r9r rIr ) __future__renumtypingrrrnarwhals.utilsrr`repyarrowpatyping_extensionsrrZr narwhals.dtypesr narwhals.typingr r IntEnumrr;r=rHr r!r"rss" (&>%-& +6+AH+ +\''['['r!