StudE-Backend/stude/student_status/serializers.py

43 lines
1.6 KiB
Python
Raw Permalink Normal View History

2023-06-26 22:06:05 +08:00
from rest_framework import serializers
from .models import StudentStatus
from subjects.models import Subject
from django.contrib.gis.geos import Point
from drf_extra_fields.geo_fields import PointField
from landmarks.models import Landmark
2023-06-26 22:06:05 +08:00
class StudentStatusSerializer(serializers.ModelSerializer):
subject = serializers.SlugRelatedField(
queryset=Subject.objects.all(), slug_field='name', required=True)
2023-07-01 16:09:20 +08:00
user = serializers.CharField(source='user.full_name', read_only=True)
location = PointField()
landmark = serializers.SlugRelatedField(
queryset=Landmark.objects.all(), many=False, slug_field='name', required=False, allow_null=True)
2023-06-27 13:40:31 +08:00
2023-06-26 22:06:05 +08:00
class Meta:
model = StudentStatus
fields = '__all__'
read_only_fields = ['user', 'landmark']
def create(self, validated_data):
user = self.context['request'].user
2023-06-27 13:40:31 +08:00
student_status = StudentStatus.objects.create(
user=user, defaults=validated_data)
return student_status
2023-07-01 16:09:20 +08:00
2023-06-27 13:40:31 +08:00
def update(self, instance, validated_data):
active = validated_data.get('active', None)
if active is not None and active is False:
validated_data['location'] = Point(0, 0)
2023-06-27 13:40:31 +08:00
validated_data['subject'] = None
validated_data['landmark'] = None
else:
# Check each landmark to see if our location is within it
for landmark in Landmark.objects.all():
if landmark.location.contains(validated_data['location']):
validated_data['landmark'] = landmark
break
2023-06-27 13:40:31 +08:00
2023-07-01 16:09:20 +08:00
return super().update(instance, validated_data)