From 79e2af8f70ab22b70b612200f5c5fe397c9e866c Mon Sep 17 00:00:00 2001 From: Oskar Persson Date: Tue, 18 Jun 2019 13:34:43 +0200 Subject: [PATCH] Fix header on archive policy admin page --- ESSArch_PP/configuration/admin.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/ESSArch_PP/configuration/admin.py b/ESSArch_PP/configuration/admin.py index 4a6cee541..f4572ed2a 100644 --- a/ESSArch_PP/configuration/admin.py +++ b/ESSArch_PP/configuration/admin.py @@ -23,6 +23,9 @@ """ from django.contrib import admin +from django.db import transaction +from django.utils.decorators import method_decorator +from django.views.decorators.csrf import csrf_protect from nested_inline.admin import NestedModelAdmin @@ -31,6 +34,9 @@ from storage.admin import StorageMethodInline +csrf_protect_m = method_decorator(csrf_protect) + + class ArchivePolicyAdmin(NestedModelAdmin): """ ArchivePolicy @@ -68,5 +74,17 @@ class ArchivePolicyAdmin(NestedModelAdmin): ) inlines = [StorageMethodInline] + @csrf_protect_m + @transaction.atomic + def add_view(self, request, form_url='', extra_context=None): + extra_context = self.admin_site.each_context(request) + return super().add_view(request, form_url=form_url, extra_context=extra_context) + + @csrf_protect_m + @transaction.atomic + def change_view(self, request, object_id, form_url='', extra_context=None): + extra_context = self.admin_site.each_context(request) + return super().change_view(request, object_id, form_url=form_url, extra_context=extra_context) + admin.site.register(ArchivePolicy, ArchivePolicyAdmin)