from django import forms
from django.contrib.auth.models import User
from .models import Dataset, Subject
from django.utils.translation import ugettext_lazy as _


class DatasetForm(forms.ModelForm):
	class Meta:
		model = Dataset
		fields = ['dataset_id', 'collection_site']
		labels={
			'dataset': _('Dataset'),
		'collection_site':_('Collection Site')
		}
		help_texts={
			'dataset': _('The name of the dataset this subject is a part of.'),
			'collection_site':_('The location the dataset was collected.')
		}

class SubjectForm(forms.ModelForm):
	class Meta:
		model = Subject
		fields = ['dataset', 'sub_id', 'sess_id', 'func_scan', 'struct_scan','an', 'slice_timing', 'dti_file', 'bvals_file', 'bvecs_file', 'state', 'bucket', 'bidsdir', 'jobdir', 'creds_file', 'datasetname', 'modality']
		labels={
			'dataset': _('Dataset'),
			'sub_id': _('Subject ID'),
			'sess_id': _('Session ID'),
			'func_scan':_('Functional Scan'),
			'struct_scan':_('Structural Scan'),
			'an':_('Structural scan type'),
			'slice_timing':_('Slice Timing Method'),
			'dti_file':_('DTI Image'),
			'bvals_file':_('DTI b-values File'),
			'bvecs_file':_('DTI b-vectors File'),
			'state':_('Analysis Level'),
			'bucket':_('S3 Bucket Name'),
			'bidsdir':_('S3 Bucket BIDS Directory'),
			'jobdir':_('Job Directory'),
			'creds_file':_('AWS Credentials File'),
			'datasetname':_('Dataset Name'),
			'modality':_('Modality')
		}
		help_texts={
			'dataset': _('The name of the dataset this subject is a part of.'),
			'sub_id': _('A unique identifier for the subject.'),
			'sess_id': _('A session name for the subject.'),
			'func_scan':_('the fMRI of this subject.'),
			'struct_scan':_('the structural MRI of this subject.'),
			'an':_('The Type of structural scan.'),
			'slice_timing':_('The method in which slices were acquired.'),
			'dti_file':_('DTI image file for NDMG pipeline'),
			'bvals_file':_('DTI b-values file for NDMG pipeline'),
			'bvecs_file':_('DRI b-vectors file for NDMG pipeline'),
			'state':_('Level of analysis to perform'),
			'bucket':_('Name of S3 bucket where data lives'),
			'bidsdir':_('Path on S3 bucket where data lives'),
			'jobdir':_('Directory of jobs to kill or check status'),
			'creds_file':_('File containing user credentials for AWS services'),
			'datasetname':_('Dataset name'),
			'modality':_('Modality of data')
		}

from django import forms
from django.contrib.auth.models import User
from .models import Submission
from django.utils.translation import ugettext_lazy as _


class SubmissionForm(forms.ModelForm):
	class Meta:
		model = Submission
		fields = ['state', 'bucket', 'bidsdir', 'jobdir', 'creds_file', 'datasetname', 'modality', 'slice_timing']
		labels={
			'state':_('Analysis Level'),
			'bucket':_('S3 Bucket Name'),
			'bidsdir':_('S3 Bucket BIDS Directory'),
			'jobdir':_('Job Directory'),
			'creds_file':_('AWS Credentials File'),
			'datasetname':_('Dataset Name'),
			'modality':_('Modality'),
			'slice_timing':_('Slice Timing Method')
		}
		help_texts={
			'state':_('Level of analysis to perform'),
			'bucket':_('Name of S3 bucket where data lives'),
			'bidsdir':_('Path on S3 bucket where data lives'),
			'jobdir':_('Directory of jobs to kill or check status'),
			'creds_file':_('File containing user credentials for AWS services'),
			'datasetname':_('Dataset name'),
			'modality':_('Modality of data'),
			'slice_timing':_('The method in which slices were acquired.')
		}

Update forms.py