feature: new model for custom status

chats/apps/projects/migrations/0021_customstatus.py

@@ -0,0 +1,41 @@
+# Generated by Django 4.1.2 on 2025-01-31 19:57
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("projects", "0020_project_org"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="CustomStatus",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("name", models.CharField(max_length=255)),
+                ("is_deleted", models.BooleanField(default=False)),
+                (
+                    "project",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="custom_statuses",
+                        to="projects.project",
+                    ),
+                ),
+            ],
+            options={
+                "unique_together": {("name", "project")},
+            },
+        ),
+    ]

chats/apps/projects/models/models.py

@@ -15,6 +15,9 @@
 
 from .permission_managers import UserPermissionsManager
 
+from django.core.exceptions import ValidationError
+from django.db import transaction
+
 User = get_user_model()
 
 # Create your models here.
@@ -500,3 +503,35 @@ def __str__(self):
     @property
     def project(self):
         return self.flow_start.project
+
+
+class CustomStatus(models.Model):
+    name = models.CharField(max_length=255)
+    project = models.ForeignKey(
+        "Project", on_delete=models.CASCADE, related_name="custom_statuses"
+    )
+    is_deleted = models.BooleanField(default=False)
+
+    def delete(self, *args, **kwargs):
+        self.is_deleted = True
+        self.save(update_fields=["is_deleted"])
+
+    def save(self, *args, **kwargs):
+        if not self.pk:
+            with transaction.atomic():
+                existing_count = (
+                    CustomStatus.objects.select_for_update()
+                    .filter(project=self.project, is_deleted=False)
+                    .count()
+                )
+                if existing_count >= 3:
+                    raise ValidationError(
+                        "A project can have a maximum of 3 custom statuses."
+                    )
+        super().save(*args, **kwargs)
+
+    def __str__(self):
+        return self.name
+
+    class Meta:
+        unique_together = ("name", "project")