Aggiungere un campo count a un serializzatore django rest framework

Question

Sto serializzando il modello del gruppo django incorporato e vorrei aggiungere un campo al serializzatore che conta il numero di utenti nel gruppo. Attualmente sto usando il seguente serializzatore:

class GroupSerializer(serializers.ModelSerializer): 
    class Meta: 
     model = Group 
     fields = ('id', 'name', 'user_set') 

Questo restituisce l'ID del gruppo e il nome e una serie di utenti (user ID) nel gruppo:

{ 
    "id": 3, 
    "name": "Test1", 
    "user_set": [ 
     9 
    ] 
} 

Quello che vorrei, invece, come l'uscita è qualcosa come:

{ 
    "id": 3, 
    "name": "Test1", 
    "user_count": 1 
} 

Qualsiasi aiuto sarebbe apprezzato. Grazie.

Answer 1

Questo dovrebbe funzionare

class GroupSerializer(serializers.ModelSerializer): 

    user_count = serializers.SerializerMethodField() 

    class Meta: 
     model = Group 
     fields = ('id', 'name','user_count') 

    def get_user_count(self, obj): 
     return obj.user_set.count() 

Questo aggiunge un campo user_count al serializzatore il cui valore viene fissato dal get_user_count, che restituirà la lunghezza del user_set.

Potete trovare ulteriori informazioni su SerializerMethodField qui: http://www.django-rest-framework.org/api-guide/fields/#serializermethodfield

Answer 2

un po 'tardi, ma risposta breve. Prova questo

user_count = serializers.IntegerField(
    source='user_set.count', 
    read_only=True 
)