Add routes and custom serialiser

Add requirements
This commit is contained in:
2019-04-18 21:43:22 +02:00
parent 31d5f7ade4
commit 1b1a19ae9d
6 changed files with 78 additions and 26 deletions

View File

@ -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