paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

websocket authentication

Browse Source
This commit is contained in:
jonaswinkler 2021-02-03 20:44:10 +01:00
parent c36fe577de
commit ba3c8308b3
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/paperless/consumers.py
View File

@ -1,14 +1,19 @@
import json import json
from asgiref.sync import async_to_sync from asgiref.sync import async_to_sync
from channels.exceptions import DenyConnection, AcceptConnection
from channels.generic.websocket import WebsocketConsumer from channels.generic.websocket import WebsocketConsumer
class StatusConsumer(WebsocketConsumer): class StatusConsumer(WebsocketConsumer):
def connect(self): def connect(self):
self.accept() if not self.scope['user'].is_authenticated:
async_to_sync(self.channel_layer.group_add)( raise DenyConnection()
'status_updates', self.channel_name) else:
async_to_sync(self.channel_layer.group_add)(
'status_updates', self.channel_name)
raise AcceptConnection()
def disconnect(self, close_code): def disconnect(self, close_code):
async_to_sync(self.channel_layer.group_discard)( async_to_sync(self.channel_layer.group_discard)(