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

added debugging directives

Browse source
This commit is contained in:
CocoByte 2023-09-21 14:17:55 -06:00
parent 125690f347
commit 7be4c300bf
No known key found for this signature in database
GPG key ID: BBFAA2526384C97F
1 changed files with 46 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -42,9 +42,29 @@ class Command(BaseCommand):
domain_statuses_filename, domain_statuses_filename,
**options **options
): ):
"""Load the data files and create the DomainInvitations.""" """Load the data files and create the DomainInvitations."""
sep = options.get("sep") sep = options.get("sep")
debug_MaxEntriesToParse = 10
class termColors:
HEADER = '\033[95m'
OKBLUE = '\033[94m'
OKCYAN = '\033[96m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
BOLD = '\033[1m'
UNDERLINE = '\033[4m'
if __debug__:
print(termColors.WARNING)
print("----------DEBUG MODE ON----------")
print(f"Parsing of entries will be limited to {debug_MaxEntriesToParse} lines per file.")
print("Detailed print statements activated.")
print(termColors.ENDC)
# STEP 1: # STEP 1:
# Create mapping of domain name -> status # Create mapping of domain name -> status
# TODO: figure out latest status # TODO: figure out latest status
@ -57,7 +77,6 @@ class Command(BaseCommand):
# print("adding "+domainName+", "+domainStatus) # print("adding "+domainName+", "+domainStatus)
domain_status_dictionary[domainName] = domainStatus domain_status_dictionary[domainName] = domainStatus
logger.info("Loaded statuses for %d domains", len(domain_status_dictionary)) logger.info("Loaded statuses for %d domains", len(domain_status_dictionary))
print(domain_status_dictionary)
# STEP 2: # STEP 2:
# Create mapping of userId -> emails NOTE: is this one to many?? # Create mapping of userId -> emails NOTE: is this one to many??
@ -148,32 +167,44 @@ class Command(BaseCommand):
) )
''' '''
newOrExistingEntry = None
try: try:
newOrExistingEntry = TransitionDomain.objects.get( existingEntry = TransitionDomain.objects.get(
username=userEmail, username=userEmail,
domain_name=domainName domain_name=domainName
) )
print("Updating entry: ", newOrExistingEntry)
print(" Status: ", newOrExistingEntry.status, " > ",domainStatus) # DEBUG:
newOrExistingEntry.status = domainStatus if __debug__:
print(" Email Sent: ", newOrExistingEntry.email_sent, " > ", emailSent) print(termColors.WARNING)
newOrExistingEntry.email_sent = emailSent print("Updating entry: ", existingEntry)
newOrExistingEntry.save() print(" Status: ", existingEntry.status, " > ",domainStatus)
print(" Email Sent: ", existingEntry.email_sent, " > ", emailSent)
existingEntry.status = domainStatus
existingEntry.email_sent = emailSent
existingEntry.save()
except TransitionDomain.DoesNotExist: except TransitionDomain.DoesNotExist:
# no matching entry, make one # no matching entry, make one
newOrExistingEntry = TransitionDomain( newEntry = TransitionDomain(
username=userEmail, username=userEmail,
domain_name=domainName, domain_name=domainName,
status = domainStatus, status = domainStatus,
email_sent = emailSent email_sent = emailSent
) )
print("Adding entry ",total_new_entries,": ", newOrExistingEntry) to_create.append(newEntry)
to_create.append(newOrExistingEntry)
# DEBUG:
if __debug__:
print("Adding entry ",total_new_entries,": ", newEntry)
total_new_entries += 1 total_new_entries += 1
if total_new_entries > 10:
print("DONE") # DEBUG:
break if __debug__:
if total_new_entries > debug_MaxEntriesToParse:
print("----BREAK----")
print(termColors.ENDC)
break
logger.info("Creating %d transition domain entries", len(to_create)) logger.info("Creating %d transition domain entries", len(to_create))
TransitionDomain.objects.bulk_create(to_create) TransitionDomain.objects.bulk_create(to_create)