Джанго: Как представлять и запросить симметричные отношения для семейного древа? ⇐ Python

[Anonymous]

Я строю приложение генеалогического древа в Джанго, где мне нужно представлять в браках симметрично. Каждый брак должен иметь только одну запись, а отношения должны включать обоих партнеров без дублирования данных. Вот соответствующая структура модели: < /p>
class Person(models.Model):
first_name = models.CharField(max_length=100)
last_name = models.CharField(max_length=100)
spouses = models.ManyToManyField(
'self', through="Marriage", symmetrical=True, related_name="partners"
)

class Marriage(models.Model):
person1 = models.ForeignKey(Person, on_delete=models.CASCADE, related_name="marriages_as_person1")
person2 = models.ForeignKey(Person, on_delete=models.CASCADE, related_name="marriages_as_person2")
start_date = models.DateField(null=True, blank=True)
end_date = models.DateField(null=True, blank=True)
< /code>
Я хочу: < /p>

[*] Убедитесь, что оба партнера появляются как супруги друг для друга симметрично. < /li>
Избегайте дублирующихся записей для одного и того же брака. < /li>
Эффективно запросить всех супругов человека. > Вот код, который я использую для запроса супругов: < /p>
# Query spouses for a person
p1 = Person.objects.create()
p2 = Person.objects.create()
Marriage.objects.create(person1=p1, person2=p2)

p1.spouses.all() # Returns list containing p2
p2.spouses.all() # Returns empty list
< /code>
Тем не менее, я сталкиваюсь с проблемами: < /p>

Если P1 < /code> супруги запрашиваются, он должен содержать P2 и если P2 супруги запрошены, он должен содержать P1
[*] Оба запроса не являются симметричными

Вопросы: < /h3>

Является ли моя структура модели правильной для симметричного представления браков? Если нет, то какие улучшения я должен сделать? ol>
Мой вариант использования - вернуть список или человека, имеющего PID < /code> (идентификатор партнера) в качестве списка идентификатора, как ниже, используя DRF < /p>
[
{
id: 1,
full_name: ‘John’,
pids: [2]
},
{
id: 2,
full_name: ‘Mary’,
pids: [1]
}
]
< /code>
Текущий код сериализатора-< /p>
def get_pids(self, obj):
"""
Returns a list of IDs of all the spouses of the person, ensuring bidirectional relationships.
"""
partner_ids = set(obj.spouses.values_list('id', flat=True))
# Ensure bidirectional relationships
for spouse in obj.spouses.all():
partner_ids.update(spouse.spouses.values_list('id', flat=True))

return list(partner_ids)


Подробнее здесь: https://stackoverflow.com/questions/793 ... amily-tree