mirror of
https://github.com/ae-utbm/sith.git
synced 2025-07-10 03:49:24 +00:00
add PermissionOrAuthorRequiredMixin
This commit is contained in:
@ -68,7 +68,10 @@ class NewsQuerySet(models.QuerySet):
|
||||
"""
|
||||
if user.has_perm("com.view_unmoderated_news"):
|
||||
return self
|
||||
return self.filter(Q(is_moderated=True) | Q(author_id=user.id))
|
||||
q_filter = Q(is_moderated=True)
|
||||
if user.is_authenticated:
|
||||
q_filter |= Q(author_id=user.id)
|
||||
return self.filter(q_filter)
|
||||
|
||||
|
||||
class News(models.Model):
|
||||
@ -149,8 +152,12 @@ class News(models.Model):
|
||||
self.author_id == user.id or user.has_perm("com.change_news")
|
||||
)
|
||||
|
||||
def can_be_viewed_by(self, user):
|
||||
return self.is_moderated or user.has_perm("com.view_unmoderated_news")
|
||||
def can_be_viewed_by(self, user: User):
|
||||
return (
|
||||
self.is_moderated
|
||||
or user.has_perm("com.view_unmoderated_news")
|
||||
or (user.is_authenticated and self.author_id == user.id)
|
||||
)
|
||||
|
||||
|
||||
def news_notification_callback(notif):
|
||||
|
23
com/views.py
23
com/views.py
@ -44,7 +44,11 @@ from club.models import Club, Mailing
|
||||
from com.calendar import IcsCalendar
|
||||
from com.forms import NewsDateForm, NewsForm, PosterForm
|
||||
from com.models import News, NewsDate, Poster, Screen, Sith, Weekmail, WeekmailArticle
|
||||
from core.auth.mixins import CanEditPropMixin, CanViewMixin
|
||||
from core.auth.mixins import (
|
||||
CanEditPropMixin,
|
||||
CanViewMixin,
|
||||
PermissionOrAuthorRequiredMixin,
|
||||
)
|
||||
from core.models import User
|
||||
from core.views.mixins import QuickNotifMixin, TabedViewMixin
|
||||
from core.views.widgets.markdown import MarkdownInput
|
||||
@ -162,24 +166,19 @@ class NewsCreateView(PermissionRequiredMixin, CreateView):
|
||||
return init
|
||||
|
||||
|
||||
class NewsUpdateView(UpdateView):
|
||||
class NewsUpdateView(PermissionOrAuthorRequiredMixin, UpdateView):
|
||||
model = News
|
||||
form_class = NewsForm
|
||||
template_name = "com/news_edit.jinja"
|
||||
pk_url_kwarg = "news_id"
|
||||
|
||||
def dispatch(self, request, *args, **kwargs):
|
||||
if (
|
||||
not request.user.has_perm("com.edit_news")
|
||||
and self.get_object().author != request.user
|
||||
):
|
||||
raise PermissionDenied
|
||||
return super().dispatch(request, *args, **kwargs)
|
||||
permission_required = "com.edit_news"
|
||||
|
||||
def form_valid(self, form):
|
||||
self.object = form.save()
|
||||
IcsCalendar.make_internal()
|
||||
return super().form_valid(form)
|
||||
# Don't call `super().form_valid()`,
|
||||
# because it would trigger a second call to `form.save()`
|
||||
return HttpResponseRedirect(self.get_success_url())
|
||||
|
||||
def get_date_form_kwargs(self) -> dict[str, Any]:
|
||||
"""Get initial data for NewsDateForm"""
|
||||
@ -202,7 +201,7 @@ class NewsUpdateView(UpdateView):
|
||||
}
|
||||
|
||||
|
||||
class NewsDeleteView(PermissionRequiredMixin, DeleteView):
|
||||
class NewsDeleteView(PermissionOrAuthorRequiredMixin, DeleteView):
|
||||
model = News
|
||||
pk_url_kwarg = "news_id"
|
||||
template_name = "core/delete_confirm.jinja"
|
||||
|
Reference in New Issue
Block a user