Один потребитель веб-сокетов Django Channels используется на всем сайте? - PullRequest
Новая
       18

Один потребитель веб-сокетов Django Channels используется на всем сайте?

1 голос
/ 12 мая 2019

У меня есть проект Django, который использует WebSockets и одного потребителя (ChatConsumer) как для части чата, так и для уведомлений приложения.

У меня routing.py установлено значение url(r"^messages/(?P<username>[\w.@+-]+)", ChatConsumer), однако, поскольку уведомления также зависят от веб-сокета, к ним необходимо обращаться с любой страницы сайта.

Причина в том, что когда пользователь нажимает на уведомление, оно помечается как read с использованием

socket.send(JSON.stringify(data));

В настоящее время уведомления работают только тогда, когда пользовательна /messages/<username>/ URL.

Если я изменю routing.py на весь сайт, (url(r"^", ChatConsumer)), я, очевидно, получаю проблему 

File "./consumers.py" in websocket_connect
    other_user = self.scope['url_route']['kwargs']['username']
  'username'

Есть ли простоеспособ решить это?Потому что поправьте меня, если я ошибаюсь, но я не думаю, что писать нового потребителя уместно, так как уведомления и чат тесно переплетены?

consumer.py 

class ChatConsumer(AsyncConsumer):
    async def websocket_connect(self, event):

        other_user = self.scope['url_route']['kwargs']['username']
        me = self.scope['user']
        thread_obj = await self.get_thread(me, other_user)
        self.thread_obj = thread_obj
        chat_room = f"thread_{thread_obj.id}"
        self.chat_room = chat_room
        # below creates the chatroom
        await self.channel_layer.group_add(
            chat_room,
            self.channel_name
        )

        await self.send({
            "type": "websocket.accept"
        })

    async def websocket_receive(self, event):
        # when a message is recieved from the websocket
        print("receive", event)

        message_type = json.loads(event.get('text','{}')).get('type')
        if message_type == "notification_read":
            user = self.scope['user']
            username = user.username if user.is_authenticated else 'default'
            # Update the notification read status flag in Notification model.
            notification = Notification.objects.filter(notification_user=user).update(notification_read=True)
            print("notification read")
            return

        front_text = event.get('text', None)
        if front_text is not None:
            loaded_dict_data = json.loads(front_text)
            msg =  loaded_dict_data.get('message')
            user = self.scope['user']
            username = user.username if user.is_authenticated else 'default'
            notification_id = 'default'
            myResponse = {
                'message': msg,
                'username': username,
                'notification': notification_id,
            }
            print(myResponse)
            await self.create_chat_message(user, msg)
            other_user = self.scope['url_route']['kwargs']['username']
            other_user = User.objects.get(username=other_user)
            await self.create_notification(other_user, msg)

            # broadcasts the message event to be sent, the group send layer
            # triggers the chat_message function for all of the group (chat_room)
            await self.channel_layer.group_send(
                self.chat_room,
                {
                    'type': 'chat_message',
                    'text': json.dumps(myResponse)
                }
            )

    # chat_method is a custom method name that we made
    async def chat_message(self, event):
        # sends the actual message
        await self.send({
                'type': 'websocket.send',
                'text': event['text']
        })

    async def websocket_disconnect(self, event):
        # when the socket disconnects
        print('disconnected', event)

    @database_sync_to_async
    def get_thread(self, user, other_username):
        return Thread.objects.get_or_new(user, other_username)[0]

    @database_sync_to_async
    def create_chat_message(self, me, msg):
        thread_obj = self.thread_obj
        return ChatMessage.objects.create(thread=thread_obj, user=me, message=msg)

    @database_sync_to_async
    def create_notification(self, other_user, msg):
        last_chat = ChatMessage.objects.latest('id')
        created_notification = Notification.objects.create(notification_user=other_user, notification_chat=last_chat)
        return created_notification

base.html

<script>
    $(document).ready(function() {
      // $('div[id^="notification-"]')
      $("#notificationLink").click(function() {
        $('span[id^="notification"]').each(function() {
          var username = '{{ request.user.username }}'
          var data = {
            "type": "notification_read",
            "username": username,
          }
          socket.send(JSON.stringify(data));
        });
    });
  </script>

<script>
...
    var endpoint = wsStart + loc.host + loc.pathname
    var socket = new ReconnectingWebSocket(endpoint)
    // all the websocket scripts - client side*
...
</script>

1 Ответ

0 голосов
/ 12 мая 2019

Я думаю, вам следует отправить other_user с сообщением, чтобы оно было таким: 

 msg =  loaded_dict_data.get('message')
 other_user =  loaded_dict_data.get('other_user')

, чтобы вам больше не нужно было использовать self.scope ['url_route'] ['kwargs'] ['username']
Кстати, вы можете поделиться кодом на github, потому что я готовлюсь сделать что-то подобное, и я думаю, что ваш проект мне очень поможет

...