Как проверить уникальность данных в полях ForeignKey?

Question

[Bjornie]

Это моя достаточно простая модель с 6 полями для выбора игрока в команду, чтобы в последствии включить команду в матч, не добавляя игроков по-отдельности.

class Team(models.Model):
    team_name = models.CharField(max_length=255)
    t_1_id = models.ForeignKey(Player, related_name='+')
    t_2_id = models.ForeignKey(Player, related_name='+')
    t_3_id = models.ForeignKey(Player, blank=True, null=True, related_name='+')
    t_4_id = models.ForeignKey(Player, blank=True, null=True, related_name='+')
    t_5_id = models.ForeignKey(Player, blank=True, null=True, related_name='+')
    t_6_id = models.ForeignKey(Player, blank=True, null=True, related_name='+')

Вопрос: как проверить уникальность выбранного игрока. т.е. чтобы в каждом поле игрок был уникален. Приходили идеи при валидации формы (говорю про админку) читать id выбранных игроков, сверять их через цикл, добавляя во временный массив и т.д. Но получается много кода. Подскажите пожалуйста, есть ли варианты по-серьезнее и по-проще?

Comments

[Артём Белоусов]

Они должны быть разные между собой и между командами?

[Bjornie]

Артём Белоусов: нет, только для созданной Team. Т.е. не может быть (1,1,2,3,4,1), как видите 3 раза выбран Иван Иванов (1), а должно быть (1,2,3,4,5,6)

[ns 5d]

Bjornie: ForeignKey => ManyToMany имхо было бы правильней.

[Артём Белоусов]

Тогда валидация модели. Проверка простая
arr = [self.t_1_id, и так далее]
if len(arr) != len(set(arr)):
raise ValidationError

[Bjornie]

Артём Белоусов: спасибо за решение, сделал как предложили ниже с many2many. Бонусом пошел приятный интерфейс для выбора!

Answer 1

[Astrohas]

делайте m2m и не изобретайте велосипеды.

class Team(model.Model):
    ...
   players  = models.ManyToManyField(Player)

В Админке делаете filter_horizontal

Comments

[Bjornie]

Хасан Истамкулов Интересное предложение, как вариант, чтобы избавиться от одинаковых полей с FK. Однако он все-равно позволяет выбрать одних и тех же игроков.

[Astrohas]

Bjornie: хм Вариант с AdminInline возможно, да. Но можно например filter_horizontal юзать и повторов не будет.

[Bjornie]

Хасан Истамкулов: то что надо. Спасибо!

[Anton Kuzmichev]

Bjornie: вариант с filter_horizontal создаст элемент управления наподобие выбора групп для пользователей Django (посмотрите в своей админке) - слева будут все игроки, которых вы можете перемещать вправо. Повторы в таком случае исключены. Проверять, что игроков выбрано именно столько, сколько нужно, можно в процессе сохранения формы на сервере (clean-методы формы или модели).

[Bjornie]

Anton Kuzmichev: да, отличная удобная форма, намного лучше полей с FK. Пока не понял как вывести игроков в list_display, пишет что manytomanyfield нельзя выводить в changelist. Писать цикл?

[Anton Kuzmichev]

Bjornie: можно кастомное "поле" создать с помощью метода класса админки, вот тут в первом ответе.