format updates

Signed-off-by: CocoByte <nicolle.leclair@gmail.com>
This commit is contained in:
CocoByte 2023-10-06 02:30:23 -06:00
parent 67b9d2f69a
commit 86acb62699
No known key found for this signature in database
GPG key ID: BBFAA2526384C97F
2 changed files with 26 additions and 25 deletions

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,16 +81,14 @@ 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", "domain_contacts_filename", help="Data file with domain contact information"
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", "domain_statuses_filename", help="Data file with domain status information"
help="Data file with domain status information"
) )
parser.add_argument("--sep", default="|", help="Delimiter character") parser.add_argument("--sep", default="|", help="Delimiter character")
@ -107,7 +105,9 @@ class Command(BaseCommand):
action=argparse.BooleanOptionalAction, action=argparse.BooleanOptionalAction,
) )
def print_debug_mode_statements(self, debug_on: bool, debug_max_entries_to_parse: int): def print_debug_mode_statements(
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:
@ -117,7 +117,7 @@ class Command(BaseCommand):
Detailed print statements activated. Detailed print statements activated.
{termColors.ENDC} {termColors.ENDC}
""" """
) )
if debug_max_entries_to_parse > 0: if debug_max_entries_to_parse > 0:
logger.info( logger.info(
f"""{termColors.OKCYAN} f"""{termColors.OKCYAN}
@ -127,9 +127,11 @@ class Command(BaseCommand):
Detailed print statements activated. Detailed print statements activated.
{termColors.ENDC} {termColors.ENDC}
""" """
) )
def get_domain_user_dict(self, domain_statuses_filename: str, sep: str) -> defaultdict(str): def get_domain_user_dict(
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)
@ -172,12 +174,12 @@ class Command(BaseCommand):
return status_maps.get(status_to_map) return status_maps.get(status_to_map)
def print_summary_duplications( def print_summary_duplications(
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:
- duplicate domains - duplicate domains
- duplicate domain - user pairs - duplicate domain - user pairs
@ -217,11 +219,10 @@ class Command(BaseCommand):
{termColors.ENDC}""" {termColors.ENDC}"""
) )
def print_summary_status_findings(self, def print_summary_status_findings(
domains_without_status: list[str], self, domains_without_status: list[str], outlier_statuses: 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:
- domains without a status - domains without a status
- any statuses not accounted for in our status mappings (see - any statuses not accounted for in our status mappings (see
@ -264,7 +265,6 @@ class Command(BaseCommand):
{termColors.ENDC}""" {termColors.ENDC}"""
) )
def handle( # noqa: C901 def handle( # noqa: C901
self, self,
domain_contacts_filename, domain_contacts_filename,
@ -275,7 +275,7 @@ class Command(BaseCommand):
"""Parse the data files and create TransitionDomains.""" """Parse the data files and create TransitionDomains."""
sep = options.get("sep") sep = options.get("sep")
# If --resetTable was used, prompt user to confirm # If --resetTable was used, prompt user to confirm
# deletion of table data # deletion of table data
if options.get("resetTable"): if options.get("resetTable"):
confirm_reset = query_yes_no( confirm_reset = query_yes_no(
@ -356,7 +356,7 @@ class Command(BaseCommand):
new_entry_status = TransitionDomain.StatusChoices.READY new_entry_status = TransitionDomain.StatusChoices.READY
new_entry_email = "" new_entry_email = ""
new_entry_emailSent = False # set to False by default new_entry_emailSent = False # set to False by default
if new_entry_domainName not in domain_status_dictionary: if new_entry_domainName not in domain_status_dictionary:
# this domain has no status...default to "Create" # this domain has no status...default to "Create"
@ -482,11 +482,11 @@ class Command(BaseCommand):
updated {total_updated_domain_entries} transition domain entries updated {total_updated_domain_entries} transition domain entries
{termColors.ENDC} {termColors.ENDC}
""" """
) )
# Print a summary of findings (duplicate entries, # Print a summary of findings (duplicate entries,
# missing data..etc.) # missing data..etc.)
self.print_summary_duplications( self.print_summary_duplications(
duplicate_domain_user_combos, duplicate_domains, users_without_email duplicate_domain_user_combos, duplicate_domains, users_without_email
) )
self.print_summary_status_findings(domains_without_status, outlier_statuses) self.print_summary_status_findings(domains_without_status, outlier_statuses)

View file

@ -12,7 +12,8 @@ class TransitionDomain(TimeStampedModel):
"""Transition Domain model stores information about the """Transition Domain model stores information about the
state of a domain upon transition between registry state of a domain upon transition between registry
providers""" providers"""
StatusChoices=StatusChoices
StatusChoices = StatusChoices
StatusChoices = StatusChoices StatusChoices = StatusChoices