#334 Refactor Branch admin deleting (#371)

chcemvediet/apps/inforequests/admin.py

@@ -319,10 +319,18 @@ def get_queryset(self, request):
     def get_inforequest(self, obj):
         return obj.inforequest
 
-    def has_delete_permission(self, request, obj=None):
-        if obj is None:
-            return True
-        return not obj.is_main
+    def delete_constraints(self, obj):
+        if obj.is_main:
+            return [format_html(u'{} is main.'.format(admin_obj_format(obj)))]
+
+    def render_delete_form(self, request, context):
+        context[u'delete_constraints'] = self.delete_constraints(context[u'object'])
+        return super(BranchAdmin, self).render_delete_form(request, context)
+
+    def delete_model(self, request, obj):
+        if self.delete_constraints(obj):
+            raise PermissionDenied
+        return super(BranchAdmin, self).delete_model(request, obj)
 
 @admin.register(Action, site=admin.site)
 class ActionAdmin(NoBulkDeleteAdminMixin, DeleteNestedInforequestEmailAdminMixin, admin.ModelAdmin):

chcemvediet/apps/inforequests/templates/admin/inforequests/branch/delete_confirmation.html

@@ -1,5 +1,5 @@
 {% extends "admin/delete_confirmation.html" %}
-{% load amend after from poleno.amend %}
+{% load amend before after set_attributes from poleno.amend %}
 
 {% block content %}
   {% amend %}
@@ -14,5 +14,14 @@
         <ul>{{ inbound|unordered_list }}</ul>
       {% endif %}
     {% endafter %}
+    {% if delete_constraints %}
+      {% before path=".//form" %}
+        <div class="errornote">
+          <p>Delete not allowed.</p>
+          <ul>{{ delete_constraints|unordered_list }}</ul>
+        </div>
+      {% endbefore %}
+      {% set_attributes path=".//form//input[@type='submit']" disabled=False %}
+    {% endif %}
   {% endamend %}
 {% endblock %}