add enrollment type

Author: Melevir

Makefile

@@ -10,3 +10,5 @@ check:
 	make isort style types test
 run:
 	cd landing_page; DEBUG=1 python manage.py runserver
+shell:
+	cd landing_page; DEBUG=1 python manage.py shell

landing_page/mainpage/migrations/0043_enrollment_type.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.4 on 2023-09-08 13:49
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mainpage', '0042_enrollment_platim_url'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='enrollment',
+            name='type',
+            field=models.CharField(blank=True, choices=[('BASE', 'BASE'), ('ADVANCED', 'ADVANCED')], max_length=10, null=True),
+        ),
+    ]

landing_page/mainpage/models.py

@@ -1,3 +1,5 @@
+import enum
+
 from django.db import models
 from django.utils.timezone import now
 
@@ -55,10 +57,20 @@ def __str__(self) -> str:
     )
 
 
+class EnrollmentType(enum.StrEnum):
+    BASE = "BASE"
+    ADVANCED = "ADVANCED"
+
+    @classmethod
+    def get_choices(cls) -> list[tuple[str, str]]:
+        return [(v, v) for v in cls]
+
+
 class Enrollment(models.Model):
     timepad_event_id = models.CharField(max_length=64, null=True, blank=True)
     platim_url = models.CharField(max_length=254, null=True, blank=True)
 
+    type = models.CharField(max_length=10, choices=EnrollmentType.get_choices(), null=True, blank=True)
     start_date = models.DateField()
     end_date = models.DateField()
     end_registration_date = models.DateField()
@@ -68,8 +80,11 @@ class Enrollment(models.Model):
     late_price_date_from = models.DateField()
 
     @staticmethod
-    def get_enrollment_with_active_registration() -> "Enrollment | None":
-        return Enrollment.objects.filter(end_registration_date__gte=now()).first()
+    def get_enrollment_with_active_registration(enrollment_type: EnrollmentType) -> "Enrollment | None":
+        return Enrollment.objects.filter(
+            type=enrollment_type,
+            end_registration_date__gte=now(),
+        ).first()
 
     def __str__(self) -> str:
-        return f"Enrollment ({self.start_date} - {self.end_date})"
+        return f"{self.type.capitalize()} enrollment ({self.start_date} - {self.end_date})"