def __init__(self, prefix='', project_id=None):
super(NIFTIImporterForm, self).__init__(prefix, project_id)
self.data_file = TraitUploadField(NIFTIImporterModel.data_file, ('.nii', '.gz', '.zip'), self, name='data_file')
self.apply_corrections = BoolField(NIFTIImporterModel.apply_corrections, self, name='apply_corrections')
self.mappings_file = TraitUploadField(NIFTIImporterModel.mappings_file, '.txt', self, name='mappings_file')
self.connectivity = TraitDataTypeSelectField(NIFTIImporterModel.connectivity, self, name='connectivity')
def __init__(self):
super(GIFTISurfaceImporterForm, self).__init__()
self.file_type = SelectField(GIFTISurfaceImporterModel.file_type, name='file_type')
self.data_file = TraitUploadField(GIFTISurfaceImporterModel.data_file, '.gii', 'data_file')
self.data_file_part2 = TraitUploadField(GIFTISurfaceImporterModel.data_file_part2, '.gii', 'data_file_part2')
self.should_center = BoolField(GIFTISurfaceImporterModel.should_center, name='should_center')
def __init__(self):
super(CSVConnectivityImporterForm, self).__init__()
self.weights = TraitUploadField(CSVConnectivityImporterModel.weights, '.csv', 'weights')
self.weights_delimiter = SelectField(CSVConnectivityImporterModel.weights_delimiter, name='weights_delimiter',
choices=DELIMITER_OPTIONS)
self.tracts = TraitUploadField(CSVConnectivityImporterModel.tracts, ['.csv'], 'tracts')
self.tracts_delimiter = SelectField(CSVConnectivityImporterModel.tracts_delimiter, name='tracts_delimiter',
choices=DELIMITER_OPTIONS)
self.input_data = TraitDataTypeSelectField(CSVConnectivityImporterModel.input_data, 'input_data')
def __init__(self, project_id=None):
super(GIFTISurfaceImporterForm, self).__init__(project_id)
self.file_type = SelectField(GIFTISurfaceImporterModel.file_type, self.project_id, name='file_type',
choices=GIFTISurfaceImporterModel.surface_types)
self.data_file = TraitUploadField(GIFTISurfaceImporterModel.data_file, '.gii', self.project_id,
'data_file', self.temporary_files)
self.data_file_part2 = TraitUploadField(GIFTISurfaceImporterModel.data_file_part2, '.gii', self.project_id,
'data_file_part2', self.temporary_files)
self.should_center = BoolField(GIFTISurfaceImporterModel.should_center, self.project_id, name='should_center')
def test_upload_field(self):
connectivity_file = os.path.join(os.path.dirname(tvb_data.__file__), 'connectivity', 'connectivity_96.zip')
data_file = Str('Test Upload Field')
required_type = '.zip'
upload_field = TraitUploadField(data_file, required_type, self.name)
post_data = {'Data_Subject': 'John Doe', self.name: connectivity_file, 'normalization': 'explicit-None-value'}
upload_field.fill_from_post(post_data)
assert post_data[self.name] == upload_field.data, "Path was not set correctly on TraitUploadField!"
def __init__(self):
super(ZIPSurfaceImporterForm, self).__init__()
self.uploaded = TraitUploadField(ZIPSurfaceImporterModel.uploaded, '.zip', 'uploaded')
self.surface_type = SelectField(ZIPSurfaceImporterModel.surface_type, 'surface_type',
choices=ALL_SURFACES_SELECTION)
self.zero_based_triangles = BoolField(ZIPSurfaceImporterModel.zero_based_triangles, name='zero_based_triangles')
self.should_center = BoolField(ZIPSurfaceImporterModel.should_center, name='should_center')
def __init__(self):
super(BRCOImporterForm, self).__init__()
self.data_file = TraitUploadField(BRCOImporterModel.data_file, '.xml',
'data_file')
self.connectivity = TraitDataTypeSelectField(
BRCOImporterModel.connectivity, 'connectivity')
def __init__(self, prefix='', project_id=None):
super(ZIPConnectivityImporterForm, self).__init__(prefix, project_id)
self.uploaded = TraitUploadField(ZIPConnectivityImporterModel.uploaded, '.zip', self,
name='uploaded')
self.normalization = SelectField(ZIPConnectivityImporterModel.normalization, self, name='normalization',
choices=NORMALIZATION_OPTIONS)
def __init__(self):
super(SensorsImporterForm, self).__init__()
self.sensors_file = TraitUploadField(SensorsImporterModel.sensors_file,
('.txt', '.bz2'), 'sensors_file')
self.sensors_type = SelectField(SensorsImporterModel.sensors_type,
name='sensors_type')
def __init__(self, prefix='', project_id=None):
super(ConnectivityMeasureImporterForm, self).__init__(prefix, project_id)
self.data_file = TraitUploadField(ConnectivityMeasureImporterModel.data_file, '.mat', self, name='data_file')
self.dataset_name = StrField(ConnectivityMeasureImporterModel.dataset_name, self, name='dataset_name')
self.connectivity = TraitDataTypeSelectField(ConnectivityMeasureImporterModel.connectivity, self,
name='connectivity')
def __init__(self, prefix='', project_id=None):
super(TVBImporterForm, self).__init__(prefix, project_id)
self.data_file = TraitUploadField(TVBImporterModel.data_file,
('.zip', '.h5'),
self,
name='data_file')
def __init__(self, project_id=None):
super(RegionMatTimeSeriesImporterForm, self).__init__(project_id)
self.data_file = TraitUploadField(
RegionMatTimeSeriesImporterModel.data_file, '.mat',
self.project_id, 'data_file', self.temporary_files)
self.dataset_name = StrField(
RegionMatTimeSeriesImporterModel.dataset_name,
self.project_id,
name='dataset_name')
self.structure_path = StrField(
RegionMatTimeSeriesImporterModel.structure_path,
self.project_id,
name='structure_path')
self.transpose = BoolField(RegionMatTimeSeriesImporterModel.transpose,
self.project_id,
name='transpose')
self.slice = StrField(RegionMatTimeSeriesImporterModel.slice,
self.project_id,
name='slice')
self.sampling_rate = IntField(
RegionMatTimeSeriesImporterModel.sampling_rate,
self.project_id,
name='sampling_rate')
self.start_time = IntField(RegionMatTimeSeriesImporterModel.start_time,
self.project_id,
name='start_time')
self.datatype = TraitDataTypeSelectField(
RegionMatTimeSeriesImporterModel.datatype,
self.project_id,
name='tstype_parameters')
def __init__(self):
super(TrackImporterForm, self).__init__()
self.data_file = TraitUploadField(TrackImporterModel.data_file, '.trk',
'data_file')
self.region_volume = TraitDataTypeSelectField(
TrackImporterModel.region_volume, name='region_volume')
def __init__(self):
super(ZIPConnectivityImporterForm, self).__init__()
self.uploaded = TraitUploadField(ZIPConnectivityImporterModel.uploaded,
'.zip', 'uploaded')
self.normalization = SelectField(
ZIPConnectivityImporterModel.normalization, name='normalization')
def __init__(self, prefix='', project_id=None):
super(ObjSurfaceImporterForm, self).__init__(prefix, project_id)
self.surface_type = SelectField(ObjSurfaceImporterModel.surface_type, self, name='surface_type',
choices=ALL_SURFACES_SELECTION)
self.data_file = TraitUploadField(ObjSurfaceImporterModel.data_file, '.obj', self, name='data_file')
self.should_center = BoolField(ObjSurfaceImporterModel.should_center, self, name='should_center')
def __init__(self, prefix='', project_id=None):
super(SensorsImporterForm, self).__init__(prefix, project_id)
self.sensors_file = TraitUploadField(SensorsImporterModel.sensors_file, ('.txt', '.bz2'), self,
name='sensors_file')
self.sensors_type = SelectField(SensorsImporterModel.sensors_type, self, name='sensors_type',
choices=SensorsImporterModel.OPTIONS)
def __init__(self):
super(ConnectivityMeasureImporterForm, self).__init__()
self.data_file = TraitUploadField(
ConnectivityMeasureImporterModel.data_file, '.mat', 'data_file')
self.dataset_name = StrField(
ConnectivityMeasureImporterModel.dataset_name, 'dataset_name')
self.connectivity = TraitDataTypeSelectField(
ConnectivityMeasureImporterModel.connectivity, name='connectivity')
def __init__(self, prefix='', project_id=None):
super(TrackImporterForm, self).__init__(prefix, project_id)
self.data_file = TraitUploadField(TrackImporterModel.data_file,
'.trk',
self,
name='data_file')
self.region_volume = TraitDataTypeSelectField(
TrackImporterModel.region_volume, self, name='region_volume')
def __init__(self, prefix='', project_id=None):
super(BRCOImporterForm, self).__init__(prefix, project_id)
self.data_file = TraitUploadField(BRCOImporterModel.data_file,
'.xml',
self,
name='data_file')
self.connectivity = TraitDataTypeSelectField(
BRCOImporterModel.connectivity, self, name='connectivity')
def __init__(self):
super(RegionMappingImporterForm, self).__init__()
self.mapping_file = TraitUploadField(RegionMappingImporterModel.mapping_file, ('.txt', '.zip', '.bz2'),
'mapping_file')
surface_conditions = FilterChain(fields=[FilterChain.datatype + '.surface_type'], operations=['=='],
values=[CORTICAL])
self.surface = TraitDataTypeSelectField(RegionMappingImporterModel.surface, name='surface',
conditions=surface_conditions)
self.connectivity = TraitDataTypeSelectField(RegionMappingImporterModel.connectivity, name='connectivity')
def __init__(self, prefix='', project_id=None):
super(NetworkxConnectivityImporterForm, self).__init__(prefix, project_id)
self.data_file = TraitUploadField(NetworkxImporterModel.data_file, '.gpickle', self, name='data_file')
self.key_edge_weight = StrField(NetworkxImporterModel.key_edge_weight, self, name='key_edge_weight')
self.key_edge_tract = StrField(NetworkxImporterModel.key_edge_tract, self, name='key_edge_tract')
self.key_node_coordinates = StrField(NetworkxImporterModel.key_node_coordinates, self,
name='key_node_coordinates')
self.key_node_label = StrField(NetworkxImporterModel.key_node_label, self, name='key_node_label')
self.key_node_region = StrField(NetworkxImporterModel.key_node_region, self, name='key_node_region')
self.key_node_hemisphere = StrField(NetworkxImporterModel.key_node_hemisphere, self, name='key_node_hemisphere')
def __init__(self, prefix='', project_id=None):
super(ProjectionMatrixImporterForm, self).__init__(prefix, project_id)
self.projection_file = TraitUploadField(ProjectionMatrixImporterModel.projection_file, ('.mat', '.npy'), self,
name='projection_file')
self.dataset_name = StrField(ProjectionMatrixImporterModel.dataset_name, self, name='dataset_name')
surface_conditions = FilterChain(fields=[FilterChain.datatype + '.surface_type'], operations=['=='],
values=['Cortical Surface'])
self.surface = TraitDataTypeSelectField(ProjectionMatrixImporterModel.surface, self, name='surface',
conditions=surface_conditions)
self.sensors = TraitDataTypeSelectField(ProjectionMatrixImporterModel.sensors, self, name='sensors')
def __init__(self):
super(ObjSurfaceImporterForm, self).__init__()
self.surface_type = SelectField(ObjSurfaceImporterModel.surface_type,
name='surface_type')
self.data_file = TraitUploadField(ObjSurfaceImporterModel.data_file,
'.obj', 'data_file')
self.should_center = BoolField(ObjSurfaceImporterModel.should_center,
name='should_center')
del self.surface_type.choices[-1]
def __init__(self):
super(ABCUploaderForm, self).__init__()
self.subject_field = StrField(UploaderViewModel.data_subject,
name='Data_Subject')
# Show Encryption field only when the current TVB installation is capable of decryption
supports_encrypted_files = (
TvbProfile.current.UPLOAD_KEY_PATH is not None
and os.path.exists(TvbProfile.current.UPLOAD_KEY_PATH))
if supports_encrypted_files:
self.encrypted_aes_key = TraitUploadField(
UploaderViewModel.encrypted_aes_key, '.pem',
'encrypted_aes_key')
def __init__(self, project_id=None):
super(CSVConnectivityImporterForm, self).__init__(project_id)
self.weights = TraitUploadField(CSVConnectivityImporterModel.weights,
'.csv', self.project_id, 'weights',
self.temporary_files)
self.weights_delimiter = SelectField(
CSVConnectivityImporterModel.weights_delimiter,
self.project_id,
name='weights_delimiter',
choices=DELIMITER_OPTIONS)
self.tracts = TraitUploadField(CSVConnectivityImporterModel.tracts,
['.csv'], self.project_id, 'tracts',
self.temporary_files)
self.tracts_delimiter = SelectField(
CSVConnectivityImporterModel.tracts_delimiter,
self.project_id,
name='tracts_delimiter',
choices=DELIMITER_OPTIONS)
self.input_data = TraitDataTypeSelectField(
CSVConnectivityImporterModel.input_data, self.project_id,
'input_data')
def __init__(self):
super(GIFTITimeSeriesImporterForm, self).__init__()
self.data_file = TraitUploadField(
GIFTITimeSeriesImporterModel.data_file, '.gii', 'data_file')
surface_conditions = FilterChain(
fields=[FilterChain.datatype + '.surface_type'],
operations=["=="],
values=['Cortical Surface'])
self.surface = TraitDataTypeSelectField(
GIFTITimeSeriesImporterModel.surface,
name='surface',
conditions=surface_conditions)
def __init__(self, project_id=None):
super(ZIPSurfaceImporterForm, self).__init__(project_id)
self.uploaded = TraitUploadField(ZIPSurfaceImporterModel.uploaded,
'.zip', self.project_id, 'uploaded',
self.temporary_files)
self.surface_type = SelectField(ZIPSurfaceImporterModel.surface_type,
self.project_id,
'surface_type',
choices=ALL_SURFACES_SELECTION)
self.zero_based_triangles = BoolField(
ZIPSurfaceImporterModel.zero_based_triangles,
self.project_id,
name='zero_based_triangles')
self.should_center = BoolField(ZIPSurfaceImporterModel.should_center,
self.project_id,
name='should_center')
def __init__(self):
super(TVBImporterForm, self).__init__()
self.data_file = TraitUploadField(TVBImporterModel.data_file,
('.zip', '.h5'), 'data_file')
def __init__(self, project_id=None):
super(FooDataImporterForm, self).__init__(project_id)
self.array_data = TraitUploadField(FooDataImporterModel.array_data,
'.npy', self.project_id,
'array_data', self.temporary_files)
def __init__(self, project_id=None):
super(BRCOImporterForm, self).__init__(project_id)
self.data_file = TraitUploadField(BRCOImporterModel.data_file, '.xml', self.project_id, 'data_file',
self.temporary_files)
self.connectivity = TraitDataTypeSelectField(BRCOImporterModel.connectivity, self.project_id, 'connectivity')