def build(cls,
manager,
extension,
default_parser=None,
form_code=None,
is_datasender=False):
channels = dict({
".xls": Channel.XLS,
".xlsx": Channel.XLSX,
".csv": Channel.CSV
})
try:
channel = channels[extension]
except KeyError:
raise InvalidFileFormatException()
if default_parser is not None:
parser = default_parser()
elif extension == '.csv':
parser = CsvParser()
elif extension == '.xls':
parser = XlsParser()
elif extension == '.xlsx':
parser = XlsxParser()
else:
raise InvalidFileFormatException()
location_bridge = LocationBridge(
get_location_tree(), get_loc_hierarchy=get_location_hierarchy)
player = FilePlayer(manager,
parser,
channel,
location_tree=location_bridge,
is_datasender=is_datasender)
player.form_code = form_code
return player
def setUp(self):
loc_tree = Mock()
loc_tree.get_hierarchy_path.return_value = None
self.dbm = Mock(spec=DatabaseManager)
self.submission_handler_mock = Mock(spec=SubmissionHandler)
self.parser = XlsParser()
self.xls_data = """
FORM_CODE,ID,BEDS,DIRECTOR,MEDS
CLF1,CL001,10,Dr. A,201
CLF1,CL002,11,Dr. B,202
CLF2,CL003,12,Dr. C,203
CLF1,CL004,13,Dr. D,204
CLF1,CL005,14,Dr. E,205
"""
self.file_name = 'test.xls'
wb = xlwt.Workbook()
ws = wb.add_sheet('test')
for row_number, row in enumerate(self.xls_data.split('\n')):
for col_number, val in enumerate(row.split(',')):
ws.write(row_number, col_number, val)
wb.save(self.file_name)
self.player = XlsPlayer(self.dbm, self.submission_handler_mock,
self.parser, loc_tree)
self.generate_code_patcher = patch(
"mangrove.transport.player.player.Player._handle_registration_form"
)
self.generate_code_patcher.start()
def _init_xls_data(self):
self.parser = XlsParser()
self.file_name = 'test.xls'
wb = xlwt.Workbook()
ws = wb.add_sheet('test')
for row_number, row in enumerate(UPLOAD_DATA.split('\n')):
for col_number, val in enumerate(row.split(',')):
ws.write(row_number, col_number, val)
wb.save(self.file_name)
def setUp(self):
data = """
FORM_CODE,ID,BEDS,DIRECTOR,MEDS
CLF1,CL001,10,Dr. A,201
CLF1,CL002,11,Dr. B,202
CLF2,CL003,12,Dr. C,203
CLF1,CL004,13,Dr. D,204
CLF1,CL005,14,Dr. E,205
"""
self._write_to_xls(data)
self.parser = XlsParser()
def _handle_uploaded_file(file_name, file, manager):
base_name, extension = os.path.splitext(file_name)
if extension == '.csv':
csv_player = CsvPlayer(dbm=manager,
submission_handler=SubmissionHandler(manager),
parser=CsvParser())
response = csv_player.accept(file)
elif extension == '.xls':
xls_player = XlsPlayer(dbm=manager,
submission_handler=SubmissionHandler(manager),
parser=XlsParser())
response = xls_player.accept(file)
else:
raise InvalidFileFormatException()
return response
