31 lines
998 B
Python
31 lines
998 B
Python
"""passbook helper views"""
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.shortcuts import render
|
|
from django.views.generic import CreateView
|
|
from guardian.shortcuts import assign_perm
|
|
|
|
|
|
class CreateAssignPermView(CreateView):
|
|
"""Assign permissions to object after creation"""
|
|
|
|
permissions = [
|
|
"%s.view_%s",
|
|
"%s.change_%s",
|
|
"%s.delete_%s",
|
|
]
|
|
|
|
def form_valid(self, form):
|
|
response = super().form_valid(form)
|
|
for permission in self.permissions:
|
|
full_permission = permission % (
|
|
self.object._meta.app_label,
|
|
self.object._meta.model_name,
|
|
)
|
|
assign_perm(full_permission, self.request.user, self.object)
|
|
return response
|
|
|
|
|
|
def bad_request_message(request: HttpRequest, message: str) -> HttpResponse:
|
|
"""Return generic error page with message, with status code set to 400"""
|
|
return render(request, "error/400.html", {"message": message}, status=400)
|