mirror of
https://github.com/lemeow125/StudE-Backend.git
synced 2024-11-17 06:19:24 +08:00
68 lines
2.7 KiB
Python
68 lines
2.7 KiB
Python
from rest_framework import serializers
|
|
from .models import StudyGroup
|
|
from accounts.models import CustomUser
|
|
from subjects.models import Subject
|
|
from drf_extra_fields.geo_fields import PointField
|
|
from landmarks.models import Landmark
|
|
|
|
|
|
class CustomUserKeyRelatedField(serializers.PrimaryKeyRelatedField):
|
|
|
|
def to_representation(self, value):
|
|
# returns the string representation of the custom user (aka the name)
|
|
return str(value)
|
|
|
|
|
|
class StudyGroupSerializer(serializers.ModelSerializer):
|
|
name = serializers.CharField()
|
|
students = serializers.StringRelatedField(many=True)
|
|
subject = serializers.SlugRelatedField(
|
|
many=False, slug_field='name', queryset=Subject.objects.all(), required=True, allow_null=False)
|
|
location = PointField()
|
|
landmark = serializers.SlugRelatedField(
|
|
queryset=Landmark.objects.all(), many=False, slug_field='name', required=False, allow_null=True)
|
|
radius = serializers.FloatField()
|
|
|
|
class Meta:
|
|
model = StudyGroup
|
|
fields = '__all__'
|
|
read_only_fields = ['landmark', 'radius', 'students']
|
|
|
|
|
|
class FullNameSlugRelatedField(serializers.SlugRelatedField):
|
|
def to_representation(self, instance):
|
|
return instance.full_name
|
|
|
|
|
|
class StudyGroupCreateSerializer(serializers.ModelSerializer):
|
|
name = serializers.CharField()
|
|
subject = serializers.SlugRelatedField(
|
|
many=False, slug_field='name', queryset=Subject.objects.all(), required=True, allow_null=False)
|
|
location = PointField()
|
|
landmark = serializers.SlugRelatedField(
|
|
queryset=Landmark.objects.all(), many=False, slug_field='name', required=False, allow_null=True)
|
|
|
|
def create(self, validated_data):
|
|
user = self.context['request'].user
|
|
study_group = StudyGroup.objects.create(
|
|
name=validated_data['name'], location=validated_data['location'], subject=validated_data['subject'])
|
|
for landmark in Landmark.objects.all():
|
|
if landmark.location.contains(validated_data['location']):
|
|
validated_data['landmark'] = landmark
|
|
break
|
|
validated_data['location'].read_only = True
|
|
return study_group
|
|
|
|
def update(self, instance, validated_data):
|
|
# Check each landmark to see if our location is within it
|
|
if ('location' in validated_data):
|
|
for landmark in Landmark.objects.all():
|
|
if landmark.location.contains(validated_data['location']):
|
|
validated_data['landmark'] = landmark
|
|
break
|
|
return super().update(instance, validated_data)
|
|
|
|
class Meta:
|
|
model = StudyGroup
|
|
fields = '__all__'
|
|
read_only_fields = ['landmark']
|