Skip to content

Instantly share code, notes, and snippets.

@toransahu
Created April 13, 2018 12:12
Show Gist options
  • Select an option

  • Save toransahu/e02fc30cd0e55e968971c46e59862acb to your computer and use it in GitHub Desktop.

Select an option

Save toransahu/e02fc30cd0e55e968971c46e59862acb to your computer and use it in GitHub Desktop.
#!/usr/bin/env python3.6.4
# coding="UTF-8"
__author__ = 'Toran Sahu <toran.sahu@yahoo.com>'
__copyright__ = 'Copyright (C) 2018 Ethereal Machines Pvt. Ltd. All rights reserved'
from rest_framework import viewsets, mixins
from .models import Job, JobEnquiry
from .serializers import JobSerializer, JobEnquirySerializer
from backend.permissions import AllowAny, IsSuperUser
# Create your views here.
class JobViewSet(viewsets.ModelViewSet):
queryset = Job.objects.all()
serializer_class = JobSerializer
def get_permissions(self):
if self.action == 'list':
self.permission_classes = [AllowAny, ]
elif self.action == 'retrieve':
self.permission_classes = [AllowAny, ]
elif self.action == 'create':
self.permission_classes = [IsSuperUser, ]
elif self.action == 'update':
self.permission_classes = [IsSuperUser, ]
elif self.action == 'destroy':
self.permission_classes = [IsSuperUser, ]
return super(JobViewSet, self).get_permissions()
class JobEnquiryViewSet(mixins.ListModelMixin, mixins.RetrieveModelMixin, mixins.CreateModelMixin, mixins.DestroyModelMixin, viewsets.GenericViewSet):
queryset = JobEnquiry.objects.all()
serializer_class = JobEnquirySerializer
def get_permissions(self):
if self.action == 'list':
self.permission_classes = [IsSuperUser, ]
elif self.action == 'retrieve':
self.permission_classes = [IsSuperUser, ]
elif self.action == 'create':
self.permission_classes = [AllowAny, ]
elif self.action == 'destroy':
self.permission_classes = [IsSuperUser, ]
return super(JobEnquiryViewSet, self).get_permissions()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment