|
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