mirror of
https://github.com/lemeow125/StudE-Backend.git
synced 2024-11-17 06:19:24 +08:00
85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
from django.shortcuts import get_object_or_404
|
|
from django.contrib.gis.geos import Point
|
|
from requests import Response
|
|
from rest_framework import generics, viewsets, exceptions
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from django.contrib.gis.db.models.functions import Distance
|
|
from django.contrib.gis.geos import fromstr
|
|
from .models import StudentStatus
|
|
from .serializers import StudentStatusLocationSerializer, StudentStatusSerializer
|
|
from subjects.models import Subject, SubjectInstance
|
|
|
|
|
|
class StudentStatusAPIView(generics.RetrieveUpdateAPIView):
|
|
serializer_class = StudentStatusSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_object(self):
|
|
user = self.request.user
|
|
queryset = StudentStatus.objects.filter(user=user)
|
|
obj = get_object_or_404(queryset)
|
|
return obj
|
|
|
|
|
|
class ActiveStudentStatusListAPIView(generics.ListAPIView):
|
|
serializer_class = StudentStatusSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
user = self.request.user
|
|
return StudentStatus.objects.exclude(user=user).filter(active=True)
|
|
|
|
|
|
class StudentStatusListByStudentStatusLocation(generics.ListAPIView):
|
|
serializer_class = StudentStatusLocationSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
http_method_names = ['get']
|
|
|
|
def get_queryset(self):
|
|
user = self.request.user
|
|
user_status = StudentStatus.objects.filter(user=user).first()
|
|
|
|
user_location = fromstr(
|
|
user_status.location, srid=4326)
|
|
|
|
if user_status.active is False:
|
|
raise exceptions.ValidationError("Student Status is not active")
|
|
|
|
# Get names of all subjects of the user
|
|
user_subject_names = user.subjects.values_list('subject', flat=True)
|
|
|
|
# Exclude user
|
|
# Filter those only with the same subjects as the user
|
|
# Annotate the queryset with distance to the user
|
|
# Then filter so that only those within 100m remain
|
|
return StudentStatus.objects.exclude(user=user).filter(active=True).filter(
|
|
subject__name__in=user_subject_names).annotate(distance=Distance('location', user_location)).filter(distance__lte=100)
|
|
|
|
|
|
class StudentStatusListByCurrentLocation(viewsets.ViewSet):
|
|
serializer_class = StudentStatusLocationSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
http_method_names = ['post']
|
|
|
|
def create(self, request):
|
|
user = self.request.user
|
|
location_str = request.data.get('location')
|
|
|
|
# If location is not specified in request, throw error
|
|
if not location_str:
|
|
raise exceptions.ValidationError("Location is required")
|
|
|
|
# Parse user location from the POST request
|
|
user_location = fromstr(location_str, srid=4326)
|
|
|
|
# Get names of all subjects of the user
|
|
user_subject_names = user.subjects.values_list('subject', flat=True)
|
|
|
|
# Exclude user
|
|
# Filter those only with the same subjects as the user
|
|
# Annotate the queryset with distance to the user
|
|
# Then filter so that only those within 100m remain
|
|
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=100)
|
|
serializer = StudentStatusLocationSerializer(queryset, many=True)
|
|
return Response(serializer.data)
|