From c93ea945b94a7fa4ce6dbc465d9cda6fbcd23bb8 Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@uni.lu>
Date: Wed, 22 Apr 2020 19:18:20 +0200
Subject: [PATCH] if SHOW_VISITS_FROM_ZERO, then change virus visit labels to
 start from 0

---
 smash/web/forms/study_subject_forms.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/smash/web/forms/study_subject_forms.py b/smash/web/forms/study_subject_forms.py
index 1f75992f..d2d8bb50 100644
--- a/smash/web/forms/study_subject_forms.py
+++ b/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):
-- 
GitLab