mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-16 01:27:03 +02:00
Add formfield
This commit is contained in:
parent
daf640ded3
commit
7ea5503c91
2 changed files with 8 additions and 0 deletions
|
@ -660,6 +660,13 @@ class DomainApplicationAdmin(ListHeaderAdmin):
|
|||
kwargs["queryset"] = models.Website.objects.all().order_by("website") # Sort websites
|
||||
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
||||
|
||||
def formfield_for_foreignkey(self, db_field, request, **kwargs):
|
||||
# Removes invalid investigator options from the investigator dropdown
|
||||
if db_field.name == "investigator":
|
||||
kwargs["queryset"] = User.objects.filter(is_staff=True)
|
||||
return db_field.formfield(**kwargs)
|
||||
return super().formfield_for_foreignkey(db_field, request, **kwargs)
|
||||
|
||||
def get_queryset(self, request):
|
||||
"""Queryset reimplementation to order the table alphabetically"""
|
||||
query_set = super().get_queryset(request)
|
||||
|
|
|
@ -5,6 +5,7 @@ import logging
|
|||
|
||||
from django.apps import apps
|
||||
from django.db import models
|
||||
from django.forms import ValidationError
|
||||
from django_fsm import FSMField, transition # type: ignore
|
||||
from registrar.models.domain import Domain
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue