This commit is contained in:
David Kennedy 2024-08-23 17:52:57 -04:00
parent 8a9971ac79
commit ef7739dc55
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
3 changed files with 8 additions and 7 deletions

View file

@ -26,7 +26,7 @@ from registrar.views.domain_request import Step
from registrar.views.domain_requests_json import get_domain_requests_json from registrar.views.domain_requests_json import get_domain_requests_json
from registrar.views.utility.api_views import ( from registrar.views.utility.api_views import (
get_senior_official_from_federal_agency_json, get_senior_official_from_federal_agency_json,
get_federal_and_portfolio_types_from_federal_agency_json get_federal_and_portfolio_types_from_federal_agency_json,
) )
from registrar.views.domains_json import get_domains_json from registrar.views.domains_json import get_domains_json
from registrar.views.utility import always_404 from registrar.views.utility import always_404

View file

@ -150,7 +150,7 @@ class Portfolio(TimeStampedModel):
@classmethod @classmethod
def get_federal_type(cls, federal_agency): def get_federal_type(cls, federal_agency):
return federal_agency.federal_type if federal_agency else None return federal_agency.federal_type if federal_agency else None
# == Getters for domains == # # == Getters for domains == #
def get_domains(self): def get_domains(self):
"""Returns all DomainInformations associated with this portfolio""" """Returns all DomainInformations associated with this portfolio"""

View file

@ -37,7 +37,8 @@ def get_senior_official_from_federal_agency_json(request):
return JsonResponse(so_dict) return JsonResponse(so_dict)
else: else:
return JsonResponse({"error": "Senior Official not found"}, status=404) return JsonResponse({"error": "Senior Official not found"}, status=404)
@login_required @login_required
@staff_member_required @staff_member_required
def get_federal_and_portfolio_types_from_federal_agency_json(request): def get_federal_and_portfolio_types_from_federal_agency_json(request):
@ -62,8 +63,8 @@ def get_federal_and_portfolio_types_from_federal_agency_json(request):
federal_type = BranchChoices.get_branch_label(federal_type) if federal_type else "-" federal_type = BranchChoices.get_branch_label(federal_type) if federal_type else "-"
response_data = { response_data = {
'portfolio_type': portfolio_type, "portfolio_type": portfolio_type,
'federal_type': federal_type, "federal_type": federal_type,
} }
return JsonResponse(response_data) return JsonResponse(response_data)