from django.contrib.auth.models import User, Group
from rest_framework import serializers

from.models import Vehicles_infos, Vehicles



class cl_Vehicles_infos_Serializer(serializers.ModelSerializer):
    class Meta:
        model = Vehicles_infos
        fields = '__all__'

class cl_Vehicles_Serializer(serializers.ModelSerializer):
    technique_infos = serializers.SerializerMethodField()
    matos_infos = serializers.SerializerMethodField()
    perempt_infos = serializers.SerializerMethodField()
    
    class Meta:
        model = Vehicles
        fields = '__all__'
    
    def get_technique_infos(self, obj):
        enabled_infos = Vehicles_infos.objects.filter(Vehicle=obj, bEnabled=True, nType=1)
        return cl_Vehicles_infos_Serializer(enabled_infos, many=True).data
    def get_matos_infos(self, obj):
        enabled_infos = Vehicles_infos.objects.filter(Vehicle=obj, bEnabled=True, nType=2)
        return cl_Vehicles_infos_Serializer(enabled_infos, many=True).data
    def get_perempt_infos(self, obj):
        enabled_infos = Vehicles_infos.objects.filter(Vehicle=obj, bEnabled=True, nType=3)
        return cl_Vehicles_infos_Serializer(enabled_infos, many=True).data