Coverage for apps/invoices/views.py: 71%

1from rest_framework.mixins import CreateModelMixin as Create 

2from rest_framework.mixins import DestroyModelMixin as Delete 

3from rest_framework.mixins import ListModelMixin as List 

4from rest_framework.mixins import RetrieveModelMixin as Detail 

5from rest_framework.mixins import UpdateModelMixin as Update 

6from rest_framework.viewsets import GenericViewSet 

7 

8from apps.users.mixins import PublicMixin, UserMixin 

9 

10from .serializers import ( 

11 InvoiceUseSerializer, 

12 PractitionerInvoiceConfigurationSerializer, 

13 ProductInvoiceConfigurationSerializer, 

14) 

15 

16 

17class PublicInvoiceUseViewSet(PublicMixin, GenericViewSet, List): 

18 """ 

19 Public ViewSet for Invoice Use 

20 """ 

21 

22 serializer_class = InvoiceUseSerializer 

23 

24 

25class UserPractitionerInvoiceConfigurationViewSet(UserMixin, GenericViewSet, Create, List, Detail, Update): 

26 """ 

27 User ViewSet for Practitioner Invoice Configuration 

28 """ 

29 

30 serializer_class = PractitionerInvoiceConfigurationSerializer 

31 

32 def get_object(self): 

33 practitioner = getattr(self.request.user, 'practitioner', None) 

34 if practitioner: 

35 return getattr(practitioner, 'invoice_configuration', None) 

36 return None 

37 

38 def perform_create(self, serializer): 

39 serializer.save(practitioner=self.user.practitioner) 

40 

41 def get_queryset(self): 

42 queryset = super().get_queryset() 

43 return queryset.filter(practitioner=self.request.user.practitioner) 

44 

45 def list(self, request, *args, **kwargs): 

46 return self.retrieve(request, *args, **kwargs) 

47 

48 

49class UserProductInvoiceConfigurationViewSet(UserMixin, GenericViewSet, Create, Detail, Update, Delete): 

50 """ 

51 User ViewSet for Product Invoice Configuration 

52 """ 

53 

54 serializer_class = ProductInvoiceConfigurationSerializer