feat(nodes): added extra fields to node serializers

unknowncoder05 · Jul 28, 2022 · e64998f · e64998f
1 parent 35174fb
commit e64998f
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 6 deletions.
diff --git a/api/feed/serializers/comment.py b/api/feed/serializers/comment.py
@@ -26,8 +26,15 @@ class WriteCommentSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Comment
-        fields = ('created_by', 'content', 'parent')
+        fields = ('created_by', 'feed', 'content', 'parent')
         read_only_fields = ('created_by',)
         extra_kwargs = {
-            'created_by': {'write_only': True}
+            'created_by': {'write_only': True},
+            'feed': {'write_only': True},
         }
+
+    def validate(self, data):
+        if data['parent']:
+            if data['parent'].feed != data['feed']:
+                raise serializers.ValidationError('Parent comment must belong to the same feed')
+        return data
diff --git a/api/feed/urls.py b/api/feed/urls.py
@@ -4,7 +4,7 @@
 
 router = DefaultRouter()
 
-router.register(r'comment', views.CommentViewSet, basename='comment')
+router.register(r'comments', views.CommentViewSet, basename='comment')
 
 
 urlpatterns = [

diff --git a/api/nodes/serializers/node.py b/api/nodes/serializers/node.py
@@ -9,14 +9,14 @@ class NodeSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Node
-        fields = ('created_by', 'name', 'private')
+        fields = ('created_by', 'name', 'private', 'type', 'feed', 'parents', 'nodes')
 
 
 class ListNodeSerializer(serializers.ModelSerializer):
 
         class Meta:
             model = Node
-            fields = ('created_by', 'name', 'private')
+            fields = ('created_by', 'name', 'private', 'type', 'feed', 'parents', 'nodes')
 
 
 class WriteNodeSerializer(serializers.ModelSerializer):
@@ -25,7 +25,7 @@ class WriteNodeSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Node
-        fields = ('created_by', 'name', 'private')
+        fields = ('created_by', 'name', 'private', 'type', 'feed', 'parents')
         read_only_fields = ('created_by',)
         extra_kwargs = {
             'created_by': {'write_only': True}