root: add Channels Message Storage and consumer
This commit is contained in:
34
passbook/root/messages/storage.py
Normal file
34
passbook/root/messages/storage.py
Normal file
@ -0,0 +1,34 @@
|
||||
"""Channels Messages storage"""
|
||||
from asgiref.sync import async_to_sync
|
||||
from channels.layers import get_channel_layer
|
||||
from django.contrib.messages.storage.base import BaseStorage, Message
|
||||
from django.core.cache import cache
|
||||
from django.http.request import HttpRequest
|
||||
|
||||
|
||||
class ChannelsStorage(BaseStorage):
|
||||
"""Send contrib.messages over websocket"""
|
||||
|
||||
def __init__(self, request: HttpRequest) -> None:
|
||||
# pyright: reportGeneralTypeIssues=false
|
||||
super().__init__(request)
|
||||
self.channel = get_channel_layer()
|
||||
|
||||
def _store(self, messages: list[Message], response, *args, **kwargs):
|
||||
prefix = f"user_{self.request.user.pk}_"
|
||||
keys = cache.keys(f"{prefix}*")
|
||||
for key in keys:
|
||||
uid = key.replace(prefix, "")
|
||||
for message in messages:
|
||||
async_to_sync(self.channel.send)(
|
||||
uid,
|
||||
{
|
||||
"type": "event.update",
|
||||
"levelTag": message.level_tag,
|
||||
"tags": message.tags,
|
||||
"message": message.message,
|
||||
},
|
||||
)
|
||||
|
||||
def _get(self, *args, **kwargs):
|
||||
return [], True
|
||||
Reference in New Issue
Block a user