From fe457d88f821b392349a95bc6dcf3fc210d3a74c Mon Sep 17 00:00:00 2001 From: Hina Khadim Date: Thu, 2 Nov 2023 17:16:14 +0500 Subject: [PATCH] feat: add partner based filtering in discovery elasticsearch apis --- .../discovery_dataloader_app/api/v1/views/dataloader_api.py | 1 - .../apps/edx_elasticsearch_dsl_extensions/viewsets.py | 5 +++++ 2 files changed, 5 insertions(+), 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 index ea77c233f5..3a1cdcdb49 100644 --- 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 index cb8c51e6f6..18e816644f 100644 --- 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