add __ne__ to Dimension so you can filter = Dimension('dim') != val

druid-io · Mar 9, 2016 · 5290823 · 5290823
1 parent 56efc78
commit 5290823
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pydruid/utils/filters.py b/pydruid/utils/filters.py
@@ -73,6 +73,9 @@ def __init__(self, dim):
     def __eq__(self, other):
         return Filter(dimension=self.dimension, value=other)
 
+    def __ne__(self, other):
+        return ~Filter(dimension=self.dimension, value=other)
+
 class JavaScript:
     def __init__(self, dim):
         self.dimension = dim

diff --git a/tests/utils/test_filters.py b/tests/utils/test_filters.py
@@ -13,6 +13,13 @@ def test_dimension(self):
         expected = {'type': 'selector', 'dimension': 'dim', 'value': 'val'}
         assert actual == expected
 
+    def test_ne_dimension(self):
+        d = filters.Dimension('dim')
+        actual = filters.Filter.build_filter(d != 'val')
+        expected = {'field': {'dimension': 'dim', 'type': 'selector', 'value': 'val'},
+                    'type': 'not'}
+        assert actual == expected
+
 
 class TestFilter: