zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-27 03:23:51 +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 12:42:07 -07:00
parent e58d68b392
commit 5e697c36a1
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -87,7 +87,7 @@ class Command(BaseCommand):
self.failed_portfolios, self.failed_portfolios,
self.skipped_portfolios, self.skipped_portfolios,
debug=False, debug=False,
skipped_header="----- SOME PORTFOLIOS WERE SKIPPED -----", skipped_header="----- SOME PORTFOLIOS WERENT CREATED -----",
display_as_str=True, display_as_str=True,
) )
@ -258,7 +258,7 @@ class Command(BaseCommand):
message = f"Added portfolio '{portfolio}' to {len(domain_requests)} domain requests." message = f"Added portfolio '{portfolio}' to {len(domain_requests)} domain requests."
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)
return domain_requests return list(domain_requests)
def handle_portfolio_domains(self, portfolio: Portfolio, federal_agency: FederalAgency): def handle_portfolio_domains(self, portfolio: Portfolio, federal_agency: FederalAgency):
""" """
@ -286,7 +286,7 @@ class Command(BaseCommand):
message = f"Added portfolio '{portfolio}' to {len(domain_infos)} domains." message = f"Added portfolio '{portfolio}' to {len(domain_infos)} domains."
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)
return domain_infos return list(domain_infos)
def post_process_suborganization_fields(self, suborganizations, domains, requests): def post_process_suborganization_fields(self, suborganizations, domains, requests):
"""Post-process suborganization fields by pulling data from related domains and requests. """Post-process suborganization fields by pulling data from related domains and requests.
@ -342,6 +342,6 @@ class Command(BaseCommand):
if suborg: if suborg:
suborg.state_territory = state_territory suborg.state_territory = state_territory
logger.info(f"{suborg}: city: {suborg.city}, state: {suborg.state}") logger.info(f"{suborg}: city: {suborg.city}, state: {suborg.state_territory}")
return Suborganization.objects.bulk_update(suborganizations, ["city", "state_territory"]) return Suborganization.objects.bulk_update(suborganizations, ["city", "state_territory"])