新的聊天消息通知 Django 频道 [英] New chat message notification Django Channels

问题描述

我已经按照教程在我的 Django 应用程序中设置了 Django Channels 2.1.2，现在需要为新消息设置通知系统.我想以最简单的方式做到这一点.

我可以通过浏览器推送通知来做到这一点，但我不想那样做.我希望它像 Stack Overflow 一样，其中有一个红色数字代表新消息的实例.

这里的一个答案说

<块引用>

对于通知，您只需要两个模型:User 和 Notification.在connect 将范围设置为当前经过身份验证的用户.设置一个post_save 信号在您的 Notification 模型上触​​发消费者向通知对象的用户发送消息的方法.——

我正在努力思考这会是什么样子，我已经有一个 User 模型，但没有 Notification 模型.

聊天仅在 2 个用户之间进行，它不是聊天室，而是更多的聊天线程.这两个 html 模板是 inbox.html 和 thread.html

感谢任何帮助！

我的 Django 频道代码如下！

consumers.py

class ChatConsumer(AsyncConsumer):async def websocket_connect(self, event):打印('连接'，事件)other_user = self.scope['url_route']['kwargs']['username']me = self.scope['user']#print(other_user, 我)thread_obj = await self.get_thread(me, other_user)self.thread_obj = thread_objchat_room = f"thread_{thread_obj.id}"self.chat_room = chat_room# 下面创建聊天室等待 self.channel_layer.group_add(聊天室，self.channel_name)等待 self.send({类型":websocket.accept"})async def websocket_receive(self, event):# 当收到来自 websocket 的消息时打印(接收"，事件)message_type = event.get('type', None) #检查消息类型，采取相应措施如果 message_type == "notification_read":# 更新通知模型中的通知读取状态标志.通知 = Notification.object.get(id=notification_id)notification.notification_read = True通知.save() #提交到数据库打印(通知已读")front_text = event.get('text', None)如果 front_text 不是 None:Loaded_dict_data = json.loads(front_text)msg = loaded_dict_data.get('message')用户 = self.scope['用户']用户名 = '默认'如果 user.is_authenticated:用户名 = 用户.用户名我的回应 = {'消息':消息，'用户名':用户名，}等待 self.create_chat_message(user, msg)# 广播要发送的消息事件，群发层# 为所有群组(chat_room)触发chat_message功能等待 self.channel_layer.group_send(self.chat_room,{'type': 'chat_message','文本':json.dumps(myResponse)})# chat_method 是我们自定义的方法名异步 def chat_message(self, event):# 发送实际消息等待 self.send({'type': 'websocket.send','文本':事件['文本']})async def websocket_disconnect(self, event):# 当socket断开时打印('断开连接'，事件)@database_sync_to_asyncdef get_thread(self, user, other_username):返回 Thread.objects.get_or_new(user, other_username)[0]@database_sync_to_asyncdef create_chat_message(self, me, msg):thread_obj = self.thread_obj返回 ChatMessage.objects.create(thread=thread_obj, user=me, message=msg)

经理

class ThreadManager(models.Manager):def by_user(self, user):qlookup = Q(first=user) |Q(第二个=用户)qlookup2 = Q(first=user) &Q(第二个=用户)qs = self.get_queryset().filter(qlookup).exclude(qlookup2).distinct()返回 qs# 为2个用户抓取线程的方法def get_or_new(self, user, other_username): # get_or_create用户名 = 用户.用户名如果用户名 == 其他用户名:返回无，无# 看起来基于任一用户名qlookup1 = Q(first__username=username) &Q(second__username=other_username)qlookup2 = Q(first__username=other_username) &Q(second__username=username)qs = self.get_queryset().filter(qlookup1 | qlookup2).distinct()如果 qs.count() == 1:返回 qs.first(), Falseelif qs.count() >1:返回 qs.order_by('timestamp').first(), False别的:类 = user.__class__尝试:user2 = Klass.objects.get(username=other_username)除了 Klass.DoesNotExist:用户 2 = 无如果用户 != 用户 2:obj = self.model(第一个=用户，第二=用户2)obj.save()返回对象，真返回无，假

models.py

class Thread(models.Model):first = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE,related_name='chat_thread_first')second = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE,related_name='chat_thread_second')更新 = models.DateTimeField(auto_now=True)时间戳 =models.DateTimeField(auto_now_add=True)对象 = 线程管理器()def __str__(self):返回 f'{self.id}'@财产def room_group_name(self):返回 f'chat_{self.id}'定义广播(自我，味精=无):如果 msg 不是 None:broadcast_msg_to_chat(msg, group_name=self.room_group_name, user='admin')返回真返回错误类 ChatMessage(models.Model):线程=models.ForeignKey(线程，空=真，空白=真，on_delete=models.SET_NULL)user = models.ForeignKey(settings.AUTH_USER_MODEL,verbose_name='sender', on_delete=models.CASCADE)消息 = 模型.TextField()时间戳 =models.DateTimeField(auto_now_add=True)def __str__(self):返回 f'{self.id}'类通知(模型.模型):Notification_user = models.ForeignKey(用户，on_delete=models.CASCADE)notification_chat = models.ForeignKey(ChatMessage, on_delete=models.CASCADE)notification_read = models.BooleanField(默认=False)def __str__(self):返回 f'{self.id}'

views.py

class InboxView(LoginRequiredMixin, ListView):template_name = '聊天/收件箱.html'context_object_name = '线程'def get_queryset(self):返回 Thread.objects.by_user(self.request.user).exclude(chatmessage__isnull=True).order_by('timestamp')# by_user(self.request.user)类 ThreadView(LoginRequiredMixin, FormMixin, DetailView):template_name = 'chat/thread.html'form_class = ComposeForm成功网址 = '#'def get_queryset(self):返回 Thread.objects.by_user(self.request.user)def get_object(self):other_username = self.kwargs.get("用户名")obj, created = Thread.objects.get_or_new(self.request.user, other_username)如果 obj == 无:提高Http404返回对象def get_context_data(self, **kwargs):context = super().get_context_data(**kwargs)context['form'] = self.get_form()返回上下文def post(self, request, *args, **kwargs):如果不是 request.user.is_authenticated:返回 HttpResponseForbidden()self.object = self.get_object()form = self.get_form()如果 form.is_valid():返回 self.form_valid(form)别的:返回 self.form_invalid(form)def form_valid(self, form):线程 = self.get_object()用户 = self.request.usermessage = form.cleaned_data.get("message")ChatMessage.objects.create(用户=用户，线程=线程，消息=消息)返回 super().form_valid(form)

thread.html

{% 区块头 %}<title>聊天</title><script src="{% static '/channels/js/websocketbridge.js' %}" type="text/javascript"></script>{% 结束块 %}{% 块内容 %}<脚本>$(#notification-element).on("点击", function(){数据 = {类型":notification_read"，用户名":用户名，notification_id":notification_id}；socket.send(JSON.stringify(data));});<!-- 返回带有通知示例的收件箱按钮--><a class="btn btn-light" id="notification_id" href="{% url 'chat:inbox' %}">返回收件箱</a><div class="msg_history">{% 用于 object.chatmessage_set.all 中的聊天 %}{% if chat.user == user %}<div class="outgoing_msg"><div class="outgoing_msg_img"><img src="{{ chat.user.profile.image.url }}">

<div class="sent_msg"><p>{{ chat.message }}</p><span class="time_date">{{ chat.timestamp }}</span>

{％ 别的 ％}<div class="incoming_msg"><div class="incoming_msg_img"><img src="{{ chat.user.profile.image.url }}">

<div class="received_msg"><div class="received_withd_msg"><p>{{ chat.message }}</p><span class="time_date">{{ chat.timestamp }}</span>