Add routes and custom serialiser
Add requirements
This commit is contained in:
@ -1,5 +1,4 @@
|
||||
from rest_framework import serializers
|
||||
from django.contrib.auth.models import User
|
||||
|
||||
from .models import Driver, Kart, Params, Relay
|
||||
|
||||
@ -41,6 +40,23 @@ class KartSerializer(serializers.ModelSerializer):
|
||||
return Kart.objects.create(**validated_data)
|
||||
|
||||
|
||||
class KartSimpleSerializer(serializers.ModelSerializer):
|
||||
|
||||
class Meta:
|
||||
model = Kart
|
||||
fields = ('id', 'name', 'autonomy')
|
||||
|
||||
def update(self, instance, validated_data):
|
||||
instance.name = validated_data.get("name", instance.name)
|
||||
instance.autonomy = validated_data.get("autonomy", instance.autonomy)
|
||||
instance.comment = validated_data.get("comment", instance.comment)
|
||||
instance.save()
|
||||
return instance
|
||||
|
||||
def create(self, validated_data):
|
||||
return Kart.objects.create(**validated_data)
|
||||
|
||||
|
||||
class RelaySerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Relay
|
||||
@ -66,12 +82,13 @@ class RelaySerializer(serializers.ModelSerializer):
|
||||
class ParamSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Params
|
||||
fields = ('id', 'default_relay', 'autonomy', 'stand_minimum_time')
|
||||
fields = ('id', 'default_relay', 'autonomy', 'stand_minimum_time', 'comment')
|
||||
|
||||
def update(self, instance, validated_data):
|
||||
instance.default_relay = validated_data.get("default_relay", instance.default_relay)
|
||||
instance.autonomy = validated_data.get("autonomy", instance.autonomy)
|
||||
instance.stand_minimum_time = validated_data.get("stand_minimum_time", instance.stand_minimum_time)
|
||||
instance.comment = validated_data.get("comment", instance.comment)
|
||||
instance.save()
|
||||
return instance
|
||||
|
||||
|
Reference in New Issue
Block a user