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
def setUp(self): self.dbm = Mock(spec=DatabaseManager) loc_tree = Mock() loc_tree.get_hierarchy_path.return_value = None self.submission_handler_mock = Mock(spec=SubmissionHandler) self.parser = CsvParser() self.csv_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._mock_short_code_generator() self.player = CsvPlayer(self.dbm, self.submission_handler_mock, self.parser, loc_tree)