mirror of
https://github.com/lemeow125/StudE-Backend.git
synced 2024-11-17 06:19:24 +08:00
129 lines
4.6 KiB
Python
129 lines
4.6 KiB
Python
from django.shortcuts import render
|
|
from rest_framework import generics, mixins
|
|
from rest_framework.exceptions import PermissionDenied
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from .serializers import StudyGroupSerializer, StudyGroupCreateSerializer, StudyGroupDistanceSerializer, StudyGroupDetailSerializer, CustomUserAvatarSerializer
|
|
from .models import StudyGroup
|
|
from subjects.models import Subject, SubjectInstance
|
|
from student_status.models import StudentStatus
|
|
from rest_framework import generics, viewsets, exceptions
|
|
from django.contrib.gis.geos import fromstr
|
|
from django.contrib.gis.db.models.functions import Distance
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework import permissions
|
|
from accounts.models import CustomUser
|
|
|
|
# Create your views here.
|
|
|
|
|
|
class StudyGroupAvatarsView(generics.ListAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = CustomUserAvatarSerializer
|
|
queryset = CustomUser.objects.all()
|
|
|
|
def get_queryset(self):
|
|
# Get user
|
|
user = self.request.user
|
|
|
|
# Get status of user
|
|
user_status = StudentStatus.objects.filter(user=user).first()
|
|
|
|
# Return error if not in study group
|
|
if (not user_status.study_group):
|
|
raise exceptions.ValidationError("You are not in a study group")
|
|
|
|
# Get study group of user
|
|
user_study_group = user_status.study_group
|
|
|
|
# Get students in that study group
|
|
study_group_student_ids = user_study_group.students.values_list(
|
|
'user', flat=True)
|
|
|
|
# Then get user instances of those students
|
|
study_group_users = CustomUser.objects.filter(
|
|
id__in=study_group_student_ids)
|
|
|
|
# Return to be parsed by serializer
|
|
return study_group_users
|
|
|
|
|
|
class StudyGroupMembersAvatarView(generics.ListAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = StudyGroupCreateSerializer
|
|
queryset = CustomUser.objects.all()
|
|
|
|
|
|
class StudyGroupCreateView(viewsets.ModelViewSet):
|
|
http_method_names = ['patch', 'post', 'delete']
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = StudyGroupCreateSerializer
|
|
queryset = StudyGroup.objects.all()
|
|
|
|
|
|
class StudyGroupDetailView(generics.RetrieveAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = StudyGroupDetailSerializer
|
|
queryset = StudyGroup.objects.all()
|
|
lookup_field = 'name'
|
|
|
|
|
|
class StudyGroupListView(generics.ListAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = StudyGroupSerializer
|
|
queryset = StudyGroup.objects.all()
|
|
|
|
def get_queryset(self):
|
|
user = self.request.user
|
|
|
|
# Get the user's course
|
|
user_course = user.course
|
|
print(user_course)
|
|
|
|
# Get the subject name of the student's subjects from SubjectInstance
|
|
user_subject_names = user.subjects.values_list('subject', flat=True)
|
|
print('user subjects:', user_subject_names)
|
|
# Get the corresponding Subject models
|
|
user_subjects = Subject.objects.filter(name__in=user_subject_names)
|
|
# Now fetch the StudyGroups with the matching Subject models
|
|
studygroups = StudyGroup.objects.filter(subject__in=user_subjects)
|
|
|
|
return studygroups
|
|
|
|
|
|
class StudyGroupListNearView(generics.ListAPIView):
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = StudyGroupDistanceSerializer
|
|
queryset = StudyGroup.objects.all()
|
|
|
|
def get_queryset(self):
|
|
|
|
user = self.request.user
|
|
user_status = StudentStatus.objects.filter(user=user).first()
|
|
|
|
if user_status.active is False:
|
|
raise exceptions.ValidationError("Student Status is not active")
|
|
|
|
user_location = fromstr(
|
|
user_status.location, srid=4326)
|
|
|
|
# Get the user's course
|
|
user_course = user.course
|
|
print(user_course)
|
|
|
|
# Get the subject name of the student's subjects from SubjectInstance
|
|
user_subject_names = user.subjects.values_list('subject', flat=True)
|
|
# Get the corresponding Subject models
|
|
user_subjects = Subject.objects.filter(name__in=user_subject_names)
|
|
|
|
# Now fetch the StudyGroups with the matching Subject models that are within 100m
|
|
studygroups = StudyGroup.objects.filter(subject__in=user_subjects).annotate(
|
|
distance=Distance('location', user_location)).filter(distance__lte=100)
|
|
|
|
return studygroups
|
|
|
|
|
|
class IsOwnerOrReadOnly(permissions.BasePermission):
|
|
def has_object_permission(self, request, view, obj):
|
|
# Check if the requesting user's student status matches the user field
|
|
return obj.user == request.user.studentstatus
|