devicehub-django/action/models.py

from django.db import models, connection
from django.db.models import Max
from user.models import User, Institution
from django.core.exceptions import ValidationError

class State(models.Model):
    date = models.DateTimeField(auto_now_add=True)
    institution = models.ForeignKey(Institution, on_delete=models.SET_NULL, null=True)
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)

    state = models.CharField(max_length=50)
    snapshot_uuid = models.UUIDField()

    def clean(self):
        if not StateDefinition.objects.filter(institution=self.institution, state=self.state).exists():
            raise ValidationError(f"The state '{self.state}' is not valid for the institution '{self.institution.name}'.")
    
    def save(self, *args, **kwargs):
        self.clean()
        super().save(*args, **kwargs)

    def __str__(self):
        return f"{self.institution.name} - {self.state} - {self.snapshot_uuid}"

class StateDefinition(models.Model):
    institution = models.ForeignKey(Institution, on_delete=models.CASCADE)

    order = models.PositiveIntegerField(default=0)
    state = models.CharField(max_length=50)

    class Meta:
        ordering = ['order']
        constraints = [
            models.UniqueConstraint(fields=['institution', 'state'], name='unique_institution_state')
        ]

    def save(self, *args, **kwargs):
        if not self.pk:
            # set the order to be last
            max_order = StateDefinition.objects.filter(institution=self.institution).aggregate(Max('order'))['order__max']
            self.order = (max_order or 0) + 1
        super().save(*args, **kwargs)

    
    def delete(self, *args, **kwargs):
        institution = self.institution
        order = self.order
        super().delete(*args, **kwargs)
        # Adjust the order of other instances
        StateDefinition.objects.filter(institution=institution, order__gt=order).update(order=models.F('order') - 1)


    def __str__(self):
        return f"{self.institution.name} - {self.state}"