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,
|
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):
|
class ListHeaderAdmin(AuditedAdmin):
|
||||||
|
|
|
@ -244,6 +244,3 @@ class DomainInformation(TimeStampedModel):
|
||||||
domain_info.domain = domain
|
domain_info.domain = domain
|
||||||
domain_info.save()
|
domain_info.save()
|
||||||
return domain_info
|
return domain_info
|
||||||
|
|
||||||
class Meta:
|
|
||||||
verbose_name_plural = "Domain Information"
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue