def handle_import_request(): """Import request handler""" dry_run, file_data = parse_import_request() csv_data = fa.get_gdrive_file(file_data) try: converter = Converter(dry_run=dry_run, csv_data=csv_data) converter.import_csv() response_data = converter.get_info() response_json = json.dumps(response_data) headers = [("Content-Type", "application/json")] return current_app.make_response((response_json, 200, headers)) except: # pylint: disable=bare-except logger.exception("Import failed") raise BadRequest("Import failed due to server error.")
def test_getter_csv(self, file_data, disco_mock, auth_mock): """Test getter csv data over GDrive.""" disco_files = disco_mock.build.return_value.files.return_value disco_files.get.return_value.execute.return_value = file_data if "mimeType" in file_data: data_getter = disco_files.get_media.return_value.execute else: data_getter = disco_files.export_media.return_value.execute data_getter.return_value = self.CSV_DATA self.assertEqual([[u'Object_tyle', u'Code*', u'Title*', u'LIST*'], [u'', u'OBJ-1185', u'OBJ_title', u'user1\nuser2']], file_actions.get_gdrive_file(file_data)) auth_mock.assert_called_once_with() disco_mock.build.assert_called_once_with("drive", "v3", http=auth_mock.return_value) disco_files.get.assert_called_once_with(fileId=file_data["id"])
def test_getter_csv(self, file_data, disco_mock, auth_mock): """Test getter csv data over GDrive.""" disco_files = disco_mock.build.return_value.files.return_value disco_files.get.return_value.execute.return_value = file_data if "mimeType" in file_data: data_getter = disco_files.get_media.return_value.execute else: data_getter = disco_files.export_media.return_value.execute data_getter.return_value = self.CSV_DATA self.assertEqual( [[u'Object_tyle', u'Code*', u'Title*', u'LIST*'], [u'', u'OBJ-1185', u'OBJ_title', u'user1\nuser2']], file_actions.get_gdrive_file(file_data)) auth_mock.assert_called_once_with() disco_mock.build.assert_called_once_with("drive", "v3", http=auth_mock.return_value) disco_files.get.assert_called_once_with(fileId=file_data["id"])
def handle_import_request(): """Import request handler""" dry_run, file_data = parse_import_request() csv_data = fa.get_gdrive_file(file_data) return make_response(make_import(csv_data, dry_run))
def handle_import_request(): """Import request handler""" dry_run, file_data = parse_import_request() csv_data = fa.get_gdrive_file(file_data) return make_response(make_import(csv_data, dry_run))