36 lines
1.0 KiB
Python
36 lines
1.0 KiB
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/generic.html",
|
|
{"message": message, "card_title": "Bad Request",},
|
|
status=400,
|
|
)
|