if SHOW_VISITS_FROM_ZERO, then change virus visit labels to start from 0

smash/web/forms/study_subject_forms.py

@@ -3,6 +3,7 @@ import re
 
 from django import forms
 from django.forms import ModelForm
+from django.conf import settings
 
 from web.forms.forms import DATETIMEPICKER_DATE_ATTRS, get_worker_from_args
 from web.models import StudySubject, Study, StudyColumns, VoucherType, Worker
@@ -22,6 +23,15 @@ class StudySubjectForm(ModelForm):
         super(StudySubjectForm, self).__init__(*args, **kwargs)
         self.fields['health_partner'].queryset = Worker.get_workers_by_worker_type(
             WORKER_HEALTH_PARTNER)
+        
+        if getattr(settings, "SHOW_VISITS_FROM_ZERO", False):
+            virus_visit_numbers = range(0, 5)
+            for one_based_idx, virus_visit_number in enumerate(virus_visit_numbers, 1):
+                field = 'virus_test_{}'.format(one_based_idx)
+                self.fields[field].label = 'Visit {} virus result'.format(virus_visit_number)
+                date_field = 'virus_test_{}_updated'.format(one_based_idx)
+                self.fields[date_field].label = 'Visit {} virus result date'.format(virus_visit_number)
+
 
 
 class StudySubjectAddForm(StudySubjectForm):