klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2025-07-10 11:59:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix NewsQuerySet.viewable_by

Browse Source
This commit is contained in:
imperosol
2025-01-10 22:05:37 +01:00
parent 8d73ec797b
commit 9f35f5356b
2 changed files with 53 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
com/models.py
View File

@ -21,7 +21,7 @@
# Place - Suite 330, Boston, MA 02111-1307, USA.
#
#
from typing import Self
from django.conf import settings
from django.core.exceptions import ValidationError
@ -55,13 +55,20 @@ class Sith(models.Model):
class NewsQuerySet(models.QuerySet):
def moderated(self):
def moderated(self) -> Self:
return self.filter(is_moderated=True)
def viewable_by(self, user: User):
def viewable_by(self, user: User) -> Self:
"""Filter news that the given user can view.
If the user has the `com.view_unmoderated_news` permission,
all news are viewable.
Else the viewable news are those that are either moderated
or authored by the user.
"""
if user.has_perm("com.view_unmoderated_news"):
return self
return self.moderated()
return self.filter(Q(is_moderated=True) | Q(author_id=user.id))
class News(models.Model):