Django-ORM多重distinct和order by

2024-10-01 09:42:21 发布

您现在位置:Python中文网/ 问答频道 /正文

我在开发一个简单的聊天软件。我需要查询数据库中的一个用户,以获取用户对话的最后一条消息给其他用户。与whatsapp和telegram的主页相同。你知道吗

型号:

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: ted@ted.com Message: hello 4 To: saeed@saeed.com>, <CHAT: marshal@marshal.com Message: hello6 To: saeed@saeed.com>, <CHAT: saeed@saeed.com Message: hello 5 To: ted@ted.com>]>

我的问题是我从每个会话中得到最后两条消息(如果两个用户都互相发送消息),其中一个用户是发送方,另一个用户是接收方。你知道吗

现在我用下面的代码来处理它:

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: saeed@saeed.com Message: hello 5 To: ted@ted.com>, <CHAT: marshal@marshal.com Message: hello6 To: saeed@saeed.com>]

我的问题是如何在一个查询中得到这个结果?问题是用户可以是消息的发送者和接收者,我无法区分哪一个是会话的最后一条消息。如何过滤或区分?你知道吗


Tags: to用户com消息messageuidmodelschat
1条回答
网友
1楼 · 发布于 2024-10-01 09:42:21

根据对问题的描述,你把它弄得有点太复杂了。您可以通过conditional expression [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端。你知道吗

相关问题 更多 >