Progress save

Needs some code cleanup
This commit is contained in:
zandercymatics 2023-08-08 14:42:08 -06:00
parent a3661243ac
commit 95927753aa
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 23 additions and 3 deletions

View file

@ -24,6 +24,29 @@ class AuditedAdmin(admin.ModelAdmin):
)
def formfield_for_foreignkey(self, db_field, request, **kwargs):
"""Used to sort dropdown fields alphabetically but can be expanded upon"""
# Determines what we want to sort by, ex: by name
order_by_list = []
if db_field.name == "submitter" or db_field.name == "authorizing_official" or db_field.name == "creator" or db_field.name == "investigator":
order_by_list = ['first_name', 'last_name']
elif db_field.name == "requested_domain":
order_by_list = ['name']
return self.formfield_order_helper(order_by_list, db_field, request, **kwargs)
def formfield_order_helper(self, order_by_list, db_field, request, **kwargs):
"""A helper function to order a dropdown field in Django Admin, takes the fields you want to order by as an array"""
formfield = super(AuditedAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
# Only order if we choose to do so
if order_by_list:
formfield.queryset = formfield.queryset.order_by(*order_by_list)
return formfield
class ListHeaderAdmin(AuditedAdmin):
"""Custom admin to add a descriptive subheader to list views."""

View file

@ -244,6 +244,3 @@ class DomainInformation(TimeStampedModel):
domain_info.domain = domain
domain_info.save()
return domain_info
class Meta:
verbose_name_plural = "Domain Information"