Flowfile 0.3.5__py3-none-any.whl → 0.3.7__py3-none-any.whl

flowfile_frame/cloud_storage/secret_manager.py

@@ -0,0 +1,73 @@
+from typing import List
+
+from flowfile_core.schemas.cloud_storage_schemas import FullCloudStorageConnection, FullCloudStorageConnectionInterface
+from flowfile_core.flowfile.database_connection_manager.db_connections import (store_cloud_connection,
+                                                                               get_all_cloud_connections_interface,
+                                                                               delete_cloud_connection)
+from flowfile_core.database.connection import get_db_context
+from flowfile_core.auth.jwt import  get_current_user_sync, create_access_token
+from asyncio import run
+
+
+def get_current_user_id() -> int | None:
+    access_token = create_access_token(data={"sub": "local_user"})
+    with get_db_context() as db:
+        current_user_id = get_current_user_sync(
+            access_token,
+            db
+        ).id
+    return current_user_id
+
+
+def create_cloud_storage_connection(connection: FullCloudStorageConnection) -> None:
+    """
+    Create a cloud storage connection using the provided connection details.
+
+    Args:
+        connection (FullCloudStorageConnection): The connection details for cloud storage.
+
+    Returns:
+        None
+    """
+    access_token = create_access_token(data={"sub": "local_user"})
+
+    with get_db_context() as db:
+        current_user_id = get_current_user_sync(
+            access_token,
+            db
+        ).id
+        store_cloud_connection(
+            db,
+            connection,
+            current_user_id
+        )
+
+
+def create_cloud_storage_connection_if_not_exists(connection: FullCloudStorageConnection) -> None:
+    """
+    Create a cloud storage connection if it does not already exist.
+
+    Args:
+        connection (FullCloudStorageConnection): The connection details for cloud storage.
+
+    Returns:
+        None
+    """
+    all_connections = get_all_available_cloud_storage_connections()
+    if not any(conn.connection_name == connection.connection_name for conn in all_connections):
+        create_cloud_storage_connection(connection)
+
+
+def get_all_available_cloud_storage_connections() -> List[FullCloudStorageConnectionInterface]:
+    with get_db_context() as db:
+        all_connections = get_all_cloud_connections_interface(
+            db,
+            get_current_user_id()
+        )
+    return all_connections
+
+
+def del_cloud_storage_connection(connection_name: str) -> None:
+    with get_db_context() as db:
+        user_id = get_current_user_id()
+        delete_cloud_connection(db, connection_name, user_id)

flowfile_frame/expr.py

@@ -20,6 +20,13 @@ if TYPE_CHECKING:
     from flowfile_frame.selectors import Selector
     ExprType = TypeVar('ExprType', bound='Expr')
     ColumnType = "Column"  # Use string literal instead of direct class reference
+    from polars._typing import (
+        Ambiguous,
+        IntoExpr,
+        IntoExprColumn,
+        PolarsDataType,
+        PolarsTemporalType,
+        TimeUnit)
 
 ExprOrStr = Union['Expr', str]
 ExprOrStrList = List[ExprOrStr]
@@ -110,6 +117,11 @@ class StringMethods:
         res_expr = self.expr.to_uppercase() if self.expr is not None else None
         return self._create_next_expr(method_name="to_uppercase", result_expr=res_expr, is_complex=True)
 
+    def slice(self, offset: int | IntoExprColumn, length: int | IntoExprColumn | None = None) -> Expr:
+        res_expr = self.expr.slice(offset=offset, length=length)
+        return self._create_next_expr(method_name="slice", result_expr=res_expr, is_complex=True,
+                                      offset=offset, length=length)
+
     def to_lowercase(self):
         res_expr = self.expr.to_lowercase() if self.expr is not None else None
         return self._create_next_expr(method_name="to_lowercase", result_expr=res_expr, is_complex=True)
@@ -138,13 +150,28 @@ class StringMethods:
                     strict: bool = True,
                     exact: bool = True,
                     cache: bool = True,
-                    ambiguous: Literal["earliest", "latest", "raise", "null"] | Expr = "raise",):
+                    ambiguous: Literal["earliest", "latest", "raise", "null"] | Expr = "raise",) -> 'Expr':
         res_expr = self.expr.to_datetime(format, time_unit=time_unit, time_zone=time_zone, strict=strict,
                                          exact=exact, cache=cache, ambiguous=ambiguous)
         return self._create_next_expr(method_name="to_datetime", result_expr=res_expr, is_complex=True,
                                       format=format, time_unit=time_unit, time_zone=time_zone, strict=strict,
                                       exact=exact, cache=cache, ambiguous=ambiguous)
 
+    def strptime(self,
+                 dtype: PolarsTemporalType,
+                 format: str | None = None,
+                 *,
+                 strict: bool = True,
+                 exact: bool = True,
+                 cache: bool = True,
+                 ambiguous: Literal["earliest", "latest", "raise", "null"] | Expr = "raise",) -> 'Expr':
+        res_expr = self.expr.strptime(dtype, format, strict=strict, exact=exact, cache=cache, ambiguous=ambiguous)
+        return self._create_next_expr(method_name="strptime", dtype=dtype, result_expr=res_expr, is_complex=True,
+                                      format=format, strict=strict,
+                                      exact=exact, cache=cache, ambiguous=ambiguous)
+
+
+
     def __getattr__(self, name):
         if self.expr is None or not hasattr(self.expr, name):
             if self.expr is None:

flowfile_frame/expr.pyi

@@ -56,6 +56,12 @@ class StringMethods:
 
     def starts_with(self, prefix) -> Any: ...
 
+    def strptime(self, dtype: PolarsTemporalType, format: str | None=None, strict: bool=True, exact: bool=True, cache: bool=True, ambiguous: Literal['earliest', 'latest', 'raise', 'null'] | Expr='raise') -> 'Expr': ...
+
+    def to_date(self, format: str, strict: bool=True, exact: bool=True, cache: bool=True) -> Any: ...
+
+    def to_datetime(self, format: str | None=None, time_unit: Literal['ns', 'us', 'ms'] | None=None, time_zone: str | None=None, strict: bool=True, exact: bool=True, cache: bool=True, ambiguous: Literal['earliest', 'latest', 'raise', 'null'] | Expr='raise') -> 'Expr': ...
+
     def to_lowercase(self, ) -> Any: ...
 
     def to_titlecase(self, ) -> Any: ...
@@ -297,7 +303,7 @@ class Expr:
     def cum_sum(self, reverse: bool=False) -> Expr: ...
 
     # Run an expression over a sliding window that increases `1` slot every iteration.
-    def cumulative_eval(self, expr: Expr, min_samples: int=1, parallel: bool=False) -> Expr: ...
+    def cumulative_eval(self, expr: Expr, min_samples: int=1) -> Expr: ...
 
     # Bin continuous values into discrete categories.
     def cut(self, breaks: Sequence[float], labels: Sequence[str] | None=None, left_closed: bool=False, include_breaks: bool=False) -> Expr: ...
@@ -309,7 +315,7 @@ class Expr:
     def deserialize(self, source: str | Path | IOBase | bytes, format: SerializationFormat='binary') -> Expr: ...
 
     # Calculate the first discrete difference between shifted items.
-    def diff(self, n: int=1, null_behavior: NullBehavior='ignore') -> Expr: ...
+    def diff(self, n: int | IntoExpr=1, null_behavior: NullBehavior='ignore') -> Expr: ...
 
     # Compute the dot/inner product between two Expressions.
     def dot(self, other: Expr | str) -> Expr: ...
@@ -414,7 +420,7 @@ class Expr:
     # Print the value that this expression evaluates to and pass on the value.
     def inspect(self, fmt: str='{}') -> Expr: ...
 
-    # Fill null values using interpolation.
+    # Interpolate intermediate values.
     def interpolate(self, method: InterpolationMethod='linear') -> Expr: ...
 
     # Fill null values using interpolation based on another column.
@@ -483,7 +489,7 @@ class Expr:
     def lt(self, other: Any) -> Expr: ...
 
     # Apply a custom python function to a whole Series or sequence of Series.
-    def map_batches(self, function: Callable[[Series], Series | Any], return_dtype: PolarsDataType | None=None, agg_list: bool=False, is_elementwise: bool=False, returns_scalar: bool=False) -> Expr: ...
+    def map_batches(self, function: Callable[[Series], Series | Any], return_dtype: PolarsDataType | pl.DataTypeExpr | None=None, agg_list: bool=False, is_elementwise: bool=False, returns_scalar: bool=False) -> Expr: ...
 
     # Map a custom/user-defined function (UDF) to each element of a column.
     def map_elements(self, function: Callable[[Any], Any], return_dtype: PolarsDataType | None=None, skip_nulls: bool=True, pass_name: bool=False, strategy: MapElementsStrategy='thread_local', returns_scalar: bool=False) -> Expr: ...
@@ -556,7 +562,7 @@ class Expr:
     def qcut(self, quantiles: Sequence[float] | int, labels: Sequence[str] | None=None, left_closed: bool=False, allow_duplicates: bool=False, include_breaks: bool=False) -> Expr: ...
 
     # Get quantile value.
-    def quantile(self, quantile: float | Expr, interpolation: RollingInterpolationMethod='nearest') -> Expr: ...
+    def quantile(self, quantile: float | Expr, interpolation: QuantileMethod='nearest') -> Expr: ...
 
     # Convert from degrees to radians.
     def radians(self, ) -> Expr: ...
@@ -580,7 +586,7 @@ class Expr:
     def replace(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | None=None) -> Expr: ...
 
     # Replace all values by different values.
-    def replace_strict(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | None=None) -> Expr: ...
+    def replace_strict(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | pl.DataTypeExpr | None=None) -> Expr: ...
 
     # Reshape this Expr to a flat column or an Array column.
     def reshape(self, dimensions: tuple[int, ...]) -> Expr: ...
@@ -597,6 +603,9 @@ class Expr:
     # Create rolling groups based on a temporal or integer column.
     def rolling(self, index_column: str, period: str | timedelta, offset: str | timedelta | None=None, closed: ClosedInterval='right') -> Expr: ...
 
+    # Compute a rolling kurtosis.
+    def rolling_kurtosis(self, window_size: int, fisher: bool=True, bias: bool=True, min_samples: int | None=None, center: bool=False) -> Expr: ...
+
     # Compute a custom rolling window function.
     def rolling_map(self, function: Callable[[Series], Any], window_size: int, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
@@ -625,13 +634,13 @@ class Expr:
     def rolling_min_by(self, by: IntoExpr, window_size: timedelta | str, min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
 
     # Compute a rolling quantile.
-    def rolling_quantile(self, quantile: float, interpolation: RollingInterpolationMethod='nearest', window_size: int=2, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
+    def rolling_quantile(self, quantile: float, interpolation: QuantileMethod='nearest', window_size: int=2, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
     # Compute a rolling quantile based on another column.
-    def rolling_quantile_by(self, by: IntoExpr, window_size: timedelta | str, quantile: float, interpolation: RollingInterpolationMethod='nearest', min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
+    def rolling_quantile_by(self, by: IntoExpr, window_size: timedelta | str, quantile: float, interpolation: QuantileMethod='nearest', min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
 
     # Compute a rolling skew.
-    def rolling_skew(self, window_size: int, bias: bool=True) -> Expr: ...
+    def rolling_skew(self, window_size: int, bias: bool=True, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
     # Compute a rolling standard deviation.
     def rolling_std(self, window_size: int, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False, ddof: int=1) -> Expr: ...
@@ -652,7 +661,7 @@ class Expr:
     def rolling_var_by(self, by: IntoExpr, window_size: timedelta | str, min_samples: int=1, closed: ClosedInterval='right', ddof: int=1) -> Expr: ...
 
     # Round underlying floating point data by `decimals` digits.
-    def round(self, decimals: int=0) -> Expr: ...
+    def round(self, decimals: int=0, mode: RoundMode='half_to_even') -> Expr: ...
 
     # Round to a number of significant figures.
     def round_sig_figs(self, digits: int) -> Expr: ...
@@ -661,7 +670,7 @@ class Expr:
     def sample(self, n: int | IntoExprColumn | None=None, fraction: float | IntoExprColumn | None=None, with_replacement: bool=False, shuffle: bool=False, seed: int | None=None) -> Expr: ...
 
     # Find indices where elements should be inserted to maintain order.
-    def search_sorted(self, element: IntoExpr | np.ndarray[Any, Any], side: SearchSortedSide='any') -> Expr: ...
+    def search_sorted(self, element: IntoExpr | np.ndarray[Any, Any], side: SearchSortedSide='any', descending: bool=False) -> Expr: ...
 
     # Flags the expression as 'sorted'.
     def set_sorted(self, descending: bool=False) -> Expr: ...
@@ -735,7 +744,7 @@ class Expr:
     # Calculate the upper bound.
     def upper_bound(self, ) -> Expr: ...
 
-    # Count the occurrences of unique values.
+    # Count the occurrence of unique values.
     def value_counts(self, sort: bool=False, parallel: bool=False, name: str | None=None, normalize: bool=False) -> Expr: ...
 
     def var(self, ddof=1) -> Any: ...
@@ -746,59 +755,59 @@ class Expr:
     # Method equivalent of bitwise exclusive-or operator `expr ^ other`.
     def xor(self, other: Any) -> Expr: ...
 
-    def __init_subclass__(self, *args, **kwargs) -> 'Expr': ...
+    def __pos__(self, *args, **kwargs) -> 'Expr': ...
 
-    def _repr_html_(self, *args, **kwargs) -> 'Expr': ...
+    def __rxor__(self, *args, **kwargs) -> 'Expr': ...
 
-    def _from_pyexpr(self, *args, **kwargs) -> 'Expr': ...
+    def __sizeof__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __bool__(self, *args, **kwargs) -> 'Expr': ...
+    def __init_subclass__(self, *args, **kwargs) -> 'Expr': ...
 
-    def _map_batches_wrapper(self, *args, **kwargs) -> 'Expr': ...
+    def __new__(self, *args, **kwargs) -> 'Expr': ...
 
-    def _skip_batch_predicate(self, *args, **kwargs) -> 'Expr': ...
+    def __reduce__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __delattr__(self, *args, **kwargs) -> 'Expr': ...
+    def __setattr__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __reduce_ex__(self, *args, **kwargs) -> 'Expr': ...
+    def __subclasshook__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __new__(self, *args, **kwargs) -> 'Expr': ...
+    def _skip_batch_predicate(self, *args, **kwargs) -> 'Expr': ...
 
-    def __reduce__(self, *args, **kwargs) -> 'Expr': ...
+    def __getattribute__(self, *args, **kwargs) -> 'Expr': ...
 
     def __class__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __sizeof__(self, *args, **kwargs) -> 'Expr': ...
+    def _from_pyexpr(self, *args, **kwargs) -> 'Expr': ...
 
-    def __setstate__(self, *args, **kwargs) -> 'Expr': ...
+    def __bool__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __dir__(self, *args, **kwargs) -> 'Expr': ...
+    def __setstate__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __subclasshook__(self, *args, **kwargs) -> 'Expr': ...
+    def __format__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __pos__(self, *args, **kwargs) -> 'Expr': ...
+    def __getstate__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __rxor__(self, *args, **kwargs) -> 'Expr': ...
+    def __array_ufunc__(self, *args, **kwargs) -> 'Expr': ...
 
     def __xor__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __array_ufunc__(self, *args, **kwargs) -> 'Expr': ...
-
-    def __rand__(self, *args, **kwargs) -> 'Expr': ...
+    def __reduce_ex__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __str__(self, *args, **kwargs) -> 'Expr': ...
+    def _repr_html_(self, *args, **kwargs) -> 'Expr': ...
 
-    def __ror__(self, *args, **kwargs) -> 'Expr': ...
+    def __dir__(self, *args, **kwargs) -> 'Expr': ...
 
     def __abs__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __setattr__(self, *args, **kwargs) -> 'Expr': ...
+    def __ror__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __format__(self, *args, **kwargs) -> 'Expr': ...
+    def _map_batches_wrapper(self, *args, **kwargs) -> 'Expr': ...
 
-    def __getattribute__(self, *args, **kwargs) -> 'Expr': ...
+    def __rand__(self, *args, **kwargs) -> 'Expr': ...
 
-    def __getstate__(self, *args, **kwargs) -> 'Expr': ...
+    def __str__(self, *args, **kwargs) -> 'Expr': ...
+
+    def __delattr__(self, *args, **kwargs) -> 'Expr': ...
 
 
 class Column(Expr):
@@ -1005,7 +1014,7 @@ class Column(Expr):
     def cum_sum(self, reverse: bool=False) -> Expr: ...
 
     # Run an expression over a sliding window that increases `1` slot every iteration.
-    def cumulative_eval(self, expr: Expr, min_samples: int=1, parallel: bool=False) -> Expr: ...
+    def cumulative_eval(self, expr: Expr, min_samples: int=1) -> Expr: ...
 
     # Bin continuous values into discrete categories.
     def cut(self, breaks: Sequence[float], labels: Sequence[str] | None=None, left_closed: bool=False, include_breaks: bool=False) -> Expr: ...
@@ -1017,7 +1026,7 @@ class Column(Expr):
     def deserialize(self, source: str | Path | IOBase | bytes, format: SerializationFormat='binary') -> Expr: ...
 
     # Calculate the first discrete difference between shifted items.
-    def diff(self, n: int=1, null_behavior: NullBehavior='ignore') -> Expr: ...
+    def diff(self, n: int | IntoExpr=1, null_behavior: NullBehavior='ignore') -> Expr: ...
 
     # Compute the dot/inner product between two Expressions.
     def dot(self, other: Expr | str) -> Expr: ...
@@ -1122,7 +1131,7 @@ class Column(Expr):
     # Print the value that this expression evaluates to and pass on the value.
     def inspect(self, fmt: str='{}') -> Expr: ...
 
-    # Fill null values using interpolation.
+    # Interpolate intermediate values.
     def interpolate(self, method: InterpolationMethod='linear') -> Expr: ...
 
     # Fill null values using interpolation based on another column.
@@ -1191,7 +1200,7 @@ class Column(Expr):
     def lt(self, other: Any) -> Expr: ...
 
     # Apply a custom python function to a whole Series or sequence of Series.
-    def map_batches(self, function: Callable[[Series], Series | Any], return_dtype: PolarsDataType | None=None, agg_list: bool=False, is_elementwise: bool=False, returns_scalar: bool=False) -> Expr: ...
+    def map_batches(self, function: Callable[[Series], Series | Any], return_dtype: PolarsDataType | pl.DataTypeExpr | None=None, agg_list: bool=False, is_elementwise: bool=False, returns_scalar: bool=False) -> Expr: ...
 
     # Map a custom/user-defined function (UDF) to each element of a column.
     def map_elements(self, function: Callable[[Any], Any], return_dtype: PolarsDataType | None=None, skip_nulls: bool=True, pass_name: bool=False, strategy: MapElementsStrategy='thread_local', returns_scalar: bool=False) -> Expr: ...
@@ -1264,7 +1273,7 @@ class Column(Expr):
     def qcut(self, quantiles: Sequence[float] | int, labels: Sequence[str] | None=None, left_closed: bool=False, allow_duplicates: bool=False, include_breaks: bool=False) -> Expr: ...
 
     # Get quantile value.
-    def quantile(self, quantile: float | Expr, interpolation: RollingInterpolationMethod='nearest') -> Expr: ...
+    def quantile(self, quantile: float | Expr, interpolation: QuantileMethod='nearest') -> Expr: ...
 
     # Convert from degrees to radians.
     def radians(self, ) -> Expr: ...
@@ -1288,7 +1297,7 @@ class Column(Expr):
     def replace(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | None=None) -> Expr: ...
 
     # Replace all values by different values.
-    def replace_strict(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | None=None) -> Expr: ...
+    def replace_strict(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | pl.DataTypeExpr | None=None) -> Expr: ...
 
     # Reshape this Expr to a flat column or an Array column.
     def reshape(self, dimensions: tuple[int, ...]) -> Expr: ...
@@ -1305,6 +1314,9 @@ class Column(Expr):
     # Create rolling groups based on a temporal or integer column.
     def rolling(self, index_column: str, period: str | timedelta, offset: str | timedelta | None=None, closed: ClosedInterval='right') -> Expr: ...
 
+    # Compute a rolling kurtosis.
+    def rolling_kurtosis(self, window_size: int, fisher: bool=True, bias: bool=True, min_samples: int | None=None, center: bool=False) -> Expr: ...
+
     # Compute a custom rolling window function.
     def rolling_map(self, function: Callable[[Series], Any], window_size: int, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
@@ -1333,13 +1345,13 @@ class Column(Expr):
     def rolling_min_by(self, by: IntoExpr, window_size: timedelta | str, min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
 
     # Compute a rolling quantile.
-    def rolling_quantile(self, quantile: float, interpolation: RollingInterpolationMethod='nearest', window_size: int=2, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
+    def rolling_quantile(self, quantile: float, interpolation: QuantileMethod='nearest', window_size: int=2, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
     # Compute a rolling quantile based on another column.
-    def rolling_quantile_by(self, by: IntoExpr, window_size: timedelta | str, quantile: float, interpolation: RollingInterpolationMethod='nearest', min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
+    def rolling_quantile_by(self, by: IntoExpr, window_size: timedelta | str, quantile: float, interpolation: QuantileMethod='nearest', min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
 
     # Compute a rolling skew.
-    def rolling_skew(self, window_size: int, bias: bool=True) -> Expr: ...
+    def rolling_skew(self, window_size: int, bias: bool=True, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
     # Compute a rolling standard deviation.
     def rolling_std(self, window_size: int, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False, ddof: int=1) -> Expr: ...
@@ -1360,7 +1372,7 @@ class Column(Expr):
     def rolling_var_by(self, by: IntoExpr, window_size: timedelta | str, min_samples: int=1, closed: ClosedInterval='right', ddof: int=1) -> Expr: ...
 
     # Round underlying floating point data by `decimals` digits.
-    def round(self, decimals: int=0) -> Expr: ...
+    def round(self, decimals: int=0, mode: RoundMode='half_to_even') -> Expr: ...
 
     # Round to a number of significant figures.
     def round_sig_figs(self, digits: int) -> Expr: ...
@@ -1369,7 +1381,7 @@ class Column(Expr):
     def sample(self, n: int | IntoExprColumn | None=None, fraction: float | IntoExprColumn | None=None, with_replacement: bool=False, shuffle: bool=False, seed: int | None=None) -> Expr: ...
 
     # Find indices where elements should be inserted to maintain order.
-    def search_sorted(self, element: IntoExpr | np.ndarray[Any, Any], side: SearchSortedSide='any') -> Expr: ...
+    def search_sorted(self, element: IntoExpr | np.ndarray[Any, Any], side: SearchSortedSide='any', descending: bool=False) -> Expr: ...
 
     # Flags the expression as 'sorted'.
     def set_sorted(self, descending: bool=False) -> Expr: ...
@@ -1446,7 +1458,7 @@ class Column(Expr):
     # Calculate the upper bound.
     def upper_bound(self, ) -> Expr: ...
 
-    # Count the occurrences of unique values.
+    # Count the occurrence of unique values.
     def value_counts(self, sort: bool=False, parallel: bool=False, name: str | None=None, normalize: bool=False) -> Expr: ...
 
     def var(self, ddof=1) -> Any: ...
@@ -1666,7 +1678,7 @@ class When(Expr):
     def cum_sum(self, reverse: bool=False) -> Expr: ...
 
     # Run an expression over a sliding window that increases `1` slot every iteration.
-    def cumulative_eval(self, expr: Expr, min_samples: int=1, parallel: bool=False) -> Expr: ...
+    def cumulative_eval(self, expr: Expr, min_samples: int=1) -> Expr: ...
 
     # Bin continuous values into discrete categories.
     def cut(self, breaks: Sequence[float], labels: Sequence[str] | None=None, left_closed: bool=False, include_breaks: bool=False) -> Expr: ...
@@ -1678,7 +1690,7 @@ class When(Expr):
     def deserialize(self, source: str | Path | IOBase | bytes, format: SerializationFormat='binary') -> Expr: ...
 
     # Calculate the first discrete difference between shifted items.
-    def diff(self, n: int=1, null_behavior: NullBehavior='ignore') -> Expr: ...
+    def diff(self, n: int | IntoExpr=1, null_behavior: NullBehavior='ignore') -> Expr: ...
 
     # Compute the dot/inner product between two Expressions.
     def dot(self, other: Expr | str) -> Expr: ...
@@ -1783,7 +1795,7 @@ class When(Expr):
     # Print the value that this expression evaluates to and pass on the value.
     def inspect(self, fmt: str='{}') -> Expr: ...
 
-    # Fill null values using interpolation.
+    # Interpolate intermediate values.
     def interpolate(self, method: InterpolationMethod='linear') -> Expr: ...
 
     # Fill null values using interpolation based on another column.
@@ -1852,7 +1864,7 @@ class When(Expr):
     def lt(self, other: Any) -> Expr: ...
 
     # Apply a custom python function to a whole Series or sequence of Series.
-    def map_batches(self, function: Callable[[Series], Series | Any], return_dtype: PolarsDataType | None=None, agg_list: bool=False, is_elementwise: bool=False, returns_scalar: bool=False) -> Expr: ...
+    def map_batches(self, function: Callable[[Series], Series | Any], return_dtype: PolarsDataType | pl.DataTypeExpr | None=None, agg_list: bool=False, is_elementwise: bool=False, returns_scalar: bool=False) -> Expr: ...
 
     # Map a custom/user-defined function (UDF) to each element of a column.
     def map_elements(self, function: Callable[[Any], Any], return_dtype: PolarsDataType | None=None, skip_nulls: bool=True, pass_name: bool=False, strategy: MapElementsStrategy='thread_local', returns_scalar: bool=False) -> Expr: ...
@@ -1928,7 +1940,7 @@ class When(Expr):
     def qcut(self, quantiles: Sequence[float] | int, labels: Sequence[str] | None=None, left_closed: bool=False, allow_duplicates: bool=False, include_breaks: bool=False) -> Expr: ...
 
     # Get quantile value.
-    def quantile(self, quantile: float | Expr, interpolation: RollingInterpolationMethod='nearest') -> Expr: ...
+    def quantile(self, quantile: float | Expr, interpolation: QuantileMethod='nearest') -> Expr: ...
 
     # Convert from degrees to radians.
     def radians(self, ) -> Expr: ...
@@ -1952,7 +1964,7 @@ class When(Expr):
     def replace(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | None=None) -> Expr: ...
 
     # Replace all values by different values.
-    def replace_strict(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | None=None) -> Expr: ...
+    def replace_strict(self, old: IntoExpr | Sequence[Any] | Mapping[Any, Any], new: IntoExpr | Sequence[Any], default: IntoExpr, return_dtype: PolarsDataType | pl.DataTypeExpr | None=None) -> Expr: ...
 
     # Reshape this Expr to a flat column or an Array column.
     def reshape(self, dimensions: tuple[int, ...]) -> Expr: ...
@@ -1969,6 +1981,9 @@ class When(Expr):
     # Create rolling groups based on a temporal or integer column.
     def rolling(self, index_column: str, period: str | timedelta, offset: str | timedelta | None=None, closed: ClosedInterval='right') -> Expr: ...
 
+    # Compute a rolling kurtosis.
+    def rolling_kurtosis(self, window_size: int, fisher: bool=True, bias: bool=True, min_samples: int | None=None, center: bool=False) -> Expr: ...
+
     # Compute a custom rolling window function.
     def rolling_map(self, function: Callable[[Series], Any], window_size: int, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
@@ -1997,13 +2012,13 @@ class When(Expr):
     def rolling_min_by(self, by: IntoExpr, window_size: timedelta | str, min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
 
     # Compute a rolling quantile.
-    def rolling_quantile(self, quantile: float, interpolation: RollingInterpolationMethod='nearest', window_size: int=2, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
+    def rolling_quantile(self, quantile: float, interpolation: QuantileMethod='nearest', window_size: int=2, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
     # Compute a rolling quantile based on another column.
-    def rolling_quantile_by(self, by: IntoExpr, window_size: timedelta | str, quantile: float, interpolation: RollingInterpolationMethod='nearest', min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
+    def rolling_quantile_by(self, by: IntoExpr, window_size: timedelta | str, quantile: float, interpolation: QuantileMethod='nearest', min_samples: int=1, closed: ClosedInterval='right') -> Expr: ...
 
     # Compute a rolling skew.
-    def rolling_skew(self, window_size: int, bias: bool=True) -> Expr: ...
+    def rolling_skew(self, window_size: int, bias: bool=True, min_samples: int | None=None, center: bool=False) -> Expr: ...
 
     # Compute a rolling standard deviation.
     def rolling_std(self, window_size: int, weights: list[float] | None=None, min_samples: int | None=None, center: bool=False, ddof: int=1) -> Expr: ...
@@ -2024,7 +2039,7 @@ class When(Expr):
     def rolling_var_by(self, by: IntoExpr, window_size: timedelta | str, min_samples: int=1, closed: ClosedInterval='right', ddof: int=1) -> Expr: ...
 
     # Round underlying floating point data by `decimals` digits.
-    def round(self, decimals: int=0) -> Expr: ...
+    def round(self, decimals: int=0, mode: RoundMode='half_to_even') -> Expr: ...
 
     # Round to a number of significant figures.
     def round_sig_figs(self, digits: int) -> Expr: ...
@@ -2033,7 +2048,7 @@ class When(Expr):
     def sample(self, n: int | IntoExprColumn | None=None, fraction: float | IntoExprColumn | None=None, with_replacement: bool=False, shuffle: bool=False, seed: int | None=None) -> Expr: ...
 
     # Find indices where elements should be inserted to maintain order.
-    def search_sorted(self, element: IntoExpr | np.ndarray[Any, Any], side: SearchSortedSide='any') -> Expr: ...
+    def search_sorted(self, element: IntoExpr | np.ndarray[Any, Any], side: SearchSortedSide='any', descending: bool=False) -> Expr: ...
 
     # Flags the expression as 'sorted'.
     def set_sorted(self, descending: bool=False) -> Expr: ...
@@ -2110,7 +2125,7 @@ class When(Expr):
     # Calculate the upper bound.
     def upper_bound(self, ) -> Expr: ...
 
-    # Count the occurrences of unique values.
+    # Count the occurrence of unique values.
     def value_counts(self, sort: bool=False, parallel: bool=False, name: str | None=None, normalize: bool=False) -> Expr: ...
 
     def var(self, ddof=1) -> Any: ...