From c597af5523f701f8e8c7c5971067a613c6f3f28a Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Fri, 26 Aug 2022 07:20:30 +0100
Subject: [PATCH] Fix restricted source handling

---
 core/lib/opensearch.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/core/lib/opensearch.py b/core/lib/opensearch.py
index 81e7310..9a64feb 100644
--- a/core/lib/opensearch.py
+++ b/core/lib/opensearch.py
@@ -347,7 +347,7 @@ def query_results(
         source = query_params["source"]
 
         if source in settings.OPENSEARCH_SOURCES_RESTRICTED:
-            if not request.user.has_perm("restricted_sources"):
+            if not request.user.has_perm("core.restricted_sources"):
                 message = "Access denied"
                 message_class = "danger"
                 return {"message": message, "class": message_class}
@@ -363,11 +363,11 @@ def query_results(
         sources = [source]
     else:
         sources = settings.OPENSEARCH_MAIN_SOURCES
-        if request.user.has_perm("restricted_sources"):
+        print("SOURCES", sources)
+        if request.user.has_perm("core.restricted_sources"):
+            print("HAS PERM")
             for source_iter in settings.OPENSEARCH_SOURCES_RESTRICTED:
                 sources.append(source_iter)
-        else:
-            sources = settings.OPENSEARCH_MAIN_SOURCES
 
     add_top_tmp = {"bool": {"should": []}}
     for source_iter in sources: