Django ORM多个不同和排序

Question

我在一个简单的聊天应用程序上工作。我需要查询用户的数据库以获取用户对话的最后一条消息给其他用户。与whatsapp和电报的主页相同。

型：

class CHAT(models.Model):
sender_uid = models.CharField(max_length=150, db_index=True)
receiver_uid = models.CharField(max_length=150, db_index=True)
message = models.TextField(verbose_name='Message Text')
created = models.IntegerField(default=created_time_epoch)

我尝试了这个查询：

message_list = self.model.objects.filter(Q(sender_uid=user_uid)|Q(receiver_uid=user_uid)).order_by('receiver_uid', 'sender_uid', '-created').distinct('receiver_uid', 'sender_uid')

输出：

<QuerySet [<CHAT: [e-mail] Message: hello 4 To: [e-mail]>, <CHAT: [e-mail] Message: hello6 To: [e-mail]>, <CHAT: [e-mail] Message: hello 5 To: [e-mail]>]>

我的问题是我从每个对话中收到最后两条消息(如果两个用户都互相发送消息(，其中一个用户是发送者，另一个用户是接收者。

现在我使用以下代码处理它：

message_send_list = list(self.model.objects.filter(sender_uid=user_uid).order_by('receiver_uid', '-created').distinct('receiver_uid'))
message_receive_list = list(self.model.objects.filter(receiver_uid=user_uid).order_by('sender_uid', '-created').distinct('sender_uid'))
temp_list = []
for s_message in message_send_list:
r_message = next((item for item in message_receive_list if item.sender_uid == s_message.receiver_uid), None)
if r_message is not None:
message_receive_list.pop(message_receive_list.index(r_message))
if s_message.created > r_message.created:
temp_list.append(s_message)
else:
temp_list.append(r_message)
else:
temp_list.append(s_message)
temp_list.extend(message_receive_list)

输出：

[<CHAT: [e-mail] Message: hello 5 To: [e-mail]>, <CHAT: [e-mail] Message: hello6 To: [e-mail]>]

我的问题是如何在一个查询中获得此结果？问题是用户可能是消息的发送者和接收者，我无法区分哪一个是对话的最后一条消息。如何过滤或区分？

Answer 1

根据问题的描述，你让它有点太复杂了。你可以用条件表达式 [Django-doc] 获取另一个人。因此，通过首先进行"还原"，我们将另一个人带走，然后我们可以为此使用唯一性过滤器：

from django.db.models import Case, F, When
last_messages = self.model.objects.filter(
Q(sender_uid=user_uid) | Q(receiver_uid=user_uid)
).annotate(
other=Case(
When(sender_uid=user_uid, then=F('receiver_uid')),
default=F('sender_uid'),
output_field=CharField()
)
).order_by('other', '-created').distinct('other')

此外，所有Chat对象都将有一个额外的属性：other，因此包含非user_uid端。