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

close connection when not authenticated anymore.

Browse Source
This commit is contained in:
jonaswinkler 2021-02-03 21:53:39 +01:00
parent e300ee0c92
commit c51fa60172
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/paperless/consumers.py
View File

@ -7,8 +7,11 @@ from channels.generic.websocket import WebsocketConsumer
class StatusConsumer(WebsocketConsumer): class StatusConsumer(WebsocketConsumer):
def _authenticated(self):
return 'user' in self.scope and self.scope['user'].is_authenticated
def connect(self): def connect(self):
if not self.scope['user'].is_authenticated: if not self._authenticated():
raise DenyConnection() raise DenyConnection()
else: else:
async_to_sync(self.channel_layer.group_add)( async_to_sync(self.channel_layer.group_add)(
@ -20,4 +23,7 @@ class StatusConsumer(WebsocketConsumer):
'status_updates', self.channel_name) 'status_updates', self.channel_name)
def status_update(self, event): def status_update(self, event):
self.send(json.dumps(event['data'])) if not self._authenticated():
self.close()
else:
self.send(json.dumps(event['data']))