Преглед на файлове

Change permission class

Mohidul Islam преди 4 години
родител
ревизия
bb1f05806a
променени са 2 файла, в които са добавени 6 реда и са изтрити 4 реда
  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):
 class UnRepliedReviewList(LoginRequiredMixin, PermissionRequiredMixin, View):
-    permission_required = 'is_staff'
+    permission_required = 'is_admin'
 
 
     def post(self, request, *args, **kwargs):
     def post(self, request, *args, **kwargs):
         form = ReplyForm(self.request.POST)
         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.views.generic import View
 from django.contrib import messages
 from django.contrib import messages
 from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
 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 django.utils import timezone
 
 
 from nlu_job.nlu_utils import model_inference, analyze_model_inference, clean_text
 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
 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):
     def post(self, request, *args, **kwargs):
         form = ReplyForm(self.request.POST)
         form = ReplyForm(self.request.POST)
@@ -67,7 +68,8 @@ class FacebookReviews(LoginRequiredMixin, View):
         return render(request, 'facebook-reviews.html', context=context)
         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):
     def get(self, request, id,  *args, **kwargs):
         review = FacebookReview.objects.get(pk=id)
         review = FacebookReview.objects.get(pk=id)