Update create_federal_portfolio.py

This commit is contained in:
zandercymatics 2024-08-29 15:18:14 -06:00
parent 91e62a6cdf
commit f2288673a0
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -193,7 +193,14 @@ class Command(BaseCommand):
Associate portfolio with domain requests for a federal agency.
Updates all relevant domain request records.
"""
domain_requests = DomainInformation.objects.filter(federal_agency=federal_agency)
invalid_states = [
DomainRequest.DomainRequestStatus.STARTED,
DomainRequest.DomainRequestStatus.INELIGIBLE,
DomainRequest.DomainRequestStatus.REJECTED,
]
domain_requests = DomainRequest.objects.filter(
federal_agency=federal_agency
).exclude(status__in=invalid_states)
if not domain_requests.exists():
message = "Portfolios not added to domain requests: no valid records found"
TerminalHelper.colorful_logger(logger.info, TerminalColors.YELLOW, message)