zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-31 01:33:56 +02:00
Code Issues Projects Releases Packages Wiki Activity

(no changes, but git detects a change)

Browse source 
Signed-off-by: CocoByte <nicolle.leclair@gmail.com>
This commit is contained in:
CocoByte 2023-10-08 20:00:16 -06:00
parent c8a8d0c34a
commit c971d838fe
No known key found for this signature in database
GPG key ID: BBFAA2526384C97F
1 changed files with 28 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/registrar/management/commands/load_transition_domain.py
View file

@ -30,7 +30,7 @@ class termColors:
BackgroundLightYellow = "\033[103m" BackgroundLightYellow = "\033[103m"
def query_yes_no(question: str, default="yes") -> dict[str, bool]: def query_yes_no(question: str, default="yes") -> dict[str,bool]:
"""Ask a yes/no question via raw_input() and return their answer. """Ask a yes/no question via raw_input() and return their answer.
"question" is a string that is presented to the user. "question" is a string that is presented to the user.
@ -81,14 +81,16 @@ class Command(BaseCommand):
for testing purposes, but USE WITH CAUTION for testing purposes, but USE WITH CAUTION
""" """
parser.add_argument( parser.add_argument(
"domain_contacts_filename", help="Data file with domain contact information" "domain_contacts_filename",
help="Data file with domain contact information"
) )
parser.add_argument( parser.add_argument(
"contacts_filename", "contacts_filename",
help="Data file with contact information", help="Data file with contact information",
) )
parser.add_argument( parser.add_argument(
"domain_statuses_filename", help="Data file with domain status information" "domain_statuses_filename",
help="Data file with domain status information"
) )
parser.add_argument("--sep", default="|", help="Delimiter character") parser.add_argument("--sep", default="|", help="Delimiter character")
@ -105,9 +107,7 @@ class Command(BaseCommand):
action=argparse.BooleanOptionalAction, action=argparse.BooleanOptionalAction,
) )
def print_debug_mode_statements( def print_debug_mode_statements(self, debug_on: bool, debug_max_entries_to_parse: int):
self, debug_on: bool, debug_max_entries_to_parse: int
):
"""Prints additional terminal statements to indicate if --debug """Prints additional terminal statements to indicate if --debug
or --limitParse are in use""" or --limitParse are in use"""
if debug_on: if debug_on:
@ -129,9 +129,7 @@ class Command(BaseCommand):
""" """
) )
def get_domain_user_dict( def get_domain_user_dict(self, domain_statuses_filename: str, sep: str) -> defaultdict(str):
self, domain_statuses_filename: str, sep: str
) -> defaultdict(str):
"""Creates a mapping of domain name -> status""" """Creates a mapping of domain name -> status"""
domain_status_dictionary = defaultdict(str) domain_status_dictionary = defaultdict(str)
logger.info("Reading domain statuses data file %s", domain_statuses_filename) logger.info("Reading domain statuses data file %s", domain_statuses_filename)
@ -165,9 +163,9 @@ class Command(BaseCommand):
and any ok state should map to Ready. and any ok state should map to Ready.
""" """
status_maps = { status_maps = {
"hold": TransitionDomain.StatusChoices.HOLD, "hold": TransitionDomain.StatusChoices.ON_HOLD,
"serverhold": TransitionDomain.StatusChoices.HOLD, "serverhold": TransitionDomain.StatusChoices.ON_HOLD,
"clienthold": TransitionDomain.StatusChoices.HOLD, "clienthold": TransitionDomain.StatusChoices.ON_HOLD,
"created": TransitionDomain.StatusChoices.READY, "created": TransitionDomain.StatusChoices.READY,
"ok": TransitionDomain.StatusChoices.READY, "ok": TransitionDomain.StatusChoices.READY,
} }
@ -177,7 +175,7 @@ class Command(BaseCommand):
self, self,
duplicate_domain_user_combos: list[TransitionDomain], duplicate_domain_user_combos: list[TransitionDomain],
duplicate_domains: list[TransitionDomain], duplicate_domains: list[TransitionDomain],
users_without_email: list[str], users_without_email: list[str]
): ):
"""Called at the end of the script execution to print out a summary of """Called at the end of the script execution to print out a summary of
data anomalies in the imported Verisign data. Currently, we check for: data anomalies in the imported Verisign data. Currently, we check for:
@ -219,8 +217,9 @@ class Command(BaseCommand):
{termColors.ENDC}""" {termColors.ENDC}"""
) )
def print_summary_status_findings( def print_summary_status_findings(self,
self, domains_without_status: list[str], outlier_statuses: list[str] domains_without_status: list[str],
outlier_statuses: list[str]
): ):
"""Called at the end of the script execution to print out a summary of """Called at the end of the script execution to print out a summary of
status anomolies in the imported Verisign data. Currently, we check for: status anomolies in the imported Verisign data. Currently, we check for:
@ -265,6 +264,7 @@ class Command(BaseCommand):
{termColors.ENDC}""" {termColors.ENDC}"""
) )
def handle( # noqa: C901 def handle( # noqa: C901
self, self,
domain_contacts_filename, domain_contacts_filename,
@ -463,7 +463,7 @@ class Command(BaseCommand):
# DEBUG: # DEBUG:
if debug_on or debug_max_entries_to_parse > 0: if debug_on or debug_max_entries_to_parse > 0:
if ( if (
total_rows_parsed > debug_max_entries_to_parse total_rows_parsed >= debug_max_entries_to_parse
and debug_max_entries_to_parse != 0 and debug_max_entries_to_parse != 0
): ):
logger.info( logger.info(