Files
polylan-submitter/polylan_submitter/submissions/consumers.py
T

35 lines
1.2 KiB
Python

import json
from channels.generic.websocket import AsyncWebsocketConsumer
from django.contrib.auth.models import AnonymousUser
class SubmissionsConsumer(AsyncWebsocketConsumer):
async def connect(self):
self.user = self.scope["user"]
if isinstance(self.user, AnonymousUser):
await self.close()
return
self.user_group_name = f"submissions_user_{self.user.id}"
await self.channel_layer.group_add(self.user_group_name, self.channel_name)
await self.accept()
async def disconnect(self, close_code):
await self.channel_layer.group_discard(self.user_group_name, self.channel_name)
async def submission_updated(self, event):
submission_data = event["submission"]
await self.send(text_data=json.dumps(submission_data))
async def submission_file_updated(self, event):
file_data = event["file"]
await self.send(text_data=json.dumps(file_data))
async def submission_crop_updated(self, event):
crop_data = event["crop"]
await self.send(text_data=json.dumps(crop_data))
async def scope_updated(self, event):
await self.send(text_data=json.dumps({"scope": event["scope"]}))