zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-12 20:49:41 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update create_federal_portfolio.py

Browse source
This commit is contained in:
zandercymatics 2025-01-10 13:55:25 -07:00
parent 19e4dff7f9
commit ac08b17f94
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 18 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

37
src/registrar/management/commands/create_federal_portfolio.py
View file

@ -102,25 +102,25 @@ class Command(BaseCommand):
for federal_agency in agencies: for federal_agency in agencies:
message = f"Processing federal agency '{federal_agency.agency}'..." message = f"Processing federal agency '{federal_agency.agency}'..."
TerminalHelper.colorful_logger(logger.info, TerminalColors.MAGENTA, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.MAGENTA, message)
# try: try:
portfolio, created = self.create_portfolio(federal_agency) portfolio, created = self.create_portfolio(federal_agency)
suborganizations = self.create_suborganizations(portfolio, federal_agency) suborganizations = self.create_suborganizations(portfolio, federal_agency)
domains = [] domains = []
domain_requests = [] domain_requests = []
if created and parse_domains or both: if created and parse_domains or both:
domains = self.handle_portfolio_domains(portfolio, federal_agency) domains = self.handle_portfolio_domains(portfolio, federal_agency)
if parse_requests or both: if parse_requests or both:
domain_requests = self.handle_portfolio_requests(portfolio, federal_agency) domain_requests = self.handle_portfolio_requests(portfolio, federal_agency)
all_suborganizations.extend(suborganizations) all_suborganizations.extend(suborganizations)
all_domains.extend(domains) all_domains.extend(domains)
all_domain_requests.extend(domain_requests) all_domain_requests.extend(domain_requests)
# except Exception as exec: except Exception as exec:
# self.failed_portfolios.add(federal_agency) self.failed_portfolios.add(federal_agency)
# logger.error(exec) logger.error(exec)
# message = f"Failed to create portfolio '{federal_agency.agency}'" message = f"Failed to create portfolio '{federal_agency.agency}'"
# TerminalHelper.colorful_logger(logger.info, TerminalColors.FAIL, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.FAIL, message)
# Post process steps # Post process steps
# Add suborg info to created or existing suborgs. # Add suborg info to created or existing suborgs.
@ -179,7 +179,6 @@ class Command(BaseCommand):
federal_agency=federal_agency, organization_name__isnull=False federal_agency=federal_agency, organization_name__isnull=False
) )
org_names = set(valid_agencies.values_list("organization_name", flat=True)) org_names = set(valid_agencies.values_list("organization_name", flat=True))
if not org_names: if not org_names:
message = ( message = (
"Could not add any suborganizations." "Could not add any suborganizations."
@ -219,7 +218,7 @@ class Command(BaseCommand):
else: else:
TerminalHelper.colorful_logger(logger.warning, TerminalColors.YELLOW, "No suborganizations added") TerminalHelper.colorful_logger(logger.warning, TerminalColors.YELLOW, "No suborganizations added")
return new_suborgs return new_suborgs if len(new_suborgs) > 0 else []
def handle_portfolio_requests(self, portfolio: Portfolio, federal_agency: FederalAgency): def handle_portfolio_requests(self, portfolio: Portfolio, federal_agency: FederalAgency):
""" """