Merge pull request #72 from edly-io/feat/add-partner-based-filtering

Add partner based filtering in elasticsearch apis
edly-io · Nov 2, 2023 · db9118c · db9118c
2 parents c72b66f + fe457d8
commit db9118c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/course_discovery/apps/discovery_dataloader_app/api/v1/views/dataloader_api.py b/course_discovery/apps/discovery_dataloader_app/api/v1/views/dataloader_api.py
@@ -100,5 +100,4 @@ class DataLoaderCourseRunSearchViewSet(CourseRunSearchViewSet):
         'number': 'number',
         'org': {'field': 'org.raw', 'lookups': [LOOKUP_FILTER_TERM, LOOKUP_FILTER_TERMS]},
         'category': {'field': 'subjects.raw', 'lookups': [LOOKUP_FILTER_TERM, LOOKUP_FILTER_TERMS, LOOKUP_QUERY_IN]},
-
     }
diff --git a/course_discovery/apps/edx_elasticsearch_dsl_extensions/viewsets.py b/course_discovery/apps/edx_elasticsearch_dsl_extensions/viewsets.py
@@ -122,3 +122,8 @@ def filter_facet_queryset(self, queryset):
             queryset = queryset.filter('term', published=True).exclude('term', hidden=True)
 
         return queryset
+
+    def get_queryset(self):
+        queryset = super().get_queryset()
+        queryset = queryset.filter('term', partner=self.request.site.partner.short_code)
+        return queryset