index.community/apiv1/serializers.py

from rest_framework import serializers
from collections import OrderedDict
from scraper.models import Instance, PeerRelationship


class InstanceListSerializer(serializers.ModelSerializer):
    class Meta:
        model = Instance
        fields = ('name', 'user_count')

    def to_representation(self, instance):
        """
        Object instance -> Dict of primitive datatypes.
        We use a custom to_representation function to exclude empty fields in the serialized JSON.
        """
        ret = super(InstanceListSerializer, self).to_representation(instance)
        ret = OrderedDict(list(filter(lambda x: x[1], ret.items())))
        return ret


class InstanceDetailSerializer(serializers.ModelSerializer):
    userCount = serializers.SerializerMethodField()
    statusCount = serializers.SerializerMethodField()
    domainCount = serializers.SerializerMethodField()
    lastUpdated = serializers.SerializerMethodField()
    peers = InstanceListSerializer(many=True, read_only=True)

    def get_userCount(self, obj):
        return obj.user_count

    def get_statusCount(self, obj):
        return obj.status_count

    def get_domainCount(self, obj):
        return obj.domain_count

    def get_lastUpdated(self, obj):
        return obj.last_updated

    class Meta:
        model = Instance
        fields = ('name', 'description', 'version', 'userCount', 'statusCount', 'domainCount', 'peers', 'lastUpdated')


class EdgeSerializer(serializers.ModelSerializer):
    id = serializers.SerializerMethodField('get_pk')

    class Meta:
        model = PeerRelationship
        fields = ('source', 'target', 'id')

    def get_pk(self, obj):
        return obj.pk


class NodeSerializer(serializers.ModelSerializer):
    id = serializers.SerializerMethodField('get_name')
    label = serializers.SerializerMethodField('get_name')
    size = serializers.SerializerMethodField()
    x = serializers.SerializerMethodField()
    y = serializers.SerializerMethodField()

    class Meta:
        model = Instance
        fields = ('id', 'label', 'size', 'x', 'y')

    def get_name(self, obj):
        return obj.name

    def get_size(self, obj):
        return obj.user_count or 1

    def get_x(self, obj):
        return obj.x_coord

    def get_y(self, obj):
        return obj.y_coord

    def to_representation(self, instance):
        """
        Object instance -> Dict of primitive datatypes.
        We use a custom to_representation function to exclude empty fields in the serialized JSON.
        """
        ret = super(NodeSerializer, self).to_representation(instance)
        ret = OrderedDict(list(filter(lambda x: x[1], ret.items())))
        return ret