zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-30 01:10:04 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update extend_expiration_dates.py

Browse source
This commit is contained in:
zandercymatics 2023-12-08 15:20:45 -07:00
parent 40d7588731
commit b4658e6da3
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -14,7 +14,6 @@ try:
except ImportError: except ImportError:
pass pass
from dateutil.relativedelta import relativedelta
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -237,7 +236,7 @@ class Command(BaseCommand):
) )
# We use this manual approach rather than relative delta due to our # We use this manual approach rather than relative delta due to our
# github localenv not having the package installed. # github localenv not having the package installed.
# Credit: https://stackoverflow.com/questions/15741618/add-one-year-in-current-date-python # Credit: https://stackoverflow.com/questions/15741618/add-one-year-in-current-date-python
def add_years(self, old_date, years): def add_years(self, old_date, years):
"""Return a date that's `years` years after the date (or datetime) """Return a date that's `years` years after the date (or datetime)
@ -247,6 +246,6 @@ class Command(BaseCommand):
""" """
try: try:
return old_date.replace(year = old_date.year + years) return old_date.replace(year=old_date.year + years)
except ValueError: except ValueError:
return old_date + (date(old_date.year + years, 1, 1) - date(old_date.year, 1, 1)) return old_date + (date(old_date.year + years, 1, 1) - date(old_date.year, 1, 1))