Update create_federal_portfolio.py

This commit is contained in:
zandercymatics 2024-09-06 10:14:06 -06:00
parent 23d471f11c
commit 44e363da04
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -53,13 +53,11 @@ class Command(BaseCommand):
federal_agency = FederalAgency.objects.filter(agency__iexact=agency_name).first() federal_agency = FederalAgency.objects.filter(agency__iexact=agency_name).first()
if not federal_agency: if not federal_agency:
message = ( raise ValueError(
f"Cannot find the federal agency '{agency_name}' in our database. " f"Cannot find the federal agency '{agency_name}' in our database. "
"The value you enter for `agency_name` must be " "The value you enter for `agency_name` must be "
"prepopulated in the FederalAgency table before proceeding." "prepopulated in the FederalAgency table before proceeding."
) )
TerminalHelper.colorful_logger(logger.error, TerminalColors.FAIL, message)
return None
portfolio = self.create_or_modify_portfolio(federal_agency) portfolio = self.create_or_modify_portfolio(federal_agency)
self.create_suborganizations(portfolio, federal_agency) self.create_suborganizations(portfolio, federal_agency)