Add viewsets and serializers.

1 files changed, 18 insertions, 5 deletions

diff --git a/api/serializers.py b/api/serializers.py
index f8d15bbf..e39cd4a3 100644
--- a/api/serializers.py
+++ b/api/serializers.py
@@ -2,14 +2,27 @@ from rest_framework.serializers import ModelSerializer, PrimaryKeyRelatedField
 from rest_framework_bulk import BulkSerializerMixin
 
 from .models import (
-    DocumentationLink,
-    Member, OffTopicChannelName,
-    Role, SnakeFact,
-    SnakeIdiom, SnakeName,
-    SpecialSnake, Tag
+    DeletedMessage, DocumentationLink,
+    Member, MessageDeletionContext,
+    OffTopicChannelName, Role,
+    SnakeFact, SnakeIdiom,
+    SnakeName, SpecialSnake,
+    Tag
 )
 
 
+class MessageDeletionContextSerializer(BulkSerializerMixin, ModelSerializer):
+    deleted_messages = PrimaryKeyRelatedField(
+        many=True,
+        queryset=DeletedMessage.objects.all()
+    )
+
+    class Meta:
+        model = MessageDeletionContext
+        fields = ('actor', 'creation', 'messages')
+        depth = 1
+
+
 class DocumentationLinkSerializer(ModelSerializer):
     class Meta:
         model = DocumentationLink