mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-16 01:27:03 +02:00
Update admin.py
This commit is contained in:
parent
c66ac66018
commit
528c77a8b3
1 changed files with 11 additions and 4 deletions
|
@ -548,17 +548,24 @@ class DomainApplicationAdminForm(forms.ModelForm):
|
||||||
class DomainApplicationAdmin(ListHeaderAdmin):
|
class DomainApplicationAdmin(ListHeaderAdmin):
|
||||||
|
|
||||||
"""Custom domain applications admin class."""
|
"""Custom domain applications admin class."""
|
||||||
|
|
||||||
class InvestigatorFilter(admin.SimpleListFilter):
|
class InvestigatorFilter(admin.SimpleListFilter):
|
||||||
title = 'investigator'
|
"""Custom investigator filter that only displays users with the manager role"""
|
||||||
parameter_name = 'investigator'
|
title = "investigator"
|
||||||
|
parameter_name = "investigator"
|
||||||
|
|
||||||
def lookups(self, request, model_admin):
|
def lookups(self, request, model_admin):
|
||||||
valid_user_ids = UserDomainRole.objects.filter(role=UserDomainRole.Roles.MANAGER).values_list('user__id', flat=True)
|
"""Lookup reimplementation, gets users by the MANAGER role.
|
||||||
|
Returns a list of tuples consisting of (user.id, user)
|
||||||
|
"""
|
||||||
|
valid_user_ids = UserDomainRole.objects.filter(role=UserDomainRole.Roles.MANAGER).values_list(
|
||||||
|
"user__id", flat=True
|
||||||
|
)
|
||||||
privileged_users = User.objects.filter(id__in=valid_user_ids)
|
privileged_users = User.objects.filter(id__in=valid_user_ids)
|
||||||
return [(user.id, user) for user in privileged_users]
|
return [(user.id, user) for user in privileged_users]
|
||||||
|
|
||||||
def queryset(self, request, queryset):
|
def queryset(self, request, queryset):
|
||||||
print(f"look here: {self.value()}")
|
"""Custom queryset implementation, filters by investigator"""
|
||||||
return queryset.filter(investigator=self.value())
|
return queryset.filter(investigator=self.value())
|
||||||
|
|
||||||
# Columns
|
# Columns
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue