Skip to content

Commit f1864e8

Browse files
committed
fixed filter
1 parent 6a70527 commit f1864e8

1 file changed

Lines changed: 7 additions & 8 deletions

File tree

feed/views.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -31,12 +31,11 @@ class FeedList(APIView):
3131
}
3232
)
3333
def get(self, request: Request, *args, **kwargs) -> Response:
34-
models = []
35-
if request.query_params.get("news") != "false":
36-
models.append(News)
37-
if request.query_params.get("vacancies") != "false":
38-
models.append(Vacancy)
39-
if request.query_params.get("projects") != "false":
40-
models.append(Project)
41-
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)
4241
return Response(status=status.HTTP_200_OK, data=collect_feed(models, 3))

0 commit comments

Comments
 (0)