linted

Signed-off-by: CocoByte <nicolle.leclair@gmail.com>

src/registrar/management/commands/load_transition_domain.py

@@ -30,7 +30,7 @@ class termColors:
     BackgroundLightYellow = "\033[103m"
 
 
-def query_yes_no(question: str, default="yes") -> dict[str,bool]:
+def query_yes_no(question: str, default="yes") -> bool:
     """Ask a yes/no question via raw_input() and return their answer.
 
     "question" is a string that is presented to the user.
@@ -81,16 +81,14 @@ class Command(BaseCommand):
         for testing purposes, but USE WITH CAUTION
         """
         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(
             "contacts_filename",
             help="Data file with contact information",
         )
         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")
@@ -107,7 +105,9 @@ class Command(BaseCommand):
             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
         or --limitParse are in use"""
         if debug_on:
@@ -129,7 +129,9 @@ class Command(BaseCommand):
                 """
             )
 
-    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, str]:
         """Creates a mapping of domain name -> status"""
         domain_status_dictionary = defaultdict(str)
         logger.info("Reading domain statuses data file %s", domain_statuses_filename)
@@ -141,9 +143,11 @@ class Command(BaseCommand):
         logger.info("Loaded statuses for %d domains", len(domain_status_dictionary))
         return domain_status_dictionary
 
-    def get_user_emails_dict(self, contacts_filename: str, sep) -> defaultdict(str):
+    def get_user_emails_dict(self,
+                             contacts_filename: str,
+                             sep) -> defaultdict[str, str]:
         """Creates mapping of userId -> emails"""
-        user_emails_dictionary = defaultdict(list)
+        user_emails_dictionary = defaultdict(str)
         logger.info("Reading domain-contacts data file %s", contacts_filename)
         with open(contacts_filename, "r") as contacts_file:
             for row in csv.reader(contacts_file, delimiter=sep):
@@ -153,7 +157,8 @@ class Command(BaseCommand):
         logger.info("Loaded emails for %d users", len(user_emails_dictionary))
         return user_emails_dictionary
 
-    def get_mapped_status(self, status_to_map: str) -> str:
+    def get_mapped_status(self,
+                          status_to_map: str):
         """
         Given a verisign domain status, return a corresponding
         status defined for our domains.
@@ -169,13 +174,14 @@ class Command(BaseCommand):
             "created": TransitionDomain.StatusChoices.READY,
             "ok": TransitionDomain.StatusChoices.READY,
         }
-        return status_maps.get(status_to_map)
+        mapped_status = status_maps.get(status_to_map)
+        return mapped_status
 
     def print_summary_duplications(
         self,
         duplicate_domain_user_combos: 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
         data anomalies in the imported Verisign data.  Currently, we check for:
@@ -217,9 +223,8 @@ class Command(BaseCommand):
                     {termColors.ENDC}"""
             )
 
-    def print_summary_status_findings(self, 
-                                      domains_without_status: list[str], 
-                                      outlier_statuses: list[str]
+    def print_summary_status_findings(
+        self, domains_without_status: list[str], outlier_statuses: list[str]
     ):
         """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:
@@ -264,7 +269,6 @@ class Command(BaseCommand):
                 {termColors.ENDC}"""
             )
 
-
     def handle(  # noqa: C901
         self,
         domain_contacts_filename,
@@ -425,8 +429,8 @@ class Command(BaseCommand):
                                 logger.info(
                                     f"{termColors.OKCYAN}"
                                     f"Updating entry: {existing_entry}"
-                                    f"Status: {existing_entry.status} > {new_entry_status}"
-                                    f"Email Sent: {existing_entry.email_sent} > {new_entry_emailSent}"
+                                    f"Status: {existing_entry.status} > {new_entry_status}" # noqa
+                                    f"Email Sent: {existing_entry.email_sent} > {new_entry_emailSent}" # noqa
                                     f"{termColors.ENDC}"
                                 )