Coverage for api/mixins.py: 68%

1from rest_framework.permissions import AllowAny 

2 

3from api.permissions import ProscaiPermissions 

4from django_audit_log.rest_framework.viewsets import AuditLogViewSet 

5 

6 

7class BaseMixin(AuditLogViewSet): 

8 @property 

9 def model(self): 

10 """ 

11 Retrieve model used in serializer 

12 """ 

13 return self.serializer_class.Meta.model 

14 

15 def get_queryset(self): 

16 """ 

17 Get Only Organization Related objects 

18 """ 

19 if self.queryset: 

20 return self.queryset 

21 return self.model.objects.all() 

22 

23 

24class ProscaiMixin(BaseMixin): 

25 permission_classes = [ProscaiPermissions]