From 0920af7e77acc6e154fed4ae2c86381d58b9ba88 Mon Sep 17 00:00:00 2001 From: Untone Date: Wed, 29 Nov 2023 12:16:37 +0300 Subject: [PATCH] visibility-filter-fix --- resolvers/reader.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/resolvers/reader.py b/resolvers/reader.py index bb606133..3168e4da 100644 --- a/resolvers/reader.py +++ b/resolvers/reader.py @@ -47,11 +47,14 @@ def apply_filters(q, filters, author_id=None): if filters.get("reacted") and author_id: q.join(Reaction, Reaction.created_by == author_id) - v = filters.get("visibility") - if v == "public": - q = q.filter(Shout.visibility == filters.get("visibility")) - if v == "community": - q = q.filter(Shout.visibility.in_(["public", "community"])) + by_visibility = filters.get("visibility") + if by_visibility: + visibility = { + 'public': ShoutVisibility.PUBLIC, + 'community': ShoutVisibility.COMMUNITY, + 'authors': ShoutVisibility.AUTHORS + } + q = q.filter(Shout.visibility == visibility.get(by_visibility)) if filters.get("layouts"): q = q.filter(Shout.layout.in_(filters.get("layouts")))