serializer

serializers.py

from rest_framework import serializers

# from dashboard.serializers import HarvestListSerializer
from team.serializers import DeviceTeamListSerializer  # , DeviceSerializer
from team.models import DeviceTeam

from .models import Harvest


class BaseMeta:
    exclude = ("polymorphic_ctype", "rght", 'tree_id', "level")


class BaseSerializer(serializers.ModelSerializer):
    class_name = serializers.SerializerMethodField()

    def get_class_name(self, obj):
        return obj.__class__.__name__


class HarvestSerializer(BaseSerializer):
    devices = serializers.SerializerMethodField()

    def get_devices(self, obj):
        print obj
        device_team = DeviceTeam.objects.filter(form=obj)
        return DeviceTeamListSerializer(device_team).data

    class Meta:
        model = Harvest
        exclude = ('node_ptr',)