Автозаполнение полей модели Django при создании экземпляра другой модели - PullRequest
0 голосов
/ 01 мая 2018

У меня есть три модели Student, Question и StudentAndQuestion

class Student(models.Model):
    class Meta:
        verbose_name_plural = "Students"
    name = models.CharField(max_length=150)
    surname = models.CharField(max_length=150)
    code = models.CharField(max_length=10)
#    group = models.ForeignKey(Group,on_delete=models.CASCADE)
    points = models.IntegerField(default=0)


class Question(models.Model):
    class Meta:
        verbose_name_plural = "Questions"

    text = models.CharField(max_length=1500)
    variants = models.CharField(max_length=1500)
    theme = models.OneToOneField(Theme, on_delete=models.CASCADE)
    anwser = models.CharField(max_length=1500)


class StudentAndQuestion(models.Model):
    question = models.OneToOneField(Question,on_delete=models.CASCADE)
    student = models.OneToOneField(Student,on_delete=models.CASCADE)
    is_learned = models.BooleanField(default=0)
    points = models.IntegerField(default=0)

Я хочу, чтобы Django сделал заполнение в StudentAndQuestion, когда я создаю новый экземпляр учеников (например, создаю строки для всех Вопросов и устанавливаю нулевые значения), а когда я создаю новый экземпляр Вопроса, добавляю его ко всем старым студентам

1 Ответ

0 голосов
/ 01 мая 2018

Вы можете прослушать сигнал Post Save , излучаемый при сохранении этих моделей, и выполнить необходимое создание в обработчике. Этот сигнал вызывает обработчик с аргументом created, равным True, если была создана новая запись.

Обратитесь к документации по сигналам , чтобы узнать, как зарегистрировать ваши обработчики. Образец выглядит следующим образом

from django.db.models.signals import post_delete, post_save

from .models import Student, Question, StudentAndQuestion


def add_question_to_students(sender, instance, created, *args, **kwargs):    
question = instance
if created:
    StudentAndQuestion.objects.bulk_create([
        StudentAndQuestion(question=question, student_id=student_id)
        for student_id in Student.objects.values_list("id", flat=True)
    ])

def add_student_to_question(sender, instance, created, *args, **kwargs):    
student = instance
if created:
    StudentAndQuestion.objects.bulk_create([
        StudentAndQuestion(question=question_id, student=student)
        for question_id in Question.objects.values_list("id", flat=True)
    ])

post_save.connect(add_question_to_students, sender=Question)
post_save.connect(add_student_to_question, sender=Student)

Кроме того, вы можете переопределить сохранение в обеих моделях, чтобы сделать то же самое

class Question(models.Model):
    ....
    def save(*args, **kwargs):
        is_create = self.pk is None
        instance = super().save(*args, **kwargs)
        add_question_to_students(Question, self, is_created)

class Student(models.Model):
    ....
    def save(*args, **kwargs):
        is_create = self.pk is None
        instance = super().save(*args, **kwargs)
        add_student_to_question(Student, self, is_created)

Я чувствую, что позже это будет понятнее, так как легче будет выяснить, что все происходит при сохранении, но это может привести к круговым зависимостям, если модели находятся в разных приложениях.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...