mirror of
https://code.eliotberriot.com/funkwhale/funkwhale.git
synced 2025-10-03 22:39:16 +02:00
20 lines
622 B
Python
20 lines
622 B
Python
from channels.generic.websocket import JsonWebsocketConsumer
|
|
|
|
from funkwhale_api.common import channels
|
|
|
|
|
|
class JsonAuthConsumer(JsonWebsocketConsumer):
|
|
def connect(self):
|
|
try:
|
|
assert self.scope["user"].pk is not None
|
|
except (AssertionError, AttributeError, KeyError):
|
|
return self.close()
|
|
|
|
return self.accept()
|
|
|
|
def accept(self):
|
|
super().accept()
|
|
for group in self.groups:
|
|
channels.group_add(group, self.channel_name)
|
|
for group in self.scope["user"].get_channels_groups():
|
|
channels.group_add(group, self.channel_name)
|