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

Migration change, script work

Browse source
This commit is contained in:
zandercymatics 2023-10-27 08:29:50 -06:00
parent 00f44f2f84
commit 519595ba9b
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
6 changed files with 107 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

57
src/registrar/management/commands/load_extra_transition_domain.py
View file

@ -8,34 +8,18 @@ import os
from typing import List
from enum import Enum
from django.core.management import BaseCommand
from .utility.extra_transition_domain import ExtraTransitionDomain
from registrar.models.transition_domain import TransitionDomain
from .utility.extra_transition_domain import ExtraTransitionDomain
from .utility.epp_data_containers import AgencyAdhoc, DomainAdditionalData, DomainTypeAdhoc, OrganizationAdhoc, EnumFilenames
logger = logging.getLogger(__name__)
class EnumFilenames(Enum):
AGENCY_ADHOC = "agency.adhoc.dotgov.txt"
DOMAIN_ADDITIONAL = "domainadditionaldatalink.adhoc.dotgov.txt"
DOMAIN_ADHOC = "domaintypes.adhoc.dotgov.txt"
ORGANIZATION_ADHOC = "organization.adhoc.dotgov.txt"
class Command(BaseCommand):
help = ""
filenames = EnumFilenames
strip_date_regex = re.compile(r'\d+\.(.+)')
# While the prefix of these files typically includes the date,
# the rest of them following a predefined pattern. Define this here,
# and search for that to infer what is wanted.
filename_pattern_mapping = {
# filename - regex to use when encountered
filenames.AGENCY_ADHOC: strip_date_regex,
filenames.DOMAIN_ADDITIONAL: strip_date_regex,
filenames.DOMAIN_ADHOC: strip_date_regex,
filenames.ORGANIZATION_ADHOC: strip_date_regex
}
def add_arguments(self, parser):
"""Add filename arguments."""
parser.add_argument(
@ -45,22 +29,22 @@ class Command(BaseCommand):
)
parser.add_argument(
"--agency_adhoc_filename",
default=self.filenames.AGENCY_ADHOC,
default=self.filenames.AGENCY_ADHOC[1],
help="Defines the filename for agency adhocs",
)
parser.add_argument(
"--domain_additional_filename",
default=self.filenames.DOMAIN_ADDITIONAL,
default=self.filenames.DOMAIN_ADDITIONAL[1],
help="Defines the filename for additional domain data",
)
parser.add_argument(
"--domain_adhoc_filename",
default=self.filenames.DOMAIN_ADHOC,
default=self.filenames.DOMAIN_ADHOC[1],
help="Defines the filename for domain type adhocs",
)
parser.add_argument(
"--organization_adhoc_filename",
default=self.filenames.ORGANIZATION_ADHOC,
default=self.filenames.ORGANIZATION_ADHOC[1],
help="Defines the filename for domain type adhocs",
)
parser.add_argument("--sep", default="|", help="Delimiter character")
@ -79,5 +63,30 @@ class Command(BaseCommand):
except Exception as err:
logger.error(f"Could not load additional data. Error: {err}")
else:
for transition_domain in TransitionDomain.objects.all():
transition_domain.organization_type
def get_organization_adhoc(self, desired_id):
"""Grabs adhoc information for organizations. Returns an organization
dictionary
returns:
{
"
}
"""
return self.get_object_by_id(self.filenames.ORGANIZATION_ADHOC, desired_id)
def get_domain_adhoc(self, desired_id):
""""""
return self.get_object_by_id(self.filenames.DOMAIN_ADHOC, desired_id)
def get_agency_adhoc(self, desired_id):
""""""
return self.get_object_by_id(self.filenames.AGENCY_ADHOC, desired_id)
def get_object_by_id(self, file_type: EnumFilenames, desired_id):
""""""
desired_type = self.domain_object.csv_data.get(file_type)
obj = desired_type.get(desired_id)
return obj