Coverage for clustersoftware / views.py: 95%

1from django.contrib.auth.mixins import LoginRequiredMixin 

2from django.contrib import messages 

3from django.http import HttpResponseRedirect 

4from sm.views import SafeDeleteMixin 

5 

6from .models import Model 

7from .forms import Form, FormDisabled 

8from . import app_label 

9 

10from django.views.generic import ListView as GenericListView 

11from django.views.generic.edit import UpdateView as GenericUpdateView 

12from django.views.generic.edit import CreateView as GenericCreateView 

13from django.views.generic.edit import DeleteView as GenericDeleteView 

14from django.contrib.messages.views import SuccessMessageMixin 

15 

16from django.utils.translation import gettext as _ 

17 

18try: 

19 from django.urls import reverse_lazy 

20except Exception: # pragma: no cover 

21 from django.urls import reverse_lazy # pragma: no cover 

22 

23 

24class ListView(LoginRequiredMixin, GenericListView): 

25 template_name = "%s/list.html" % app_label 

26 model = Model 

27 paginate_by = 20 

28 paginate_orphans = paginate_by / 4 

29 ordering = "name" 

30 

31 def get_queryset(self): 

32 return Model.objects.all().order_by("name") 

33 

34 

35class DetailView(LoginRequiredMixin, GenericUpdateView): 

36 template_name = "%s/detail.html" % app_label 

37 model = Model 

38 form_class = FormDisabled 

39 

40 

41class UpdateView(SuccessMessageMixin, LoginRequiredMixin, GenericUpdateView): 

42 success_message = "%(name)s " + _("was updated successfully") 

43 

44 template_name = "%s/edit.html" % app_label 

45 model = Model 

46 

47 def form_valid(self, form): 

48 self.object = form.save() 

49 messages.success(self.request, self.success_message % self.object.__dict__) 

50 

51 return HttpResponseRedirect(self.get_success_url()) 

52 

53 form_class = Form 

54 success_url = reverse_lazy("%s:index" % app_label) 

55 

56 

57class CreateView(SuccessMessageMixin, LoginRequiredMixin, GenericCreateView): 

58 success_message = "%(name)s " + _("was created successfully") 

59 

60 template_name = "%s/edit.html" % app_label 

61 form_class = Form 

62 model = Model 

63 success_url = reverse_lazy("%s:index" % app_label) 

64 

65 def get_initial(self): 

66 from vendor.models import Model as VendorModel 

67 

68 initial = super().get_initial() 

69 if "vendor" in self.kwargs: 

70 initial["vendor"] = VendorModel.objects.filter( 

71 pk=self.kwargs["vendor"] 

72 ).first() 

73 return initial 

74 

75 def form_valid(self, form): 

76 self.object = form.save() 

77 messages.success(self.request, self.success_message % self.object.__dict__) 

78 

79 return HttpResponseRedirect(self.get_success_url()) 

80 

81 

82class DeleteView(SafeDeleteMixin, LoginRequiredMixin, GenericDeleteView): 

83 success_message = "%(name)s " + _("was deleted successfully") 

84 template_name = "delete.html" 

85 model = Model 

86 success_url = reverse_lazy("%s:index" % app_label)