modernize expression_lib

README.md

@@ -56,23 +56,48 @@ fn pig_latinnify(inputs: &[Series], kwargs: PigLatinKwargs) -> PolarsResult<Seri
 This can then be exposed on the Python side:
 
 ```python
+from __future__ import annotations
+from typing import TYPE_CHECKING
 import polars as pl
-from polars.type_aliases import IntoExpr
-from polars.utils.udfs import _get_shared_lib_location
+from polars.plugins import register_plugin_function
+from expression_lib._utils import LIB
 
-from expression_lib.utils import parse_into_expr
-
-lib = _get_shared_lib_location(__file__)
+if TYPE_CHECKING:
+    from expression_lib._typing import IntoExpr
 
 
 def pig_latinnify(expr: IntoExpr, capitalize: bool = False) -> pl.Expr:
-    expr = parse_into_expr(expr)
-    return expr.register_plugin(
-        lib=lib,
-        symbol="pig_latinnify",
+    return register_plugin_function(
+        plugin_path=LIB,
+        args=[expr],
+        function_name="pig_latinnify",
         is_elementwise=True,
         kwargs={"capitalize": capitalize},
     )
+
+
+def append_args(
+    expr: IntoExpr,
+    float_arg: float,
+    integer_arg: int,
+    string_arg: str,
+    boolean_arg: bool,
+) -> pl.Expr:
+    """
+    This example shows how arguments other than `Series` can be used.
+    """
+    return register_plugin_function(
+        plugin_path=LIB,
+        args=[expr],
+        kwargs={
+            "float_arg": float_arg,
+            "integer_arg": integer_arg,
+            "string_arg": string_arg,
+            "boolean_arg": boolean_arg,
+        },
+        function_name="append_kwargs",
+        is_elementwise=True,
+    )
 ```
 
 Compile/ship and then it is ready to use:

example/derive_expression/expression_lib/expression_lib/date_util.py

@@ -1,15 +1,16 @@
+from __future__ import annotations
 import polars as pl
-from polars.type_aliases import IntoExpr
+from typing import TYPE_CHECKING
 from polars.plugins import register_plugin_function
-from pathlib import Path
+from expression_lib._utils import LIB
 
-from expression_lib.utils import parse_into_expr
+if TYPE_CHECKING:
+    from expression_lib._typing import IntoExpr
 
 
 def is_leap_year(expr: IntoExpr) -> pl.Expr:
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr],
         function_name="is_leap_year",
         is_elementwise=True,
@@ -19,9 +20,10 @@ def is_leap_year(expr: IntoExpr) -> pl.Expr:
 # Note that this already exists in Polars. It is just for explanatory
 # purposes.
 def change_time_zone(expr: IntoExpr, tz: str = "Europe/Amsterdam") -> pl.Expr:
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr],
-        function_name="change_time_zone", is_elementwise=True, kwargs={"tz": tz}
+        function_name="change_time_zone",
+        is_elementwise=True,
+        kwargs={"tz": tz},
     )

example/derive_expression/expression_lib/expression_lib/dist.py

@@ -1,25 +1,25 @@
+from __future__ import annotations
 import polars as pl
-from polars.type_aliases import IntoExpr
+from typing import TYPE_CHECKING
 from polars.plugins import register_plugin_function
-from pathlib import Path
+from expression_lib._utils import LIB
 
+if TYPE_CHECKING:
+    from expression_lib._typing import IntoExpr
 
-from expression_lib.utils import parse_into_expr
 
 def hamming_distance(expr: IntoExpr, other: IntoExpr) -> pl.Expr:
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr, other],
         function_name="hamming_distance",
         is_elementwise=True,
     )
 
 
 def jaccard_similarity(expr: IntoExpr, other: IntoExpr) -> pl.Expr:
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr, other],
         function_name="jaccard_similarity",
         is_elementwise=True,
@@ -32,9 +32,8 @@ def haversine(
     end_lat: IntoExpr,
     end_long: IntoExpr,
 ) -> pl.Expr:
-    start_lat = parse_into_expr(start_lat)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[start_lat, start_long, end_lat, end_long],
         function_name="haversine",
         is_elementwise=True,

example/derive_expression/expression_lib/expression_lib/extension.py

@@ -16,11 +16,12 @@
 - static typing will not recognise your custom namespace. Errors such
   as `"Expr" has no attribute "dist"  [attr-defined]`.
 """
+
 from __future__ import annotations
 
 import polars as pl
 from typing import Any, Callable
-from expression_lib import date_util, dist, language, utils, panic
+from expression_lib import date_util, dist, language, panic
 
 
 @pl.api.register_expr_namespace("language")

example/derive_expression/expression_lib/expression_lib/language.py

@@ -1,16 +1,16 @@
+from __future__ import annotations
+from typing import TYPE_CHECKING
 import polars as pl
-from polars.type_aliases import IntoExpr
 from polars.plugins import register_plugin_function
-from pathlib import Path
-
-from expression_lib.utils import parse_into_expr
+from expression_lib._utils import LIB
 
+if TYPE_CHECKING:
+    from expression_lib._typing import IntoExpr
 
 
 def pig_latinnify(expr: IntoExpr, capitalize: bool = False) -> pl.Expr:
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr],
         function_name="pig_latinnify",
         is_elementwise=True,
@@ -28,9 +28,8 @@ def append_args(
     """
     This example shows how arguments other than `Series` can be used.
     """
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr],
         kwargs={
             "float_arg": float_arg,

example/derive_expression/expression_lib/expression_lib/panic.py

@@ -1,15 +1,16 @@
+from __future__ import annotations
 import polars as pl
-from polars.type_aliases import IntoExpr
+from typing import TYPE_CHECKING
 from polars.plugins import register_plugin_function
+from expression_lib._utils import LIB
 
-from expression_lib.utils import parse_into_expr
-from pathlib import Path
+if TYPE_CHECKING:
+    from expression_lib._typing import IntoExpr
 
 
 def panic(expr: IntoExpr) -> pl.Expr:
-    expr = parse_into_expr(expr)
     return register_plugin_function(
-        plugin_path=Path(__file__).parent,
+        plugin_path=LIB,
         args=[expr],
         function_name="panic",
     )