save owner

tam315 · Feb 11, 2019 · c437c2d · c437c2d
1 parent 27ba16d
commit c437c2d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/tutorial/snippets/models.py b/tutorial/snippets/models.py
@@ -20,6 +20,9 @@ class Snippet(models.Model):
     style = models.CharField(choices=STYLE_CHOICES,
                              default='friendly',
                              max_length=100)
+    owner = models.ForeignKey('auth.User',
+                              related_name='snippets',
+                              on_delete=models.CASCADE)
     highlighted = models.TextField()
 
     class Meta:

diff --git a/tutorial/snippets/serializers.py b/tutorial/snippets/serializers.py
@@ -3,6 +3,9 @@
 
 
 class SnippetSerializer(serializers.ModelSerializer):
+    owner = serializers.CharField(read_only=True)
+
     class Meta:
         model = Snippet
-        fields = ('id', 'title', 'code', 'linenos', 'language', 'style')
+        fields = ('id', 'title', 'code', 'linenos',
+                  'language', 'style', 'owner')
diff --git a/tutorial/snippets/views.py b/tutorial/snippets/views.py
@@ -11,6 +11,9 @@ class SnippetList(generics.ListCreateAPIView):
     queryset = Snippet.objects.all()
     serializer_class = SnippetSerializer
 
+    # 新規作成（POST）時にはユーザ情報を一緒に保存する
+    def perform_create(self, serializer):
+        serializer.save(owner=self.request.user)
 
 class SnippetDetail(generics.RetrieveUpdateDestroyAPIView):
     """