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))
