Add management command for moving all Chronos dates to current year

(cherry picked from commit 923d2157)

aleksis/apps/untis/management/commands/move_dates_for_testing.py

+from datetime import date, timedelta
+
+from django.core.management.base import BaseCommand
+from django.utils import timezone
+
+from calendarweek import CalendarWeek
+from tqdm import tqdm
+
+from aleksis.apps.chronos.models import (
+    Absence,
+    Event,
+    Exam,
+    ExtraLesson,
+    Holiday,
+    LessonSubstitution,
+    ValidityRange,
+)
+from aleksis.core.models import SchoolTerm
+
+
+class Command(BaseCommand):
+    help = "Move all dates to current school year"  # noqa
+
+    def add_arguments(self, parser):
+        parser.add_argument("--dry", action="store_true")
+
+    def translate_date_start_end(self, name: str, qs):
+        self.stdout.write(name.upper())
+        for instance in tqdm(qs):
+            date_start = instance.date_start
+            date_end = instance.date_end
+            date_start_new = date_start + self.time_delta
+            date_end_new = date_end + self.time_delta
+            self.stdout.write(f"{date_start} → {date_start_new}; {date_end} → {date_end_new}")
+            instance.date_start = date_start_new
+            instance.date_end = date_end_new
+            if not self.dry:
+                instance.save()
+
+    def translate_date(self, name: str, qs):
+        self.stdout.write(name.upper())
+        for instance in tqdm(qs):
+            date_old = instance.date
+            date_new = date_old + self.time_delta
+            self.stdout.write(f"{date_old} → {date_new}")
+            instance.date = date_new
+            if not self.dry:
+                instance.save()
+
+    def translate_week_year(self, name: str, qs):
+        self.stdout.write(name.upper())
+        for instance in tqdm(qs):
+            date_old = instance.date
+            date_new = instance.date + self.time_delta
+            week_new = CalendarWeek.from_date(date_new)
+            self.stdout.write(f"{date_old}, {instance.week} → {date_new}, {week_new.week}")
+            instance.year = date_new.year
+            instance.week = week_new.week
+            if not self.dry:
+                instance.save()
+
+    def handle(self, *args, **options):
+        self.dry = options["dry"]
+
+        school_terms = SchoolTerm.objects.order_by("-date_end")
+
+        if not school_terms.exists():
+            raise RuntimeError("No validity range available.")
+
+        school_term = school_terms.first()
+        self.stdout.write(f"Used school term: {school_term}")
+
+        date_start = school_term.date_start
+        date_end = school_term.date_end
+        current_date = timezone.now().date()
+        current_year = current_date.year
+        if date_start.month <= current_date.month and date_start.day < current_date.day:
+            current_year -= 1
+        date_end = date(year=current_year, day=date_start.day, month=date_start.month)
+        self.stdout.write(f"{current_year}, {date_end}")
+        days = (date_end - date_start).days
+        self.weeks = round(days / 7)
+        self.time_delta = timedelta(days=self.weeks * 7)
+        self.stdout.write(f"{days}, {self.weeks}")
+
+        self.translate_date_start_end("SCHOOL TERM", [school_term])
+
+        self.translate_date_start_end(
+            "VALIDITY RANGES", ValidityRange.objects.filter(school_term=school_term)
+        )
+
+        if not self.weeks:
+            raise RuntimeError("You need at least one validity range to move data.")
+
+        self.translate_week_year(
+            "SUBSTITUTIONS",
+            LessonSubstitution.objects.filter(
+                lesson_period__lesson__validity__school_term=school_term
+            ),
+        )
+        self.translate_week_year(
+            "EXTRA LESSONS", ExtraLesson.objects.filter(school_term=school_term)
+        )
+        self.translate_date_start_end("ABSENCES", Absence.objects.filter(school_term=school_term))
+        self.translate_date("EXAMS", Exam.objects.filter(school_term=school_term))
+        self.translate_date_start_end(
+            "HOLIDAYS", Holiday.objects.within_dates(date_start, date_end)
+        )
+        self.translate_date_start_end("EVENTS", Event.objects.filter(school_term=school_term))