4

I want to be able to make a model unique on a value that comes from a foreign key model. Not sure if that's possible in django.

Example: I have a model A as:

class modelA(models.Model):
    fieldA1 = models.AutoField(primary_key=True)
    fieldA2 = models.CharField(max_length=20)

And also model B as:

class modelB(models.Model):
    fieldB1 = models.CharField(max_length=20)
    fieldB2 = models.ForeignKey(modelA)

    class Meta:
        unique_together = ('fieldB1', 'fieldB2',<<'fieldA2'>>)

I want to add fieldA2 as one of the attributes in the unique_together clause of model B. Is that possible in Django? I COULD NOT do it as-

unique_together = ('fieldB1', 'fieldB2','modelA__fieldA2')
doniyor
  • 36,596
  • 57
  • 175
  • 260
Shiv
  • 332
  • 4
  • 12

2 Answers2

3

you cannot. database constraints cannot contain those kind of data. you will have to check it programmatically before creating instances or write some validator

doniyor
  • 36,596
  • 57
  • 175
  • 260
  • 2
    That's unfortunate. Thanks! EDIT: Found this that's related - http://stackoverflow.com/questions/14470585/how-to-validate-uniqueness-constraint-across-foreign-key-django – Shiv Jun 28 '16 at 18:31
0

You can override modelB's clean method like this

class ModelB(models.Model):
    fieldB1 = models.CharField(max_length=20)
    fieldB2 = models.ForeignKey(modelA)

    def clean(self):
        """
        Validate 'fieldB1', 'fieldB2' and 'fieldB2__fieldA2' are unique together
        """
        # first, check that foreign keys fields exist. Must use '_id', or else a RelatedObjectDoesNotExist error will be raised
        if self.fieldB2_id and self.fieldB2.field_A2_id:
            if self.__class__.objects.filter(fieldB1=self.fieldB1, fieldB2=self.fieldB2, fieldB2__fieldA2=self.fieldB2.fieldA2).exists():
                raise ValidationError(
                    _('Instance of ModelB with fieldB1, fieldB2 and fieldB2.fieldA2 already exists.'),
                    code='unique_together',
                )
ekauffmann
  • 150
  • 10