Джанго опрашивает модель, которая участвует в родовых отношениях - PullRequest
0 голосов
/ 19 ноября 2018

Здесь у меня 2 модели, Компания и UserSupervision . Компания связана как внешний ключ с общей моделью отношений UserSupervision .

Я хочу получить запрос следующим образом:

Company.objects.filter(**{'usersupervision__content_type': Company,
                          'usersupervision__user': some_user,
                          'usersupervision__date_cancelled': None})

Итак, я хочу, чтобы все компании some_user появились в UserSupervision модели. Я знаю, что Company и UserSupervision не имеют прямого отношения, поэтому я подумал, может быть, мне нужно использовать raw sql с соединениями? Или любое другое более подходящее решение?

class Company(models.Model):
    id = models.UUIDField(blank=True, editable=False, primary_key=True, default=uuid.uuid4)
    name = models.CharField(max_length=150, unique=True)

    users = models.ManyToManyField(User, related_name='companies', through='CompanyMembership')
    date_created = models.DateTimeField('created date', blank=True, editable=False, auto_now_add=True)
    date_updated = models.DateTimeField('updated date', blank=True, editable=False, auto_now=True)


class UserSupervision(models.Model):
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)
    supervision = models.ForeignKey(Supervision, on_delete=models.SET_NULL, null=True)

    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.UUIDField()
    content_object = GenericForeignKey()

    date_assigned = models.DateTimeField('assigned date', blank=True, editable=False, auto_now_add=True)
    date_cancelled = models.DateTimeField('cancelled date', blank=True, null=True, default=None)

1 Ответ

0 голосов
/ 19 ноября 2018

Итак, вы хотите найти все компании в UserSupervision для определенного пользователя UserSupervision.user?Что-то вроде:

user_supervisions = UserSupervision.objects.filter(
    content_type=ContentType.objects.get_for_model(Company),
    user=some_user,
    date_cancelled=None
).order_by('content_type_id', 'object_id').distinct('content_type_id', 'object_id')

companies = [us.object_id for us in usersupervisions]  # Loop to create list of company ids.

Вы можете отбросить order_by и различить , если вы не используете PostgreSQL, но цикл компаний должен быть изменен для учета дублирующихся записей.

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