38 lines
1.2 KiB
Python
Executable File
38 lines
1.2 KiB
Python
Executable File
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
from typing import Any
|
|
|
|
from narwhals._duckdb.dataframe import map_duckdb_dtype_to_narwhals_dtype
|
|
from narwhals.dependencies import get_duckdb
|
|
|
|
if TYPE_CHECKING:
|
|
from types import ModuleType
|
|
|
|
from narwhals.typing import DTypes
|
|
|
|
|
|
class DuckDBInterchangeSeries:
|
|
def __init__(self, df: Any, dtypes: DTypes) -> None:
|
|
self._native_series = df
|
|
self._dtypes = dtypes
|
|
|
|
def __narwhals_series__(self) -> Any:
|
|
return self
|
|
|
|
def __native_namespace__(self) -> ModuleType:
|
|
return get_duckdb() # type: ignore[no-any-return]
|
|
|
|
def __getattr__(self, attr: str) -> Any:
|
|
if attr == "dtype":
|
|
return map_duckdb_dtype_to_narwhals_dtype(
|
|
self._native_series.types[0], self._dtypes
|
|
)
|
|
msg = ( # pragma: no cover
|
|
f"Attribute {attr} is not supported for metadata-only dataframes.\n\n"
|
|
"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."
|
|
)
|
|
raise NotImplementedError(msg) # pragma: no cover
|