zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-21 18:25:58 +02:00
Code Issues Projects Releases Packages Wiki Activity

Convert domain lists to counters

Browse source
This commit is contained in:
Erin 2024-01-16 16:23:05 -08:00
parent 39a5875b87
commit 22ace8aee9
No known key found for this signature in database
GPG key ID: 1CAD275313C62460
1 changed files with 16 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

28
src/registrar/management/commands/disclose_security_emails.py
View file

@ -20,13 +20,13 @@ class Command(BaseCommand):
"""Sets global variables for code tidyness""" """Sets global variables for code tidyness"""
super().__init__() super().__init__()
# domains and transition domains that must be disclosed to true # domains and transition domains that must be disclosed to true
self.contacts_saved: list[str] = [] self.contacts_saved_count = 0
# domains with errors, which are not successfully updated to disclose # domains with errors, which are not successfully updated to disclose
self.domains_with_errors: list[str] = [] self.domains_with_errors: list[str] = []
# domains that are successfully disclosed # domains that are successfully disclosed
self.disclosed_domain_contacts: list[str] = [] self.disclosed_domain_contacts_counter = 0
# domains that skip disclose due to having contact registrar@dotgov.gov # domains that skip disclose due to having contact registrar@dotgov.gov
self.skipped_domain_contacts: list[str] = [] self.skipped_domain_contacts_counter = 0
def handle(self, **options): def handle(self, **options):
""" """
@ -45,33 +45,37 @@ class Command(BaseCommand):
# Call security_contact on all domains to trigger saving contact information # Call security_contact on all domains to trigger saving contact information
for domain in domains: for domain in domains:
contact = domain.security_contact contact = domain.security_contact
self.contacts_saved.append(copy.deepcopy(contact)) self.contacts_saved_count++
logger.info("Found %d security contacts.", len(self.contacts_saved)) logger.info("Found %d security contacts.", self.contacts_saved)
# Update EPP contact for domains with a security contact # Update EPP contact for domains with a security contact
for domain in domains: for domain in domains:
try: try:
logger.info("Domain %s security contact: %s", domain.domain_info, domain.security_contact.email) logger.info("Domain %s security contact: %s", domain.domain_name, domain.security_contact.email)
if domain.security_contact.email != "registrar@dotgov.gov": if domain.security_contact.email != "registrar@dotgov.gov":
domain._update_epp_contact(contact=domain.security_contact) domain._update_epp_contact(contact=domain.security_contact)
self.disclosed_domain_contacts.append(copy.deepcopy(domain.security_contact)) self.disclosed_domain_contacts++
else: else:
logger.info( logger.info(
"Skipping disclose for %s security contact %s.", "Skipping disclose for %s security contact %s.",
domain.domain_info, domain.domain_name,
domain.security_contact.email, domain.security_contact.email,
) )
self.skipped_domain_contacts.append(copy.deepcopy(domain.security_contact)) self.skipped_domain_contacts++
except Exception as err: except Exception as err:
# error condition if domain not in database # error condition if domain not in database
self.domains_with_errors.append(copy.deepcopy(domain.domain_info)) self.domains_with_errors.append(copy.deepcopy(domain.domain_info))
logger.error(f"error retrieving domain {domain.domain_info}: {err}") logger.error(f"error retrieving domain {domain.domain_info}: {err}")
# Inform user how many contacts were disclosed, skipped, and errored # Inform user how many contacts were disclosed, skipped, and errored
logger.info("Updated %d contacts to disclosed.", len(self.disclosed_domain_contacts)) logger.info("Updated %d contacts to disclosed.", self.disclosed_domain_contacts)
logger.info( logger.info(
"Skipped disclosing %d contacts with security email registrar@dotgov.gov.", "Skipped disclosing %d contacts with security email registrar@dotgov.gov.",
len(self.skipped_domain_contacts) self.skipped_domain_contacts
)
logger.info(
"Error disclosing the following %d contacts: s",
len(self.domains_with_errors),
self.domains_with_errors
) )
logger.info("Error disclosing %d contacts.", len(self.domains_with_errors))