Определить ограничения модели Django - PullRequest
Новая
       7

Определить ограничения модели Django

2 голосов
/ 19 октября 2019

Я хочу добавить ограничение на модель, которая включает ссылку на другой объект модели. В приведенном ниже коде я хочу, чтобы в модели ProjectMember поле project_member ограничивалось отличиями от project_creator в модели проекта. 

class User(models.Model):
  fullname = models.CharField(max_length=200)
  email = models.EmailField(max_length=254)
  password = models.CharField(max_length=254)


class Project(models.Model):
  name = models.CharField(max_length=200)
  timestamp = models.DateTimeField("date created", auto_now=True)
  project_creator = models.ForeignKey(User, on_delete=models.CASCADE)


class ProjectMember(models.Model):
  project = models.ForeignKey(Project, on_delete=models.CASCADE)

  project_member = models.ForeignKey(
      User,
      on_delete=models.CASCADE,
      verbose_name="member of the project"
  )

  class Meta:
      order_with_respect_to = 'user_member'
      constraints = [
          models.CheckConstraint(check= , 
             name='project creator not project member'),
      ]

1 Ответ

3 голосов
/ 19 октября 2019

Вы можете добавить его в объект Q следующим образом: 

class Meta:
  order_with_respect_to = 'user_member'
  constraints = [
      models.CheckConstraint(check=~Q(project_member_id=project__project_creator_id) , 
         name='project creator not project member'),
  ]

Надеюсь, это поможет!

...