0.1.1

src/polars_business_day_tools/polars_business_day_tools/__init__.py

@@ -0,0 +1,25 @@
+import polars as pl
+from polars.type_aliases import IntoExpr
+from polars.utils.udfs import _get_shared_lib_location
+
+lib = _get_shared_lib_location(__file__)
+
+__version__ = "0.1.1"
+
+
+@pl.api.register_expr_namespace("bdt")
+class BusinessDayTools:
+    def __init__(self, expr: pl.Expr):
+        self._expr = expr.cast(pl.Int32)
+
+
+    def advance_by_days(self, n) -> pl.Expr:
+        if not (isinstance(n, int) and n > 0):
+            raise ValueError("only positive integers are currently supported for `n`")
+
+        return self._expr._register_plugin(
+            lib=lib,
+            symbol="advance_by_days",
+            is_elementwise=True,
+            args = [n],
+        )