added model level constraints
This commit is contained in:
parent
b3c268a49e
commit
dc2418f61b
|
@ -1,4 +1,5 @@
|
||||||
from django.db import models, connection
|
from django.db import models, connection
|
||||||
|
from django.db.models import Max
|
||||||
from user.models import User, Institution
|
from user.models import User, Institution
|
||||||
from django.core.exceptions import ValidationError
|
from django.core.exceptions import ValidationError
|
||||||
|
|
||||||
|
@ -23,13 +24,31 @@ class State(models.Model):
|
||||||
|
|
||||||
class StateDefinition(models.Model):
|
class StateDefinition(models.Model):
|
||||||
institution = models.ForeignKey(Institution, on_delete=models.CASCADE)
|
institution = models.ForeignKey(Institution, on_delete=models.CASCADE)
|
||||||
order = models.AutoField(primary_key=True)
|
|
||||||
|
order = models.PositiveIntegerField(default=0)
|
||||||
state = models.CharField(max_length=50)
|
state = models.CharField(max_length=50)
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
|
ordering = ['order']
|
||||||
constraints = [
|
constraints = [
|
||||||
models.UniqueConstraint(fields=['institution', 'state'], name='unique_institution_state')
|
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):
|
def __str__(self):
|
||||||
return f"{self.institution.name} - {self.state}"
|
return f"{self.institution.name} - {self.state}"
|
||||||
|
|
Loading…
Reference in a new issue