Parcourir la source

Change permission class

Mohidul Islam il y a 4 ans
Parent
commit
bb1f05806a
2 fichiers modifiés avec 6 ajouts et 4 suppressions
  1. 1 1
      dashboard/views.py
  2. 5 3
      facebook_app/views.py

+ 1 - 1
dashboard/views.py

@@ -48,7 +48,7 @@ class ReviewListView(LoginRequiredMixin, View):
 
 
 class UnRepliedReviewList(LoginRequiredMixin, PermissionRequiredMixin, View):
-    permission_required = 'is_staff'
+    permission_required = 'is_admin'
 
     def post(self, request, *args, **kwargs):
         form = ReplyForm(self.request.POST)

+ 5 - 3
facebook_app/views.py

@@ -3,7 +3,7 @@ from django.shortcuts import redirect
 from django.views.generic import View
 from django.contrib import messages
 from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
-from django.contrib.auth.mixins import LoginRequiredMixin
+from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
 from django.utils import timezone
 
 from nlu_job.nlu_utils import model_inference, analyze_model_inference, clean_text
@@ -15,7 +15,8 @@ from .review_utils import reply2facebook_review
 from nlu_job.nlu_utils import filter_with_last_ten_reviews
 
 
-class FacebookReviews(LoginRequiredMixin, View):
+class FacebookReviews(LoginRequiredMixin, PermissionRequiredMixin, View):
+    permission_required = 'is_admin'
 
     def post(self, request, *args, **kwargs):
         form = ReplyForm(self.request.POST)
@@ -67,7 +68,8 @@ class FacebookReviews(LoginRequiredMixin, View):
         return render(request, 'facebook-reviews.html', context=context)
 
 
-class FacebookReviewPrediction(View):
+class FacebookReviewPrediction(LoginRequiredMixin, PermissionRequiredMixin, View):
+    permission_required = 'is_admin'
 
     def get(self, request, id,  *args, **kwargs):
         review = FacebookReview.objects.get(pk=id)