Add availability to programs in models.py (#2322)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

courses/constants.py

@@ -31,3 +31,8 @@
 HOMEPAGE_CACHE_AGE = 86400  # 24 hours
 
 SYNCED_COURSE_RUN_FIELD_MSG = "This value is synced automatically with edX studio."
+
+AVAILABILITY_ANYTIME = "anytime"
+AVAILABILITY_DATED = "dated"
+AVAILABILITY_TYPES = {AVAILABILITY_ANYTIME, AVAILABILITY_DATED}
+AVAILABILITY_CHOICES = list(zip(AVAILABILITY_TYPES, AVAILABILITY_TYPES))

courses/forms.py

@@ -410,6 +410,7 @@ class Meta:
             "departments",
             "live",
             "requirements",
+            "availability",
         ]
 
     class Media:

courses/migrations/0052_add_program_availability.py

@@ -0,0 +1,21 @@
+# Generated by Django 3.2.25 on 2024-08-02 02:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("courses", "0051_coursestopic"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="program",
+            name="availability",
+            field=models.CharField(
+                choices=[("anytime", "anytime"), ("dated", "dated")],
+                default="anytime",
+                max_length=255,
+            ),
+        ),
+    ]

courses/models.py

@@ -24,6 +24,8 @@
 from wagtail.models import Revision
 
 from courses.constants import (
+    AVAILABILITY_ANYTIME,
+    AVAILABILITY_CHOICES,
     ENROLL_CHANGE_STATUS_CHOICES,
     ENROLLABLE_ITEM_ID_SEPARATOR,
     SYNCED_COURSE_RUN_FIELD_MSG,
@@ -156,6 +158,9 @@ class Meta:
         null=True,
     )
     departments = models.ManyToManyField(Department, blank=False)
+    availability = models.CharField(
+        choices=AVAILABILITY_CHOICES, default=AVAILABILITY_ANYTIME, max_length=255
+    )
 
     @cached_property
     def page(self):

courses/serializers/v2/programs.py

@@ -77,6 +77,7 @@ class Meta:
             "departments",
             "live",
             "topics",
+            "availability",
         ]
 
 

courses/serializers/v2/programs_test.py

@@ -92,5 +92,6 @@ def test_serialize_program(
             "departments": [],
             "live": True,
             "topics": [{"name": topic.name} for topic in topics],
+            "availability": "anytime",
         },
     )