diff --git a/smash/web/forms/study_subject_forms.py b/smash/web/forms/study_subject_forms.py index a8743d146d029b0a26d25c66c1c472ed30fd6974..8fab21fc5fce5baeaf417452fadd26417b4c016e 100644 --- a/smash/web/forms/study_subject_forms.py +++ b/smash/web/forms/study_subject_forms.py @@ -88,6 +88,18 @@ class StudySubjectDetailForm(ModelForm): model = StudySubject fields = '__all__' + def __init__(self, *args, **kwargs): + super(StudySubjectDetailForm, self).__init__(*args, **kwargs) + instance = getattr(self, 'instance', None) + + prepare_study_subject_fields(fields=self.fields, study=self.study) + +def get_study_from_study_subject_instance(study_subject): + if instance and instance.study_id: + self.study = Study.objects.filter(id=instance.study_id)[0] + else: + self.study = Study(columns=StudyColumns()) + class StudySubjectEditForm(ModelForm): datetime_contact_reminder = forms.DateTimeField(label="Contact on",