Fix header on archive policy admin page

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)