Django: запрос для получения User.usernames через внешний ключ

Рассмотрим следующую модель дружбы, в которой для каждой установленной дружбы есть Пользователь, отправивший запрос на добавление в друзья, и Пользователь, который получил запрос на добавление в друзья:

from django.contrib.auth.models import User

class Friendship(models.Model):
    sender = models.ForeignKey(User)
    receiver = models.ForeignKey(User)

Как составить запрос, который даст мне все имена пользователей пользователей, которые находятся в дружеских отношениях, скажем, с пользователем 123?

Имейте в виду, что user123 может быть отправителем или получателем любой дружбы, в которой он состоит.

Кроме того, я хочу, чтобы возвращались только значения User.username, а не весь объект User.

На данный момент у меня есть эта неуклюжая половина/решение (я использую list(), чтобы я мог добавить как получающих друзей пользователя 123, так и отправляющих друзей пользователя 123, чтобы в конечном итоге создать список все друзья). Проблема в том, что я получаю все объекты User, и мне нужны только имена пользователей... и должен быть лучший способ сделать это.

friends_a = list(Friendship.objects.filter(sender=user123).values('receiver'))
friends_b = list(Friendship.objects.filter(receiver=user123).values('sender'))
friends_a = [] if not friends_a else friends_a[0].values()
friends_b = [] if not friends_b else friends_b[0].values()
all_friends = friends_a + friends_b

python django foreign-keys django-models
person sgarza62    schedule 13.01.2013    source источник

Ответы (1)


arrow_upward
5
arrow_downward

Q Objects для сложных поисков.

Вау, это удивительно головокружительно, чтобы получить желаемый набор результатов, поскольку на самом деле есть 2 разных логических запроса. Самый простой способ - это... написать это с помощью python.

user = User.objects.get(username='user123')
friendships = Friendship.objects.filter(Q(receiver=user) | Q(sender=user))

usernames = []

for friendship in friendships:
    if friendship.receiver == user:
        usernames.append(friendship.receiver.username)
    else:
        usernames.append(friendship.sender.username)
person Yuji 'Tomita' Tomita    schedule 13.01.2013
comment
Черт, молодец. Надеялся, что смогу добиться этого с помощью одного сложного запроса, но это определенно выполняет свою работу. - person sgarza62; 14.01.2013