mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-16 17:47:02 +02:00
Progress save
Needs some code cleanup
This commit is contained in:
parent
a3661243ac
commit
95927753aa
2 changed files with 23 additions and 3 deletions
|
@ -22,6 +22,29 @@ class AuditedAdmin(admin.ModelAdmin):
|
|||
object_id=object_id,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
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):
|
||||
|
|
|
@ -244,6 +244,3 @@ class DomainInformation(TimeStampedModel):
|
|||
domain_info.domain = domain
|
||||
domain_info.save()
|
||||
return domain_info
|
||||
|
||||
class Meta:
|
||||
verbose_name_plural = "Domain Information"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue