This commit is contained in:
zandercymatics 2023-11-06 16:16:57 -07:00
parent a4fcca23ba
commit 4e22e26065
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 9 additions and 6 deletions

View file

@ -136,9 +136,9 @@ class LoadExtraTransitionDomain:
def __init__(self, options: TransitionDomainArguments):
# Globally stores event logs and organizes them
self.parse_logs = FileTransitionLog()
arguments = options.args_extra_transition_domain()
print(f"options correct? {options.agency_adhoc_filename}")
# Reads and parses migration files
self.parsed_data_container = ExtraTransitionDomain(**arguments)
self.parsed_data_container = ExtraTransitionDomain(options)
self.parsed_data_container.parse_all_files(options.infer_filenames)
def update_transition_domain_models(self):

View file

@ -6,6 +6,11 @@ from registrar.management.commands.utility.epp_data_containers import EnumFilena
@dataclass
class TransitionDomainArguments:
"""Stores arguments for load_transition_domain"""
def __init__(self, **kwargs):
self.kwargs = kwargs
for k, v in kwargs.items():
if hasattr(self, k):
setattr(self, k, v)
# Settings #
directory: Optional[str] = field(default="migrationdata", repr=True)
@ -20,7 +25,8 @@ class TransitionDomainArguments:
authority_adhoc_filename: Optional[str] = field(default=EnumFilenames.AUTHORITY_ADHOC.value[1], repr=True)
## Data files ##
domain_additional_filename: Optional[str] = field(default=None, repr=True)
domain_escrow_filename: Optional[str] = field(default=EnumFilenames.DOMAIN_ESCROW.value[1], repr=True)
domain_additional_filename: Optional[str] = field(default=EnumFilenames.DOMAIN_ADDITIONAL.value[1], repr=True)
domain_contacts_filename: Optional[str] = field(default=None, repr=True)
domain_statuses_filename: Optional[str] = field(default=None, repr=True)

View file

@ -14,9 +14,6 @@ from django.core.management import call_command
from unittest.mock import patch
class TestMigrations(TestCase):
""" """
def setUp(self):
""" """
# self.load_transition_domain_script = "load_transition_domain",