From 714ec7ffc63dc1b930d1ae80cabd964b8e48f55d Mon Sep 17 00:00:00 2001
From: Johannes Christ <jc@jchri.st>
Date: Sun, 18 Nov 2018 22:28:02 +0100
Subject: Add viewsets and serializers.

---
 api/serializers.py | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

(limited to 'api/serializers.py')

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
-- 
cgit v1.2.3