From 53675aa69c4545471083b9a39340ee4bf24a38c7 Mon Sep 17 00:00:00 2001
From: Lorenz Padberg <lorenz.padberg@iterativ.ch>
Date: Thu, 14 Apr 2022 15:03:36 +0200
Subject: [PATCH] Added category to learning sequence

---
 .../migrations/0004_auto_20220414_1503.py     | 26 +++++++++++++++++++
 server/vbv_lernwelt/learnpath/models.py       | 16 +++++++++++-
 2 files changed, 41 insertions(+), 1 deletion(-)
 create mode 100644 server/vbv_lernwelt/learnpath/migrations/0004_auto_20220414_1503.py

diff --git a/server/vbv_lernwelt/learnpath/migrations/0004_auto_20220414_1503.py b/server/vbv_lernwelt/learnpath/migrations/0004_auto_20220414_1503.py
new file mode 100644
index 00000000..54e985c0
--- /dev/null
+++ b/server/vbv_lernwelt/learnpath/migrations/0004_auto_20220414_1503.py
@@ -0,0 +1,26 @@
+# Generated by Django 3.2.12 on 2022-04-14 13:03
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('learnpath', '0003_circle_goals'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='learningsequence',
+            options={'verbose_name': 'Learning Sequence'},
+        ),
+        migrations.AlterModelOptions(
+            name='learningunit',
+            options={'verbose_name': 'Learning Unit'},
+        ),
+        migrations.AddField(
+            model_name='learningsequence',
+            name='category',
+            field=models.CharField(choices=[('INCIRCLE', 'In Circle'), ('START', 'Start'), ('END', 'End')], default='INCIRCLE', max_length=16),
+        ),
+    ]
diff --git a/server/vbv_lernwelt/learnpath/models.py b/server/vbv_lernwelt/learnpath/models.py
index 08ec90e1..3c6a2080 100644
--- a/server/vbv_lernwelt/learnpath/models.py
+++ b/server/vbv_lernwelt/learnpath/models.py
@@ -2,8 +2,8 @@
 
 from django.db import models
 from wagtail.admin.edit_handlers import FieldPanel
-from wagtail.core.fields import RichTextField
 from wagtail.core.models import Page
+from django.utils.translation import gettext_lazy as _
 
 
 # class HomePage(Page):
@@ -64,7 +64,21 @@ class Circle(Page):
         return f"{self.title}"
 
 
+IN_CIRCLE = 'INCIRCLE'
+START = 'START'
+END = 'END'
+
+LEARNING_SEQUENCE_CATEGORIES = [
+    (IN_CIRCLE, 'In Circle'),
+    (START, 'Start'),
+    (END, 'End')
+]
+
+
 class LearningSequence(Page):
+    # TODO: How to do a icon choice field?
+    category = models.CharField(max_length=16, choices=LEARNING_SEQUENCE_CATEGORIES, default=IN_CIRCLE)
+
     parent_page_types = ['learnpath.Circle']
     subpage_types = ['learnpath.LearningUnit']