Browse Source

Try filtering in the post_search loop

Arend-Jan Tetteroo 5 years ago
parent
commit
ed3cd83591
1 changed files with 16 additions and 4 deletions
  1. 16 4
      only_show_green_results.py

+ 16 - 4
only_show_green_results.py

@@ -29,10 +29,21 @@ preference_section = 'privacy'
 #  request: flask request object
 #  request: flask request object
 #  ctx: the whole local context of the pre search hook
 #  ctx: the whole local context of the pre search hook
 def post_search(request, search):
 def post_search(request, search):
-    print search
+    print "post search"
+    results = search.result_container.get_ordered_results()
+    resultlist = enumerate(list(search.result_container._merged_results))
 
 
+    for i, result in resultlist:
+        print i
+        print result
+        if get_green(result):
+            #result = search.result_container._merged_results[i]
+            print('deleting result:', result)
+            if i < len(search.result_container._merged_results):
+                del(search.result_container._merged_results[i])
+    return True
 
 
-def on_result(request, search, result):
+def get_green(result):
     if 'parsed_url' not in result:
     if 'parsed_url' not in result:
         return True
         return True
 
 
@@ -47,6 +58,7 @@ def on_result(request, search, result):
     data = response.json()
     data = response.json()
     #print(data['green'])
     #print(data['green'])
 
 
-    result['green'] = data['green']
+    return data['green']
 
 
-    return True
+def on_result(request, search, result):
+    result['green'] = get_green(result)