66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
from auditlog.mixins import AuditlogHistoryAdminMixin
|
|
from django.contrib import admin, messages
|
|
from django.db import transaction
|
|
from django.db.models.deletion import ProtectedError
|
|
from import_export.admin import ImportExportModelAdmin
|
|
from unfold.admin import ModelAdmin as UnfoldModelAdmin
|
|
|
|
from core.admins.utils import SoftDeleteListFilter
|
|
|
|
|
|
class BaseAdmin(AuditlogHistoryAdminMixin, ImportExportModelAdmin, UnfoldModelAdmin):
|
|
show_auditlog_history_link = True
|
|
actions = ["hard_delete_selected", "restore_selected"]
|
|
list_filter = (SoftDeleteListFilter,)
|
|
|
|
def get_queryset(self, request):
|
|
return self.model.all_objects.all()
|
|
|
|
@admin.action(description="Hard delete selected (permanent)")
|
|
def hard_delete_selected(self, request, queryset):
|
|
count = queryset.count()
|
|
try:
|
|
with transaction.atomic():
|
|
queryset.hard_delete()
|
|
self.message_user(
|
|
request,
|
|
f"{count} record(s) permanently deleted.",
|
|
level=messages.SUCCESS,
|
|
)
|
|
except ProtectedError:
|
|
self.message_user(
|
|
request,
|
|
"Cannot hard delete because related protected objects exist.",
|
|
level=messages.ERROR,
|
|
)
|
|
except Exception as e:
|
|
self.message_user(request, str(e), level=messages.ERROR)
|
|
|
|
@admin.action(description="Restore selected (undo soft delete)")
|
|
def restore_selected(self, request, queryset):
|
|
restored = 0
|
|
for obj in queryset:
|
|
if getattr(obj, "is_deleted", False):
|
|
obj.restore()
|
|
restored += 1
|
|
self.message_user(
|
|
request,
|
|
f"{restored} record(s) restored.",
|
|
level=messages.SUCCESS,
|
|
)
|
|
|
|
def get_actions(self, request):
|
|
actions = super().get_actions(request)
|
|
|
|
if not request.user.is_superuser:
|
|
actions.pop("hard_delete_selected", None)
|
|
|
|
is_deleted_filter = request.GET.get("is_deleted")
|
|
should_show_restore_actions = is_deleted_filter == "1"
|
|
|
|
if not should_show_restore_actions:
|
|
actions.pop("restore_selected", None)
|
|
actions.pop("hard_delete_selected", None)
|
|
|
|
return actions
|