fix: improvements

Author: rustyconover

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "query-farm-sql-manipulation"
-version = "0.1.5"
+version = "0.1.6"
 description = "A Python library for intelligent SQL predicate manipulation using SQLGlot. This library provides tools to safely remove specific predicates from `SQL WHERE` clauses and filter SQL statements based on column availability."
 authors = [
     { name = "Rusty Conover", email = "[email protected]" }

src/query_farm_sql_manipulation/test_transforms.py

@@ -159,6 +159,7 @@ def test_where_clause_extract(sql: str, expected: str) -> None:
     "sql, expected",
     [
         ("""v1 >= v1 + 5 and z = 5""", "z = 5"),
+        ("v1 in ('foobar') and z = 5", "v1 IN ('foobar') AND z = 5"),
         ("""((v1 >= v1 + 5) or t1 = 5) and z = 5""", "(t1 = 5) AND z = 5"),
     ],
 )

src/query_farm_sql_manipulation/transforms.py

@@ -97,7 +97,9 @@ def filter_predicates_with_right_side_column_references(
 
     for predicate in where_clause.find_all(sqlglot.expressions.Predicate):
         assert predicate is not None
-        if predicate.right.find(sqlglot.expressions.Column):
+        if isinstance(predicate, sqlglot.expressions.Binary) and predicate.right.find(
+            sqlglot.expressions.Column
+        ):
             remove_expression_part(predicate)
 
     return statement