o h@sddlmZddlZddlmZddlmZddlmZddlmZddlm Z erPddl Z ddl Z ddlmZdd lmZdd lmZdd lmZdd lmZGd ddejZdddZGdddZGdddZdS)) annotationsN) TYPE_CHECKING)Any)NoReturn)import_dtypes_module) parse_version)SelfInterchangeSeries)DType) DataFrameLike)Versionc@s(eZdZdZdZdZdZdZdZdZ dS) DtypeKindrN) __name__ __module__ __qualname__INTUINTFLOATBOOLSTRINGDATETIME CATEGORICALrrS/var/www/vscode/kcb/lib/python3.10/site-packages/narwhals/_interchange/dataframe.pyrsrinterchange_dtypetuple[DtypeKind, int, Any, Any]versionr returnr cCst|}|dtjkr9|ddkr|S|ddkr|S|ddkr)|S|ddkr3|Sd}t||dtjkrn|ddkrJ| S|ddkrT| S|ddkr^| S|ddkrh| Sd}t||dtj kr|ddkr|S|ddkr|Sd }t||dtjkr|S|dtjkr|S|dtjkr|S|dtjkr|Sd |}t|) Nrr@ zInvalid bit width for INTzInvalid bit width for UINTzInvalid bit width for FLOATzInvalid dtype, got: )rrrInt64Int32Int16Int8AssertionErrorrUInt64UInt32UInt16UInt8rFloat64Float32rBooleanrStringrDatetimer Categorical)r!r#dtypesmsgrrr 'map_interchange_dtype_to_narwhals_dtype!sP           r:c@s eZdZd ddZd ddZd S) WrapInterchangeFrameinterchange_frameInterchangeFramer$NonecCs ||_dSN_interchange_frame)selfr<rrr __init__Ps zWrapInterchangeFrame.__init__cCs|jSr?r@rBrrr __dataframe__Ssz"WrapInterchangeFrame.__dataframe__N)r<r=r$r>)r$r=)rrrrCrErrrr r;Os r;c@seZdZd,ddZd-d d Zd.d d Zd/ddZd0ddZd1ddZd2ddZ e d3ddZ e d4d d!Z d5d#d$Z d6d&d'Zd7d)d*Zd+S)8r=dfr r#r r$r>cCs||_||_dSr?)rErA_version)rBrFr#rrr rCXs  zInterchangeFrame.__init__rcCs|jt|j|dS)Nr#) __class__r;rA)rBr#rrr _with_version\s zInterchangeFrame._with_versioncCs|Sr?rrDrrr __narwhals_dataframe__asz'InterchangeFrame.__narwhals_dataframe__rcCs d}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)rBr9rrr __native_namespace__dsz%InterchangeFrame.__native_namespace__namestrr cCs"ddlm}||j||jdS)Nrr rH)narwhals._interchange.seriesr rAget_column_by_namerG)rBrOr rrr get_columnls zInterchangeFrame.get_column pd.DataFramecCs8ddl}t|dkr|jj|jSd|j}t|)Nr)rrznConversion to pandas is achieved via interchange protocol which requires pandas>=1.5.0 to be installed, found )pandasrapi interchangefrom_dataframerA __version__rM)rBpdr9rrr to_pandasss zInterchangeFrame.to_pandaspa.TablecCsddlm}||jS)Nr)rY)"pyarrow.interchange.from_dataframerYrA)rBrYrrr to_arrows  zInterchangeFrame.to_arrowdict[str, DType]csfddjDS)Ncs$i|]}|tj|jjqSr)r:rArRdtyperG).0 column_namerDrr s z+InterchangeFrame.schema..)rA column_namesrDrrDr schemas zInterchangeFrame.schema list[str]cCst|jSr?)listrArerDrrr columnsszInterchangeFrame.columnsattrcCsd|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.rL)rBrjr9rrr __getattr__s zInterchangeFrame.__getattr__recGs8|jt|}t|dsd}t||j|j|jdS)N_dfzExpected interchange object to implement `_df` property to allow for recovering original object. See https://github.com/data-apis/dataframe-api/issues/360.rH)rAselect_columns_by_namerhhasattrrMrIrlrG)rBreframer9rrr simple_selects  zInterchangeFrame.simple_selectexprscGs d}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.rL)rBrqr9rrr selectszInterchangeFrame.selectN)rFr r#r r$r>)r#r r$r)r$r)r$r)rOrPr$r )r$rT)r$r])r$r`)r$rg)rjrPr$r)rerPr$r)rqrPr$r)rrrrCrJrKrNrSr\r_propertyrfrirkrprrrrrr r=Ws            r=)r!r"r#r r$r ) __future__renumtypingrrrnarwhals.utilsrrrVr[pyarrowpatyping_extensionsrrQr narwhals.dtypesr narwhals.typingr r IntEnumrr:r;r=rrrr s&            .