relationships count

api/nodes/serializers/node.py

@@ -10,7 +10,7 @@ class WriteProposedRelationshipSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = ProposedRelationship
-        fields = ('from_node', 'to_node', 'created_by')
+        fields = ('id', 'from_node', 'to_node', 'created_by')
 
     def validate(self, data):
         from_node = data['from_node']
@@ -31,7 +31,7 @@ class ReadProposedRelationshipSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = ProposedRelationship
-        fields = ('from_node', 'to_node', 'created_by')
+        fields = ('id', 'from_node', 'to_node', 'created_by')
 
 
 class DescribeNodeSerializer(serializers.ModelSerializer):
@@ -43,7 +43,7 @@ class DescribeNodeSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Node
-        fields = ('created_by', 'content', 'private', 'type', 'feed', 'parents', 'children')
+        fields = ('id', 'created_by', 'content', 'private', 'type', 'feed', 'parents', 'children')
 
     def get_parents(self, obj):
         return obj.to_node.all()
@@ -54,9 +54,18 @@ def get_children(self, obj):
 
 class ListNodeSerializer(serializers.ModelSerializer):
 
+    parents_count = serializers.SerializerMethodField()
+    children_count = serializers.SerializerMethodField()
+
     class Meta:
         model = Node
-        fields = ('created_by', 'content', 'private', 'type', 'feed')
+        fields = ('id', 'created_by', 'content', 'private', 'type', 'feed', 'parents_count', 'children_count')
+
+    def get_parents_count(self, obj):
+        return obj.to_node.count()
+
+    def get_children_count(self, obj):
+        return obj.from_node.count()
 
 
 class DescribeProposedRelationshipSerializer(serializers.ModelSerializer):
@@ -67,7 +76,7 @@ class DescribeProposedRelationshipSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = ProposedRelationship
-        fields = ('from_node', 'to_node', 'created_by')
+        fields = ('id', 'from_node', 'to_node', 'created_by')
 
 
 class WriteNodeSerializer(serializers.ModelSerializer):
@@ -78,7 +87,7 @@ class WriteNodeSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Node
-        fields = ('created_by', 'content', 'private', 'type', 'feed', 'parents', 'children')
+        fields = ('id', 'created_by', 'content', 'private', 'type', 'feed', 'parents', 'children')
         read_only_fields = ('created_by',)
         extra_kwargs = {
             'created_by': {'write_only': True},