Coverage for server / forms.py: 100%

1from .models import Model as ServerModel 

2from sm.forms import SMForm, SMFormDisabled 

3from status.models import Model as StatusModel 

4 

5from django.forms import DateInput 

6from django import forms 

7 

8 

9class Form(SMForm): 

10 """ 

11 Override default form behaviour; Make sure 'In Use' is the default/initial 

12 value 

13 """ 

14 

15 def __init__(self, *args, **kwargs): 

16 super().__init__(*args, **kwargs) 

17 try: 

18 self.fields["status"].initial = StatusModel.objects.get(name="In use").id 

19 except Exception as e: # noqa # flake8: noqa # NOQA # pragma: no cover 

20 print('No status "In Use" found: %s' % e) 

21 

22 class Meta(SMForm.Meta): 

23 model = ServerModel 

24 widgets = { 

25 "delivery_date": DateInput(attrs={"class": "date-input"}), 

26 "install_date": DateInput(attrs={"class": "date-input"}), 

27 } 

28 

29 

30class FormDisabled(Form, SMFormDisabled): 

31 pass 

32 

33 

34class BulkActionForm(forms.Form): 

35 status = forms.ModelChoiceField( 

36 queryset=StatusModel.objects.all(), 

37 required=False, 

38 label="Change Status to", 

39 empty_label="--- No Change ---", 

40 ) 

41 delete = forms.BooleanField(required=False, label="Delete Selected Servers")