zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-26 04:28:39 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update create_federal_portfolio.py

Browse source
This commit is contained in:
zandercymatics 2024-08-28 14:54:12 -06:00
parent e5b1b5a87a
commit d35acc8a30
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 39 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -77,18 +77,18 @@ class Command(BaseCommand):
existing_portfolio = Portfolio.objects.filter(organization_name=federal_agency.agency) existing_portfolio = Portfolio.objects.filter(organization_name=federal_agency.agency)
if not existing_portfolio.exists(): if not existing_portfolio.exists():
portfolio = Portfolio.objects.create(**portfolio_args) portfolio = Portfolio.objects.create(**portfolio_args)
message = f"Created portfolio '{federal_agency.agency}'" message = f"Created portfolio '{portfolio}'"
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)
return portfolio return portfolio
else: else:
if len(existing_portfolio) > 1: if len(existing_portfolio) > 1:
raise ValueError(f"Could not update portfolio '{federal_agency.agency}': multiple records exist.") raise ValueError(f"Could not update portfolio '{portfolio}': multiple records exist.")
# TODO a dialog to confirm / deny doing this # TODO a dialog to confirm / deny doing this
existing_portfolio.update(**portfolio_args) existing_portfolio.update(**portfolio_args)
message = f"Modified portfolio '{federal_agency.agency}'" message = f"Modified portfolio '{federal_agency.agency}'"
TerminalHelper.colorful_logger(logger.info, TerminalColors.MAGENTA, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.MAGENTA, message)
return existing_portfolio return existing_portfolio.get()
def create_suborganizations(self, portfolio: Portfolio, federal_agency: FederalAgency): def create_suborganizations(self, portfolio: Portfolio, federal_agency: FederalAgency):
non_federal_agency = FederalAgency.objects.get(agency="Non-Federal Agency") non_federal_agency = FederalAgency.objects.get(agency="Non-Federal Agency")
@ -115,20 +115,50 @@ class Command(BaseCommand):
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)
# Add any suborgs that don't presently exist # Add any suborgs that don't presently exist
excluded_org_names = existing_suborgs.values_list("name", flat=True)
suborgs = [] suborgs = []
for name in org_names: for name in org_names:
if name not in existing_suborgs: if name and name not in excluded_org_names:
suborg = Suborganization( if portfolio.organization_name and name.lower() == portfolio.organization_name.lower():
name=name, # If the suborg name is the name that currently exists,
portfolio=portfolio, # thats not a suborg - thats the portfolio itself!
) # In this case, we can use this as an opportunity to update
suborgs.append(suborg) # address information and the like
self._update_portfolio_location_details(portfolio, valid_agencies.get(organization_name=name))
else:
suborg = Suborganization(
name=name,
portfolio=portfolio,
)
suborgs.append(suborg)
Suborganization.objects.bulk_create(suborgs) Suborganization.objects.bulk_create(suborgs)
# TODO - this is inaccurate when the suborg name does not equal the org name - i.e. if the
# record exists already as well
message = f"Added {len(org_names)} suborganizations..." message = f"Added {len(org_names)} suborganizations..."
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message) TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)
# TODO rename
def _update_portfolio_location_details(self, portfolio: Portfolio, domain_info: DomainInformation):
location_props = [
"address_line1",
"address_line2",
"city",
"state_territory",
"zipcode",
"urbanization",
]
for prop_name in location_props:
# Copy the value from the domain info object to the portfolio object
value = getattr(domain_info, prop_name)
setattr(portfolio, prop_name, value)
portfolio.save()
message = f"Updated location details on portfolio '{portfolio}'"
TerminalHelper.colorful_logger(logger.info, TerminalColors.OKGREEN, message)
def handle_portfolio_requests(self, portfolio: Portfolio, federal_agency: FederalAgency): def handle_portfolio_requests(self, portfolio: Portfolio, federal_agency: FederalAgency):
domain_requests = DomainInformation.objects.filter(federal_agency=federal_agency) domain_requests = DomainInformation.objects.filter(federal_agency=federal_agency)
if len(domain_requests) < 1: if len(domain_requests) < 1: