done
This commit is contained in:
44
lib/python3.11/site-packages/narwhals/_duckdb/series.py
Normal file
44
lib/python3.11/site-packages/narwhals/_duckdb/series.py
Normal file
@ -0,0 +1,44 @@
|
||||
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
|
Reference in New Issue
Block a user