diff --git a/training/migrations/0006_auto_20210903_2158.py b/training/migrations/0006_auto_20210903_2158.py new file mode 100644 index 00000000..e7e5617f --- /dev/null +++ b/training/migrations/0006_auto_20210903_2158.py @@ -0,0 +1,23 @@ +# Generated by Django 3.1.7 on 2021-09-03 20:58 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('training', '0005_auto_20210819_1833'), + ] + + operations = [ + migrations.AddField( + model_name='traininglevel', + name='icon', + field=models.CharField(blank=True, max_length=20, null=True), + ), + migrations.AlterField( + model_name='traininglevelrequirement', + name='depth', + field=models.IntegerField(choices=[(0, 'Training Started'), (1, 'Training Complete'), (2, 'Passed Out')]), + ), + ] diff --git a/training/models.py b/training/models.py index 8903b98b..edda8359 100644 --- a/training/models.py +++ b/training/models.py @@ -119,6 +119,19 @@ class TrainingLevel(models.Model, RevisionMixin): department = models.IntegerField(choices=DEPARTMENTS, null=True) # N.B. Technical Assistant does not have a department level = models.IntegerField(choices=CHOICES) prerequisite_levels = models.ManyToManyField('self', related_name='prerequisites', symmetrical=False, blank=True) + icon = models.CharField(null=True, blank=True, max_length=20) + + def get_department_colour(self): + if self.department == 0: + return "info" + elif self.department == 1: + return "dark" + elif self.department == 2: + return "danger" + elif self.department == 3: + return "light" + else: + return "primary" def get_requirements_of_depth(self, depth): return self.requirements.filter(depth=depth) diff --git a/training/templates/base_training.html b/training/templates/base_training.html index 3a696d53..9e2ec8c5 100644 --- a/training/templates/base_training.html +++ b/training/templates/base_training.html @@ -20,6 +20,8 @@ Item Detail +
{{ object.description }}
Please Note:
+{{ level.description }}
+