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

Fix test case + linting

Browse source
This commit is contained in:
zandercymatics 2023-12-12 12:56:12 -07:00
parent 9cc9a8c2da
commit 038fa3b266
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 26 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/registrar/management/commands/extend_expiration_dates.py
View file

@ -9,9 +9,9 @@ from django.core.management import BaseCommand
from epplibwrapper.errors import RegistryError
from registrar.models import Domain
from registrar.management.commands.utility.terminal_helper import TerminalColors, TerminalHelper
from datetime import datetime
from registrar.models.transition_domain import TransitionDomain
try:
from epplib.exceptions import TransportError
except ImportError:
@ -94,7 +94,9 @@ class Command(BaseCommand):
domain.renew_domain(extension_amount)
self.update_success.append(domain.name)
logger.info(
f"{TerminalColors.OKCYAN}" f"Successfully updated expiration date for {domain}" f"{TerminalColors.ENDC}"
f"{TerminalColors.OKCYAN}"
f"Successfully updated expiration date for {domain}"
f"{TerminalColors.ENDC}"
)
# Catches registry errors. Failures indicate bad data, or a faulty connection.
except (RegistryError, KeyError, TransportError) as err:
@ -117,8 +119,7 @@ class Command(BaseCommand):
# CAVEAT: This is a workaround. A more robust solution would be a db flag
current_expiration_date = domain.registry_expiration_date
transition_domains = TransitionDomain.objects.filter(
domain_name=domain.name,
epp_expiration_date=current_expiration_date
domain_name=domain.name, epp_expiration_date=current_expiration_date
)
proposed_date = self.add_years(current_expiration_date, extension_amount)
minimum_extension_date = self.add_years(self.expiration_cutoff, extension_amount)