zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-16 09:37:03 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add formfield

Browse source
This commit is contained in:
zandercymatics 2023-12-14 09:58:33 -07:00
parent daf640ded3
commit 7ea5503c91
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/registrar/admin.py
View file

@ -660,6 +660,13 @@ class DomainApplicationAdmin(ListHeaderAdmin):
kwargs["queryset"] = models.Website.objects.all().order_by("website") # Sort websites kwargs["queryset"] = models.Website.objects.all().order_by("website") # Sort websites
return super().formfield_for_manytomany(db_field, request, **kwargs) 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): def get_queryset(self, request):
"""Queryset reimplementation to order the table alphabetically""" """Queryset reimplementation to order the table alphabetically"""
query_set = super().get_queryset(request) query_set = super().get_queryset(request)

1
src/registrar/models/domain_application.py
View file

@ -5,6 +5,7 @@ import logging
from django.apps import apps from django.apps import apps
from django.db import models from django.db import models
from django.forms import ValidationError
from django_fsm import FSMField, transition # type: ignore from django_fsm import FSMField, transition # type: ignore
from registrar.models.domain import Domain from registrar.models.domain import Domain