20 lines
526 B
Python
20 lines
526 B
Python
from __future__ import unicode_literals
|
|
|
|
from rest_framework import serializers
|
|
|
|
from user_management.serializers import UserSerializer
|
|
|
|
from .models import Folder
|
|
|
|
|
|
class FolderSerializer(serializers.ModelSerializer):
|
|
documents = serializers.SerializerMethodField('get_documents_count')
|
|
user = UserSerializer(read_only=True)
|
|
|
|
class Meta:
|
|
fields = ('datetime_created', 'documents', 'id', 'label', 'user')
|
|
model = Folder
|
|
|
|
def get_documents_count(self, obj):
|
|
return obj.documents.count()
|