Skip to content

Commit 78f4c00

Browse files
authored
Merge pull request #280 from PROCOLLAB-github/flexivanov237-pro-170
filter final
2 parents df51d69 + 712c09f commit 78f4c00

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

feed/views.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,13 @@ class FeedList(APIView):
3131
}
3232
)
3333
def get(self, request: Request, *args, **kwargs) -> Response:
34-
models = [News, Vacancy, Project]
35-
if not request.query_params.get("news"):
36-
models.remove(News)
37-
if not request.query_params.get("vacancy"):
38-
models.remove(Vacancy)
39-
if not request.query_params.get("project"):
40-
models.remove(Project)
34+
models = []
35+
filter = request.query_params.get("type")
36+
if "news" in filter:
37+
models.append(News)
38+
if "project" in filter:
39+
models.append(Project)
40+
if "vacancy" in filter:
41+
models.append(Vacancy)
42+
4143
return Response(status=status.HTTP_200_OK, data=collect_feed(models, 3))

0 commit comments

Comments
 (0)