Adding lucene compliance for filter parsing of a rule.

logprep/filter/lucene_filter.py

@@ -235,8 +235,18 @@ def __init__(self, tree: luqum.tree, special_fields: dict = None):
         for key in self._special_fields_map:
             self._special_fields[key] = special_fields.get(key) if special_fields.get(key) else []
 
+        if not self._special_fields['regex_fields']:
+            self.recognize_regex_and_add_special_fields()
+
         self._last_search_field = None
 
+    def recognize_regex_and_add_special_fields(self):
+        for child in self._tree.children:
+            value = child.children[0].value[1:-1]
+            if value.startswith('/') and value.endswith('/'):
+                self._special_fields['regex_fields'].append(child.name)
+                child.children[0].value = f'"{value[1:-1]}"'
+
     def build_filter(self) -> FilterExpression:
         """Transform luqum tree into FilterExpression
 

tests/unit/filter/test_lucene_filter.py

@@ -455,12 +455,6 @@ def test_create_filter_error(self, testcase, input_str, message):
     def test_creates_lucene_compliance_filter_two_matching_regex_keys_of_two(self):
         lucene_filter = LuceneFilter.create(
             'regex_key_one: "/.*value.*/" AND regex_key_two: "/.*value.*/"',
-            special_fields={"regex_fields": ["regex_key_one", "regex_key_two"]},
-        )
-
-        a = And(
-            RegExFilterExpression(["regex_key_one"], ".*value.*"),
-            RegExFilterExpression(["regex_key_two"], ".*value.*"),
         )
 
         assert lucene_filter == And(
@@ -470,8 +464,7 @@ def test_creates_lucene_compliance_filter_two_matching_regex_keys_of_two(self):
 
     def test_creates_lucene_compliance_filter_with_one_matching_and_one_missmatching_regex_key_of_two(self):
         lucene_filter = LuceneFilter.create(
-            'regex_key_one: ".*value.*" AND key_two: "value"',
-            special_fields={"regex_fields": ["regex_key_one", "i_dont_exist"]},
+            'regex_key_one: "/.*value.*/" AND key_two: "value"',
         )
 
         assert lucene_filter == And(