polars_python/expr/
name.rs

1use polars_utils::python_function::PythonObject;
2use pyo3::prelude::*;
3
4use crate::PyExpr;
5
6#[pymethods]
7impl PyExpr {
8    fn name_keep(&self) -> Self {
9        self.inner.clone().name().keep().into()
10    }
11
12    fn name_map(&self, lambda: PyObject) -> Self {
13        self.inner
14            .clone()
15            .name()
16            .map_udf(PythonObject(lambda))
17            .into()
18    }
19
20    fn name_prefix(&self, prefix: &str) -> Self {
21        self.inner.clone().name().prefix(prefix).into()
22    }
23
24    fn name_suffix(&self, suffix: &str) -> Self {
25        self.inner.clone().name().suffix(suffix).into()
26    }
27
28    fn name_to_lowercase(&self) -> Self {
29        self.inner.clone().name().to_lowercase().into()
30    }
31
32    fn name_to_uppercase(&self) -> Self {
33        self.inner.clone().name().to_uppercase().into()
34    }
35
36    fn name_map_fields(&self, name_mapper: PyObject) -> Self {
37        self.inner
38            .clone()
39            .name()
40            .map_fields_udf(PythonObject(name_mapper))
41            .into()
42    }
43
44    fn name_prefix_fields(&self, prefix: &str) -> Self {
45        self.inner.clone().name().prefix_fields(prefix).into()
46    }
47
48    fn name_suffix_fields(&self, suffix: &str) -> Self {
49        self.inner.clone().name().suffix_fields(suffix).into()
50    }
51}