ubccr#294: Created a data migration file so that existing ColdFront i…

…nstances, can get the allocation change status choices loaded into their application automatically

coldfront/core/allocation/migrations/0005_auto_20211117_1413.py

@@ -0,0 +1,20 @@
+# Generated by Django 2.2.24 on 2021-11-17 19:13
+
+from django.db import migrations
+
+
+def create_status_choices(apps, schema_editor):
+    AllocationChangeStatusChoice = apps.get_model('allocation', "AllocationChangeStatusChoice")
+    for choice in ('Pending', 'Approved', 'Denied',):
+        AllocationChangeStatusChoice.objects.get_or_create(name=choice)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('allocation', '0004_auto_20211102_1017'),
+    ]
+
+    operations = [
+        migrations.RunPython(create_status_choices),
+    ]

coldfront/core/allocation/models.py

@@ -325,6 +325,13 @@ class AllocationChangeRequest(TimeStampedModel):
     notes = models.CharField(max_length=512, blank=True, null=True)
     history = HistoricalRecords()
 
+    @property
+    def get_parent_resource(self):
+        if self.allocation.resources.count() == 1:
+            return self.allocation.resources.first()
+        else:
+            return self.allocation.resources.filter(is_allocatable=True).first()
+
     def __str__(self):
         return "%s (%s)" % (self.get_parent_resource.name, self.allocation.project.pi)