khaganat-web/pages/views.py

from django.http import HttpResponseRedirect, Http404
from django.utils.translation import get_language
from django.views import generic
from django.urls import reverse
from nsfw import views as nsfw
from .models import Page, PageContent


def index(request):
    default_page = Page.objects.filter(default=True).first()
    return HttpResponseRedirect(reverse("page", args=(default_page.slug,)))


class PageView(generic.DetailView):
    model = PageContent
    context_object_name = "page"
    template_name = "pages/page.html"

    def dispatch(self, request, *args, **kwargs):
        if self.get_object().page.is_nsfw:
            if not nsfw.is_nsfw_allowed(request):
                return nsfw.redirect(request)
            else:
                nsfw.alert(request)
        return super().dispatch(request, *args, **kwargs)

    def get_object(self):
        obj = PageContent.objects.filter(
            language=get_language(), page__slug=self.kwargs["slug"]
        ).first()
        if obj is None:
            raise Http404("Page not found: %s" % self.kwargs["slug"])
        return obj