20 lines
576 B
Python
20 lines
576 B
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from narwhals._compliant import LazyExprNamespace
|
|
from narwhals._compliant.any_namespace import StructNamespace
|
|
from narwhals._duckdb.utils import F, lit
|
|
|
|
if TYPE_CHECKING:
|
|
from narwhals._duckdb.expr import DuckDBExpr
|
|
|
|
|
|
class DuckDBExprStructNamespace(
|
|
LazyExprNamespace["DuckDBExpr"], StructNamespace["DuckDBExpr"]
|
|
):
|
|
def field(self, name: str) -> DuckDBExpr:
|
|
return self.compliant._with_elementwise(
|
|
lambda expr: F("struct_extract", expr, lit(name))
|
|
).alias(name)
|