mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-05-05 11:39:29 -05:00
31 lines
911 B
Python
31 lines
911 B
Python
import json
|
|
|
|
from asgiref.sync import async_to_sync
|
|
from channels.exceptions import DenyConnection, AcceptConnection
|
|
from channels.generic.websocket import WebsocketConsumer
|
|
|
|
|
|
class StatusConsumer(WebsocketConsumer):
|
|
def _authenticated(self):
|
|
return "user" in self.scope and self.scope["user"].is_authenticated
|
|
|
|
def connect(self):
|
|
if not self._authenticated():
|
|
raise DenyConnection()
|
|
else:
|
|
async_to_sync(self.channel_layer.group_add)(
|
|
"status_updates", self.channel_name
|
|
)
|
|
raise AcceptConnection()
|
|
|
|
def disconnect(self, close_code):
|
|
async_to_sync(self.channel_layer.group_discard)(
|
|
"status_updates", self.channel_name
|
|
)
|
|
|
|
def status_update(self, event):
|
|
if not self._authenticated():
|
|
self.close()
|
|
else:
|
|
self.send(json.dumps(event["data"]))
|