from rest_framework import serializers
from management.models.gruppo_driver import GruppoDriver

class GruppoDriverSerializer(serializers.ModelSerializer):   
    flussi = serializers.SerializerMethodField()

    class Meta:
        model = GruppoDriver
        fields = ['id', 'title', 'flussi', 'plan', 'data_ora_creazione']

    def get_flussi(self, obj):
        from .flusso_serializer import FlussoSerializer
        queryset = obj.flusso_set.all() 
        return FlussoSerializer(queryset, many=True).data