Tags not contains filter (#612)

* Fix for tag not contains * Update test
grai-io · Aug 23, 2023 · d8f7c7c · d8f7c7c
1 parent 643b5d2
commit d8f7c7c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/grai-server/app/lineage/graph_filter.py b/grai-server/app/lineage/graph_filter.py
@@ -39,7 +39,7 @@ def filter_by_filter(filter, query: GraphQuery) -> GraphQuery:
                 if row["operator"] == "contains":
                     query.where(f"'{value}' IN table.tags")
                 elif row["operator"] == "not-contains":
-                    query.where(f"'{value}' NOT IN table.tags")
+                    query.where(f"NOT '{value}' IN table.tags")
 
         elif row["type"] == "ancestor":
             if row["field"] == "tag":

diff --git a/grai-server/app/lineage/tests/test_graph_filter.py b/grai-server/app/lineage/tests/test_graph_filter.py
@@ -301,7 +301,7 @@ def test_table_tag_doesnt_contain():
     filter_by_filter(filter, query)
 
     assert len(query.clause[0].wheres) == 1
-    assert query.clause[0].wheres[0].where == "'tag1' NOT IN table.tags"
+    assert query.clause[0].wheres[0].where == "NOT 'tag1' IN table.tags"
     assert query.parameters == {}