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