add message

announcement/serializers.py

@@ -1,7 +1,8 @@
+from submission.serializers import SubmissionCodeSerializer
 from utils.api import serializers
 from utils.api._serializers import UsernameSerializer
 
-from .models import Announcement
+from .models import Announcement, Message
 
 
 class CreateAnnouncementSerializer(serializers.Serializer):
@@ -35,3 +36,29 @@ class EditAnnouncementSerializer(serializers.Serializer):
     content = serializers.CharField(max_length=1024 * 1024 * 8)
     visible = serializers.BooleanField()
     top = serializers.BooleanField()
+
+
+class MessageSerializer(serializers.Serializer):
+    sender = UsernameSerializer()
+    recipient = UsernameSerializer()
+    submission = SubmissionCodeSerializer()
+
+    class Meta:
+        model = Message
+        fields = "__all__"
+
+
+class MessageListSerializer(serializers.Serializer):
+    sender = UsernameSerializer()
+    recipient = UsernameSerializer()
+    submission = SubmissionCodeSerializer()
+
+    class Meta:
+        model = Message
+        exclude = ["message"]
+
+
+class CreateMessageSerializer(serializers.Serializer):
+    recipient = serializers.IntegerField()
+    submission = serializers.CharField()
+    message = serializers.CharField()