moviewyrm/bookwyrm/views/admin/announcements.py

116 lines
3.8 KiB
Python
Raw Normal View History

2021-05-19 21:55:01 +00:00
""" make announcements """
from django.contrib.auth.decorators import login_required, permission_required
from django.core.paginator import Paginator
2021-05-20 19:08:12 +00:00
from django.shortcuts import get_object_or_404, redirect
2021-05-19 21:55:01 +00:00
from django.template.response import TemplateResponse
from django.utils.decorators import method_decorator
from django.views import View
from bookwyrm import forms, models
from bookwyrm.settings import PAGE_LENGTH
2021-05-19 21:55:01 +00:00
# pylint: disable=no-self-use
2021-05-19 21:55:01 +00:00
@method_decorator(login_required, name="dispatch")
@method_decorator(
permission_required("bookwyrm.edit_instance_settings", raise_exception=True),
name="dispatch",
)
class Announcements(View):
"""tell everyone"""
def get(self, request):
"""view and create announcements"""
announcements = models.Announcement.objects
sort = request.GET.get("sort", "-created_date")
sort_fields = [
"created_date",
"preview",
"start_date",
"end_date",
"active",
]
2021-09-27 17:21:46 +00:00
# pylint: disable=consider-using-f-string
if sort in sort_fields + ["-{:s}".format(f) for f in sort_fields]:
announcements = announcements.order_by(sort)
2021-05-19 21:55:01 +00:00
data = {
"announcements": Paginator(announcements, PAGE_LENGTH).get_page(
request.GET.get("page")
),
2021-05-19 21:55:01 +00:00
"form": forms.AnnouncementForm(),
"sort": sort,
2021-05-19 21:55:01 +00:00
}
2021-09-28 19:39:52 +00:00
return TemplateResponse(
request, "settings/announcements/announcements.html", data
)
2021-05-19 21:55:01 +00:00
@method_decorator(login_required, name="dispatch")
@method_decorator(
permission_required("bookwyrm.edit_instance_settings", raise_exception=True),
name="dispatch",
)
class Announcement(View):
2021-05-20 19:08:12 +00:00
"""edit an announcement"""
def get(self, request, announcement_id):
"""view announcement"""
announcement = get_object_or_404(models.Announcement, id=announcement_id)
data = {
"announcement": announcement,
}
2021-09-28 19:39:52 +00:00
return TemplateResponse(
request, "settings/announcements/announcement.html", data
)
2022-02-11 19:38:13 +00:00
@method_decorator(login_required, name="dispatch")
@method_decorator(
permission_required("bookwyrm.edit_instance_settings", raise_exception=True),
name="dispatch",
)
class EditAnnouncement(View):
"""Create of edit an announcement"""
2022-02-12 18:34:16 +00:00
2022-02-11 19:38:13 +00:00
def get(self, request, announcement_id=None):
"""announcement forms"""
announcement = None
if announcement_id:
announcement = get_object_or_404(models.Announcement, id=announcement_id)
data = {
2021-05-20 15:56:17 +00:00
"announcement": announcement,
2022-02-11 19:38:13 +00:00
"form": forms.AnnouncementForm(instance=announcement),
}
2021-09-28 19:39:52 +00:00
return TemplateResponse(
2022-02-11 19:38:13 +00:00
request, "settings/announcements/edit_announcement.html", data
2021-09-28 19:39:52 +00:00
)
2021-05-20 19:08:12 +00:00
2022-02-11 19:38:13 +00:00
def post(self, request, announcement_id=None):
"""edit announcement"""
announcement = None
if announcement_id:
announcement = get_object_or_404(models.Announcement, id=announcement_id)
form = forms.AnnouncementForm(request.POST, instance=announcement)
if not form.is_valid():
data = {
"announcement": announcement,
"form": form,
}
return TemplateResponse(
request, "settings/announcements/edit_announcement.html", data
)
announcement = form.save()
return redirect("settings-announcements", announcement.id)
2021-05-20 19:08:12 +00:00
@login_required
@permission_required("bookwyrm.edit_instance_settings", raise_exception=True)
def delete_announcement(_, announcement_id):
"""delete announcement"""
announcement = get_object_or_404(models.Announcement, id=announcement_id)
announcement.delete()
return redirect("settings-announcements")