Сохранение объекта с отношением ManyToMany
У меня есть модели (упрощенный пример):
class Group(models.Model):
name = models.CharField(max_length = 32)
class Person(models.Model):
group = models.ForeignKey(Group)
class Task(models.Model):
group = models.ForeignKey(Group)
people = models.ManyToManyField(Person)
def save(self, **kwargs):
ppl = Person.objects.all().filter(group = self.group)
for p in ppl:
self.people.add(p)
super(Task, self).save(**kwargs)
Я хочу назначить задачу какой-то группе людей и добавить всех людей, которые принадлежат к этой группе, а также некоторых других людей позже (или удалить конкретного человека из задачи). Очевидно, что сохранение не будет выполнено, потому что объект не имеет идентификатора, когда он хочет добавить объекты отношения «многие ко многим». Как справиться с такой ситуацией? Я попытался сохранить непосредственно перед добавлением людей в задачу, а затем снова сохранить, но это не сработало.
С уважением
Крисс
Ответы 2
Вот пример, который не требует отмены сохранения. Это выглядит проще, чем то, что вы делаете.
Вот как бы я это сделал. Создайте отдельную группу для каждой задачи и, при желании, инициализируйте каждую группу другой:
class Person(models.Model):
name = models.CharField(max_length=64)
class Group(models.Model):
name = models.CharField(max_length=32)
people = models.ManyToManyField(Person)
parent_group = models.ForeignKey(Group, null=True, blank=True)
def save(self, **kwargs):
# copy parent's people in to this group if this is a new instance and parent_group is set
if not self.id and self.parent_group:
for p in self.parent_group.people:
self.people.add(p)
super(Group, self).save(**kwargs)
class Task(models.Model):
group = models.ForeignKey(Group)
def use_case():
Group(name='empty group').save()
group = Group.objects.get(name='Accounting')
accounting_subgroup = Group(name='Accounting Copy', parent_group=group)
accounting_subgroup.people.remove(person1)
accounting_subgroup.people.remove(person2)
accounting_subgroup.save()
task = Task(group=accounting_subgroup)
task.save()
Теперь вы можете повторно использовать свои подгруппы, а также определить, какие «базовые группы», например, «Учет и продажи» против «Бухгалтерская группа 3», проверив, имеет ли значение parent_group значение null. Однако в большинстве случаев вы не дублируете «список людей» в двух местах.