Further cleanup

This commit is contained in:
zandercymatics 2024-08-13 14:10:50 -06:00
parent 9733fde689
commit 5f0b342986
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 15 additions and 14 deletions

View file

@ -2839,10 +2839,11 @@ class PortfolioAdmin(ListHeaderAdmin):
change_form_template = "django/admin/portfolio_change_form.html"
fieldsets = [
(None, {"fields": ["portfolio_type", "organization_name", "creator", "created_at", "notes"]}),
("Portfolio members", {
"classes": ("collapse", "closed"),
"fields": ["administrators", "members"]}
),
# TODO - uncomment in #2521
# ("Portfolio members", {
# "classes": ("collapse", "closed"),
# "fields": ["administrators", "members"]}
# ),
("Portfolio domains", {
"classes": ("collapse", "closed"),
"fields": ["domains", "domain_requests"]}
@ -2882,16 +2883,8 @@ class PortfolioAdmin(ListHeaderAdmin):
"portfolio_type",
]
# TODO - this returns None when empty rather than - for some reason
def portfolio_type(self, obj: models.Portfolio):
"""Returns a concat of organization type and federal agency,
seperated by a dash (-)"""
org_choices = DomainRequest.OrganizationChoices
org_type = org_choices.get_org_label(obj.organization_type)
if obj.organization_type == org_choices.FEDERAL and obj.federal_agency:
return " - ".join([org_type, obj.federal_agency.agency])
else:
return org_type
return obj.portfolio_type if obj.portfolio_type else "-"
portfolio_type.short_description = "Portfolio type"
@ -2953,7 +2946,7 @@ class PortfolioAdmin(ListHeaderAdmin):
if item_display_value:
change_url = reverse(f"admin:registrar_{model_name}_change", args=[item.pk])
links.append(f'<a href="{change_url}">{escape(item_display_value)}</a>')
return format_html(seperator.join(links))
return format_html(seperator.join(links)) if links else "-"
def change_view(self, request, object_id, form_url="", extra_context=None):
"""Add related suborganizations and domain groups"""

View file

@ -121,6 +121,14 @@ class Portfolio(TimeStampedModel):
def __str__(self) -> str:
return str(self.organization_name)
@property
def portfolio_type(self):
org_type = self.OrganizationChoices.get_org_label(self.organization_type)
if self.organization_type == self.OrganizationChoices.FEDERAL and self.federal_agency:
return " - ".join([org_type, self.federal_agency.agency])
else:
return org_type
# == Getters for domains == #
def get_domains(self):
"""Returns all DomainInformations associated with this portfolio"""