24 - Post Owner

Author: codingforentrepreneurs

src/posts/serializers.py

@@ -28,6 +28,7 @@ class PostSerializer(serializers.ModelSerializer):
                             )
     user            = UserPublicSerializer(read_only=True)
     publish         = serializers.DateField(default=timezone.now())
+    owner           = serializers.SerializerMethodField(read_only=True)
 
     class Meta:
         model = Post
@@ -40,5 +41,14 @@ class Meta:
             'draft',
             'publish',
             'updated',
+            'owner',
             'timestamp',
-        ]
+        ]
+    def get_owner(self, obj):
+        request = self.context['request']
+        if request.user.is_authenticated:
+            if obj.user == request.user:
+                return True
+        return False
+
+

src/posts/views.py

@@ -13,10 +13,15 @@ class PostDetailAPIView(generics.RetrieveUpdateDestroyAPIView):
     lookup_field        = 'slug'
     permission_classes  = [IsOwnerOrReadOnly]
 
+    # def get_serializer_context(self):
+    #     context = super().get_serializer_context()
+    #     context['request'] = self.request
+    #     return context
+
 
 class PostListCreateAPIView(generics.ListCreateAPIView):
     queryset            = Post.objects.all()
-    serializer_class    =  PostSerializer
+    serializer_class    = PostSerializer
     permission_classes  = [permissions.IsAuthenticatedOrReadOnly]
 
     def perform_create(self, serializer):

src/reactify-ui/src/posts/PostDetail.js

@@ -1,4 +1,6 @@
 import React, {Component} from 'react'
+import 'whatwg-fetch'
+import cookie from 'react-cookies'
 import { Link } from 'react-router-dom'
 
 class PostDetail extends Component {
@@ -21,6 +23,12 @@ class PostDetail extends Component {
           }
       }
 
+      const csrfToken = cookie.load('csrftoken')
+      if (csrfToken !== undefined) {
+          lookupOptions['credentials'] = 'include'
+          lookupOptions['headers']['X-CSRFToken'] = csrfToken
+       }
+
       fetch(endpoint, lookupOptions)
       .then(function(response){
           if (response.status == 404){
@@ -71,6 +79,8 @@ class PostDetail extends Component {
                     pathname: `/posts`,
                     state: { fromDashboard: false }
                   }}>Posts</Link></p>
+
+                  {post.owner === true ? <div>Update Post</div> : ""}
                 </div>
                }
            </div> : "Loading..."}</p>