mirror of
https://github.com/lemeow125/StudE-Backend.git
synced 2024-11-17 06:19:24 +08:00
98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
# consumers.py
|
|
import json
|
|
from .models import StudentStatus
|
|
from .serializers import StudentStatusSerializer
|
|
from djangochannelsrestframework.generics import GenericAsyncAPIConsumer
|
|
from djangochannelsrestframework.decorators import action
|
|
from djangochannelsrestframework.observer import model_observer, observer
|
|
from channels.db import database_sync_to_async
|
|
import asyncio
|
|
from djangochannelsrestframework.mixins import (
|
|
ListModelMixin,
|
|
RetrieveModelMixin,
|
|
)
|
|
from djangochannelsrestframework.permissions import IsAuthenticated
|
|
from channels.layers import get_channel_layer
|
|
from asgiref.sync import async_to_sync
|
|
from django.contrib.gis.db.models.functions import Distance
|
|
from django.contrib.gis.geos import fromstr
|
|
from .models import StudentStatus
|
|
from accounts.models import CustomUser
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from rest_framework import generics, viewsets, exceptions
|
|
from channels.exceptions import DenyConnection
|
|
|
|
|
|
class StudentStatusConsumer(
|
|
ListModelMixin,
|
|
RetrieveModelMixin,
|
|
GenericAsyncAPIConsumer,
|
|
):
|
|
permission_classes = [IsAuthenticated]
|
|
queryset = StudentStatus.objects.none()
|
|
serializer_class = StudentStatusSerializer
|
|
|
|
async def send_status_update(self, event):
|
|
data = event['data']
|
|
await self.send(text_data=json.dumps(data))
|
|
|
|
async def websocket_connect(self, message):
|
|
if isinstance(self.scope['user'], AnonymousUser):
|
|
await self.close()
|
|
else:
|
|
student_status_isactive = await self.get_student_status()
|
|
if not student_status_isactive:
|
|
await self.close()
|
|
await self.channel_layer.group_add('student_status_group', self.channel_name)
|
|
await self.accept()
|
|
self.send_updates_task = asyncio.create_task(self.send_updates())
|
|
|
|
async def websocket_disconnect(self, message):
|
|
# ...
|
|
if not isinstance(self.scope['user'], AnonymousUser):
|
|
await self.channel_layer.group_discard('student_status_group', self.channel_name)
|
|
self.send_updates_task.cancel()
|
|
|
|
@database_sync_to_async
|
|
def get_student_status(self):
|
|
user = self.scope["user"]
|
|
user_status = StudentStatus.objects.filter(user=user).first()
|
|
return user_status.active
|
|
|
|
@database_sync_to_async
|
|
def get_student_statuses(self):
|
|
user = self.scope['user']
|
|
user_status = StudentStatus.objects.filter(user=user).first()
|
|
user_status_active = StudentStatus.objects.filter(
|
|
user=user).values_list('active').first()
|
|
user_location = fromstr(
|
|
user_status.location, srid=4326)
|
|
|
|
if user_status.active is False:
|
|
queryset = StudentStatus.objects.none()
|
|
return StudentStatusSerializer(queryset, many=True).data
|
|
|
|
user_subject_names = user.subjects.values_list('subject', flat=True)
|
|
queryset = StudentStatus.objects.exclude(user=user).filter(active=True).filter(subject__name__in=user_subject_names).annotate(
|
|
distance=Distance('location', user_location)).filter(distance__lte=50)
|
|
return StudentStatusSerializer(queryset, many=True).data
|
|
|
|
async def send_updates(self):
|
|
channel_layer = get_channel_layer()
|
|
|
|
while True:
|
|
try:
|
|
# print('attempting to get')
|
|
data = await self.get_student_statuses()
|
|
# print(f"Sending update: {data}")
|
|
await channel_layer.group_send(
|
|
'student_status_group',
|
|
{
|
|
'type': 'send_status_update',
|
|
'data': data,
|
|
}
|
|
)
|
|
await asyncio.sleep(3)
|
|
except Exception as e:
|
|
print(f"Exception in send_updates: {e}")
|
|
break # Break the loop on error
|