Files
Reskreen/vehicles/serializers.py
2024-07-28 15:49:18 +02:00

31 lines
1.2 KiB
Python

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