Merge pull request #280 from PROCOLLAB-github/flexivanov237-pro-170

filter final
PROCOLLAB-github · Feb 16, 2024 · 78f4c00 · 78f4c00
2 parents df51d69 + 712c09f
commit 78f4c00
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/feed/views.py b/feed/views.py
@@ -31,11 +31,13 @@ class FeedList(APIView):
         }
     )
     def get(self, request: Request, *args, **kwargs) -> Response:
-        models = [News, Vacancy, Project]
-        if not request.query_params.get("news"):
-            models.remove(News)
-        if not request.query_params.get("vacancy"):
-            models.remove(Vacancy)
-        if not request.query_params.get("project"):
-            models.remove(Project)
+        models = []
+        filter = request.query_params.get("type")
+        if "news" in filter:
+            models.append(News)
+        if "project" in filter:
+            models.append(Project)
+        if "vacancy" in filter:
+            models.append(Vacancy)
+
         return Response(status=status.HTTP_200_OK, data=collect_feed(models, 3))