Convert columns to postgres enums

Author: JelteF

pgcommitfest/commitfest/migrations/0008_cfbot_use_enum.py

@@ -0,0 +1,52 @@
+# Generated by Django 3.2.25 on 2024-12-19 19:58
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('commitfest', '0007_cfbot_add_scheduled'),
+    ]
+
+    operations = [
+        migrations.RunSQL(
+            """
+            CREATE TYPE cfbotbranch_status AS ENUM (
+                'testing',
+                'finished',
+                'failed',
+                'timeout'
+            );
+            """
+        ),
+        migrations.RunSQL(
+            """
+            CREATE TYPE cfbottask_status AS ENUM (
+                'CREATED',
+                'NEEDS_APPROVAL',
+                'TRIGGERED',
+                'EXECUTING',
+                'FAILED',
+                'COMPLETED',
+                'SCHEDULED',
+                'ABORTED',
+                'ERRORED'
+            );
+            """
+        ),
+        migrations.RunSQL(
+            """
+            ALTER TABLE commitfest_cfbotbranch
+            ALTER COLUMN status TYPE cfbotbranch_status
+            USING status::cfbotbranch_status;
+            """
+        ),
+        migrations.RunSQL(
+            """
+            ALTER TABLE commitfest_cfbottask
+            ALTER COLUMN status TYPE cfbottask_status
+            USING status::cfbottask_status;
+            """
+        ),
+    ]

pgcommitfest/commitfest/models.py

@@ -356,6 +356,7 @@ class CfbotBranch(models.Model):
     branch_name = models.TextField(null=False)
     commit_id = models.TextField(null=True, blank=True)
     apply_url = models.TextField(null=False)
+    # Actually a postgres enum column
     status = models.TextField(choices=STATUS_CHOICES, null=False)
     created = models.DateTimeField(auto_now_add=True)
     modified = models.DateTimeField(auto_now=True)
@@ -379,6 +380,7 @@ class CfbotTask(models.Model):
     patch = models.ForeignKey(Patch, on_delete=models.CASCADE, related_name="cfbot_tasks")
     branch_id = models.IntegerField(null=False)
     position = models.IntegerField(null=False)
+    # Actually a postgres enum column
     status = models.TextField(choices=STATUS_CHOICES, null=False)
     created = models.DateTimeField(auto_now_add=True)
     modified = models.DateTimeField(auto_now=True)