from rest_framework import serializers


class FunctionSerializer(serializers.Serializer):
    function = serializers.IntegerField(required=True)
    state = serializers.IntegerField(required=True)


class CabSerializer(serializers.Serializer):
    speed = serializers.IntegerField(required=True)
    direction = serializers.IntegerField(required=True)


class InfraSerializer(serializers.Serializer):
    power = serializers.BooleanField(required=True)
    track = serializers.ChoiceField(
        choices=("main", "prog", "join", "MAIN", "PROG", "JOIN"),
        required=False,
    )