python: Expr.__str__() (#2559)

pola-rs · Feb 6, 2022 · 3e04dba · 3e04dba
1 parent 194e402
commit 3e04dba
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/py-polars/polars/internals/expr.py b/py-polars/polars/internals/expr.py
@@ -47,6 +47,12 @@ class Expr:
     def __init__(self) -> None:
         self._pyexpr: PyExpr  # pragma: no cover
 
+    def __str__(self) -> str:
+        return self._pyexpr.to_str()
+
+    def _repr_html_(self) -> str:
+        return self._pyexpr.to_str()
+
     @staticmethod
     def _from_pyexpr(pyexpr: "PyExpr") -> "Expr":
         self = Expr.__new__(Expr)

diff --git a/py-polars/src/lazy/dsl.rs b/py-polars/src/lazy/dsl.rs
@@ -72,6 +72,9 @@ impl<'p> PyObjectProtocol<'p> for PyExpr {
 
 #[pymethods]
 impl PyExpr {
+    pub fn to_str(&self) -> String {
+        format!("{:?}", self.inner)
+    }
     pub fn eq(&self, other: PyExpr) -> PyExpr {
         self.clone().inner.eq(other.inner).into()
     }