simplify, support custom series extensions too

Author: MarcoGorelli

pandas/core/col.py

@@ -139,6 +139,9 @@ def __rmod__(self, other: Any) -> Expr:
 
     # Everything else
     def __getattr__(self, attr: str, /) -> Callable[..., Expr]:
+        if attr in Series._accessors:
+            return NamespaceExpr(self, attr)
+
         def func(df: DataFrame, *args: Any, **kwargs: Any) -> Any:
             parsed_args = _parse_args(df, *args)
             parsed_kwargs = _parse_kwargs(df, **kwargs)
@@ -170,31 +173,6 @@ def wrapper(*args: Any, **kwargs: Any) -> Expr:
     def __repr__(self) -> str:
         return self._repr_str or "Expr(...)"
 
-    # Namespaces
-    @property
-    def dt(self) -> NamespaceExpr:
-        return NamespaceExpr(self, "dt")
-
-    @property
-    def str(self) -> NamespaceExpr:
-        return NamespaceExpr(self, "str")
-
-    @property
-    def cat(self) -> NamespaceExpr:
-        return NamespaceExpr(self, "cat")
-
-    @property
-    def list(self) -> NamespaceExpr:
-        return NamespaceExpr(self, "list")
-
-    @property
-    def sparse(self) -> NamespaceExpr:
-        return NamespaceExpr(self, "sparse")
-
-    @property
-    def struct(self) -> NamespaceExpr:
-        return NamespaceExpr(self, "struct")
-
 
 class NamespaceExpr:
     def __init__(self, func: Expr, namespace: str) -> None: