44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from narwhals._duckdb.utils import DeferredTimeZone, native_to_narwhals_dtype
|
|
from narwhals.dependencies import get_duckdb
|
|
|
|
if TYPE_CHECKING:
|
|
from types import ModuleType
|
|
|
|
import duckdb
|
|
from typing_extensions import Never, Self
|
|
|
|
from narwhals._utils import Version
|
|
from narwhals.dtypes import DType
|
|
|
|
|
|
class DuckDBInterchangeSeries:
|
|
def __init__(self, df: duckdb.DuckDBPyRelation, version: Version) -> None:
|
|
self._native_series = df
|
|
self._version = version
|
|
|
|
def __narwhals_series__(self) -> Self:
|
|
return self
|
|
|
|
def __native_namespace__(self) -> ModuleType:
|
|
return get_duckdb() # type: ignore[no-any-return]
|
|
|
|
@property
|
|
def dtype(self) -> DType:
|
|
return native_to_narwhals_dtype(
|
|
self._native_series.types[0],
|
|
self._version,
|
|
DeferredTimeZone(self._native_series),
|
|
)
|
|
|
|
def __getattr__(self, attr: str) -> Never:
|
|
msg = ( # pragma: no cover
|
|
f"Attribute {attr} is not supported for interchange-level 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
|