ramon
/
skillbox
1
0
Fork 0
Code Issues 1 Pull Requests 1 Packages Projects 1 Releases 1 Wiki Activity

Save new expiry_dates to models

This commit is contained in:
Christian Cueni 2021-09-09 13:22:44 +02:00
parent adba163c77
commit fbc441d1ea
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
server/users/management/commands/reset_license_duration.py
View File

@ -1,5 +1,5 @@
import json
from datetime import datetime, date
from datetime import datetime, date, timedelta
from django.utils.timezone import now
from django.core.management.base import BaseCommand
@ -18,7 +18,7 @@ class Command(BaseCommand):
"""
def add_arguments(self, parser):
parser.add_argument('--dry-run', action='store_true', dest='save', default=False, help='Make dry-run')
parser.add_argument('--dry-run', action='store_true', dest='dry_run', default=False, help='Make dry-run')
def handle(self, *args, **options):
today = now().date()
@ -28,16 +28,16 @@ class Command(BaseCommand):
if isbn not in YEARLY_ISBNS:
continue
defined_duration = MYSKILLBOX_LICENSES[isbn]['duration'] # use isbn from hep due to our bug
standard_duration = MYSKILLBOX_LICENSES[isbn]['duration'] # use isbn from hep due to our bug
start_date = license.get_hep_start_date()
duration_in_days = (license.expire_date - start_date).days
if duration_in_days == defined_duration:
if duration_in_days == standard_duration:
continue
user_join_delta = (license.expire_date - license.licensee.date_joined.date()).days
if self._is_change_candidate(user_join_delta, duration_in_days, start_date):
print(license.id, duration_in_days, defined_duration, license.expire_date, start_date, user_join_delta)
self._update_expiry_date(license, standard_duration, options['dry_run'])
def _is_change_candidate(self, user_join_delta: int, duration_in_days: int, start_date: date) -> bool:
"""
@ -48,3 +48,21 @@ class Command(BaseCommand):
"""
return user_join_delta == LONGER_DURATION or duration_in_days == LONGER_DURATION \
or start_date == License.NO_DATE
def _update_expiry_date(self, license: License, standard_duration: int, dry_run: bool):
new_expiry_date = license.expire_date - timedelta(LONGER_DURATION - standard_duration)
user_expire_date = license.licensee.license_expiry_date
if user_expire_date == license.expire_date:
print(f'Resetting user expire_date for userID {license.licensee.id} from'
f' {license.licensee.license_expiry_date} to {new_expiry_date}')
if not dry_run:
license.licensee.license_expiry_date = new_expiry_date
license.licensee.save()
print(f'Resetting expiry date for license ID {license.id} ({license.expire_date}) to {new_expiry_date}')
if not dry_run:
license.expire_date = new_expiry_date
license.save()