|
4 | 4 | from django.contrib import admin
|
5 | 5 | from django.contrib.admin import helpers
|
6 | 6 | from django.contrib.admin.utils import unquote, quote
|
| 7 | +from django.contrib.admin.views.main import ChangeList |
7 | 8 | from django.contrib.auth import get_permission_codename
|
8 | 9 | from django.core.exceptions import PermissionDenied
|
9 | 10 | from django.shortcuts import get_object_or_404, render
|
@@ -286,40 +287,49 @@ def enforce_history_permissions(self):
|
286 | 287 | )
|
287 | 288 |
|
288 | 289 |
|
289 |
| -class SimpleHistoryWithDeletedAdmin(SimpleHistoryAdmin): |
290 |
| - class SimpleHistoryShowDeletedFilter(admin.SimpleListFilter): |
| 290 | +class SimpleHistoryChangeList(ChangeList): |
| 291 | + def apply_select_related(self, qs): |
| 292 | + # Our qs is different if we use the history, so the normal select_related |
| 293 | + # won't work and results in an empty QuerySet result. |
| 294 | + history = self.params.get("entries", None) == "deleted_only" |
| 295 | + if history: |
| 296 | + return qs |
| 297 | + return super().apply_select_related(qs) |
| 298 | + |
| 299 | + def url_for_result(self, result) -> str: |
| 300 | + history = self.params.get("entries", None) == "deleted_only" |
| 301 | + route_type = "history" if history else "change" |
| 302 | + route = f"{self.opts.app_label}_{self.opts.model_name}_{route_type}" |
| 303 | + pk = getattr(result, self.pk_attname) |
| 304 | + return reverse( |
| 305 | + f"admin:{route}", |
| 306 | + args=(quote(pk),), |
| 307 | + current_app=self.model_admin.admin_site.name, |
| 308 | + ) |
| 309 | + |
| 310 | + |
| 311 | +class SimpleHistoryShowDeletedFilter(admin.SimpleListFilter): |
291 | 312 | title = "Entries"
|
292 | 313 | parameter_name = "entries"
|
293 | 314 |
|
294 | 315 | def lookups(self, request, model_admin):
|
295 |
| - return ( |
296 |
| - ("deleted_only", "Only Deleted"), |
297 |
| - ) |
| 316 | + return ( |
| 317 | + ("deleted_only", "Only Deleted"), |
| 318 | + ) |
298 | 319 |
|
299 | 320 | def queryset(self, request, queryset):
|
300 |
| - if self.value(): |
301 |
| - return queryset.model.history.filter(history_type='-').distinct() |
302 |
| - return queryset |
| 321 | + if self.value(): |
| 322 | + return queryset.model.history.filter(history_type='-').distinct() |
| 323 | + return queryset |
303 | 324 |
|
304 |
| - def get_changelist(self, request, **kwargs): |
305 |
| - def url_from_result_maker(history=False): |
306 |
| - def custom_url_for_result(self, result): |
307 |
| - pk = getattr(result, self.pk_attname) |
308 |
| - route_type = 'history' if history else 'change' |
309 |
| - route = f'{self.opts.app_label}_{self.opts.model_name}_{route_type}' |
310 |
| - return reverse(f'admin:{route}', |
311 |
| - args=(quote(pk),), |
312 |
| - current_app=self.model_admin.admin_site.name) |
313 |
| - return custom_url_for_result |
314 |
| - |
315 |
| - changelist = super().get_changelist(request, **kwargs) |
316 |
| - if request.GET.get('entries', None) == 'deleted_only': |
317 |
| - changelist.url_for_result = url_from_result_maker(history=True) |
318 |
| - else: |
319 |
| - changelist.url_for_result = url_from_result_maker(history=False) |
320 |
| - return changelist |
321 |
| - |
322 |
| - def get_list_filter(self, request): |
323 |
| - return [self.SimpleHistoryShowDeletedFilter] + [ |
324 |
| - f for f in super().get_list_filter(request) |
325 |
| - ] |
| 325 | + |
| 326 | +class SimpleHistoryWithDeletedAdmin(SimpleHistoryAdmin): |
| 327 | + def get_changelist(self, request, **kwargs): |
| 328 | + return SimpleHistoryChangeList |
| 329 | + |
| 330 | + def get_list_filter(self, request): |
| 331 | + # Doing it here will add it to every inherited class. Alternatively, |
| 332 | + # add SimpleHistoryShowDeletedFilter to the list_filter and remove the below. |
| 333 | + return [SimpleHistoryShowDeletedFilter] + [ |
| 334 | + f for f in super().get_list_filter(request) |
| 335 | + ] |
0 commit comments