Add Expr::Case when_then_else support to rex_call_operands function (#…

…388) * Add Expr::Case when_then_else support to rex_call_operands function * Update gitignore and formatting * Update gitignore and formatting
apache · May 26, 2023 · d912db5 · d912db5
1 parent 82b4a95
commit d912db5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -10,6 +10,9 @@ __pycache__/
 *.py[cod]
 *$py.class
 
+# Python dist ignore
+dist
+
 # C extensions
 *.so
 
@@ -24,4 +27,4 @@ apache-rat-*.jar
 .env
 CHANGELOG.md.bak
 
-docs/mdbook/book
+docs/mdbook/book
diff --git a/src/expr.rs b/src/expr.rs
@@ -384,14 +384,21 @@ impl PyExpr {
                 let mut operands: Vec<PyExpr> = Vec::new();
 
                 if let Some(e) = expr {
-                    operands.push(PyExpr::from(*e.clone()));
+                    for (when, then) in when_then_expr {
+                        operands.push(PyExpr::from(Expr::BinaryExpr(BinaryExpr::new(
+                            Box::new(*e.clone()),
+                            Operator::Eq,
+                            Box::new(*when.clone()),
+                        ))));
+                        operands.push(PyExpr::from(*then.clone()));
+                    }
+                } else {
+                    for (when, then) in when_then_expr {
+                        operands.push(PyExpr::from(*when.clone()));
+                        operands.push(PyExpr::from(*then.clone()));
+                    }
                 };
 
-                for (when, then) in when_then_expr {
-                    operands.push(PyExpr::from(*when.clone()));
-                    operands.push(PyExpr::from(*then.clone()));
-                }
-
                 if let Some(e) = else_expr {
                     operands.push(PyExpr::from(*e.clone()));
                 };