diff --git a/smash/web/forms/subject_import_data_form.py b/smash/web/forms/subject_import_data_form.py index 7c49b51283a430ff69d69beb904cbd609d286910..94f07fabcd7fc398066ada3f08c22271646de301 100644 --- a/smash/web/forms/subject_import_data_form.py +++ b/smash/web/forms/subject_import_data_form.py @@ -1,7 +1,7 @@ from django import forms from django.forms import ModelForm -from web.models import SubjectImportData, Subject, StudySubject +from web.models import SubjectImportData, Subject, StudySubject, Language class SubjectImportDataEditForm(ModelForm): @@ -16,7 +16,8 @@ class SubjectImportDataEditForm(ModelForm): for field in Subject._meta.get_fields(): if field.get_internal_type() == "CharField" or \ field.get_internal_type() == "DateField" or \ - field.get_internal_type() == "TextField": + field.get_internal_type() == "TextField" or \ + (field.get_internal_type() == "ForeignKey" and field.related_model in (Language,)): field_id = Subject._meta.db_table + " - " + field.name value = field.name for mapping in instance.column_mappings.all(): @@ -27,7 +28,8 @@ class SubjectImportDataEditForm(ModelForm): for field in StudySubject._meta.get_fields(): if field.get_internal_type() == "CharField" or \ field.get_internal_type() == "DateField" or \ - field.get_internal_type() == "TextField": + field.get_internal_type() == "TextField" or \ + (field.get_internal_type() == "ForeignKey" and field.related_model in (Language,)): field_id = StudySubject._meta.db_table + " - " + field.name value = field.name for mapping in instance.column_mappings.all(): @@ -42,7 +44,8 @@ class SubjectImportDataEditForm(ModelForm): for field in Subject._meta.get_fields(): if field.get_internal_type() == "CharField" or \ field.get_internal_type() == "DateField" or \ - field.get_internal_type() == "TextField": + field.get_internal_type() == "TextField" or \ + (field.get_internal_type() == "ForeignKey" and field.related_model in (Language,)): field_id = Subject._meta.db_table + " - " + field.name value = self[field_id].value() instance.set_column_mapping(Subject, field.name, value) @@ -50,7 +53,8 @@ class SubjectImportDataEditForm(ModelForm): for field in StudySubject._meta.get_fields(): if field.get_internal_type() == "CharField" or \ field.get_internal_type() == "DateField" or \ - field.get_internal_type() == "TextField": + field.get_internal_type() == "TextField" or \ + (field.get_internal_type() == "ForeignKey" and field.related_model in (Language,)): field_id = StudySubject._meta.db_table + " - " + field.name value = self[field_id].value() instance.set_column_mapping(StudySubject, field.name, value)