51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
import json
|
|
from datetime import datetime, date
|
|
|
|
from django.utils.timezone import now
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from users.licenses import MYSKILLBOX_LICENSES
|
|
from users.models import License
|
|
|
|
YEARLY_ISBNS = ["978-3-0355-1860-3", "978-3-0355-1823-8"]
|
|
LONGER_DURATION = 455
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """
|
|
Reset licenses that have a duration of one year
|
|
Uses the duration of MYSKILLBOX_LICENSES as reference.
|
|
"""
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--dry-run', action='store_true', dest='save', default=False, help='Make dry-run')
|
|
|
|
def handle(self, *args, **options):
|
|
today = now().date()
|
|
|
|
for license in License.objects.filter(expire_date__gte=today):
|
|
isbn = license.get_isbn()
|
|
if isbn not in YEARLY_ISBNS:
|
|
continue
|
|
|
|
defined_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:
|
|
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)
|
|
|
|
def _is_change_candidate(self, user_join_delta: int, duration_in_days: int, start_date: date) -> bool:
|
|
"""
|
|
user_join_delta == LONGER_DURATION: user joined the same number of days as the old duration was
|
|
duration_in_days == LONGER_DURATION: delta from start date and expiry date is the same as the old duration
|
|
start_date == NO_DATE: license does not have a start date -> was created after moving to new api as
|
|
such it the longer duration was set as duration
|
|
"""
|
|
return user_join_delta == LONGER_DURATION or duration_in_days == LONGER_DURATION \
|
|
or start_date == License.NO_DATE
|