zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-06-14 00:14:51 +02:00
Code Issues Projects Releases Packages Wiki Activity

PR suggestions

Browse source 
Change variable names and coments
This commit is contained in:
zandercymatics 2023-11-22 08:37:48 -07:00
parent 4b625d32de
commit 8535d783e8
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 13 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/registrar/management/commands/load_organization_data.py
View file

@ -58,26 +58,26 @@ class Command(BaseCommand):
"""Process the objects in TransitionDomain."""
# Parse JSON file
options = self.load_json_settings(options, migration_json_filename)
args = TransitionDomainArguments(**options)
org_args = TransitionDomainArguments(**options)
# Will sys.exit() when prompt is "n"
TerminalHelper.prompt_for_execution(
system_exit_on_terminate=True,
info_to_inspect=f"""
==Master data file==
domain_additional_filename: {args.domain_additional_filename}
domain_additional_filename: {org_args.domain_additional_filename}
==Organization data==
organization_adhoc_filename: {args.organization_adhoc_filename}
organization_adhoc_filename: {org_args.organization_adhoc_filename}
==Containing directory==
directory: {args.directory}
directory: {org_args.directory}
""",
prompt_title="Do you wish to load organization data for TransitionDomains?",
)
load = OrganizationDataLoader(args)
transition_domains = load.update_organization_data_for_all()
org_load_helper = OrganizationDataLoader(org_args)
transition_domains = org_load_helper.update_organization_data_for_all()
# Reprompt the user to reinspect before updating DomainInformation
# Will sys.exit() when prompt is "n"
@ -85,13 +85,13 @@ class Command(BaseCommand):
system_exit_on_terminate=True,
info_to_inspect=f"""
==Master data file==
domain_additional_filename: {args.domain_additional_filename}
domain_additional_filename: {org_args.domain_additional_filename}
==Organization name information==
organization_adhoc_filename: {args.organization_adhoc_filename}
organization_adhoc_filename: {org_args.organization_adhoc_filename}
==Containing directory==
directory: {args.directory}
directory: {org_args.directory}
==Proposed Changes==
Number of DomainInformation objects to (potentially) change: {len(transition_domains)}
@ -105,10 +105,10 @@ class Command(BaseCommand):
"Preparing to load organization data onto DomainInformation tables..."
f"{TerminalColors.ENDC}"
)
self.prepare_update_domain_information(transition_domains, args.debug)
self.prepare_update_domain_information(transition_domains, org_args.debug)
logger.info(f"{TerminalColors.MAGENTA}" f"Beginning mass DomainInformation update..." f"{TerminalColors.ENDC}")
self.bulk_update_domain_information(args.debug)
self.bulk_update_domain_information(org_args.debug)
def load_json_settings(self, options, migration_json_filename):
"""Parses options from the given JSON file."""

2
src/registrar/management/commands/utility/extra_transition_domain_helper.py
View file

@ -800,6 +800,8 @@ class OrganizationDataLoader:
return self.tds_to_update
def prepare_transition_domains(self, transition_domains):
"""Pares org data for each transition domain,
then appends it to the tds_to_update list"""
for item in transition_domains:
updated = self.parse_org_data(item.domain_name, item)
self.tds_to_update.append(updated)