def test_get_connection(self, mocked_get_connection):
"""
Test and assert that objecstore.get_connection is called with the given
config, and that the connection is returned.
"""
mocked_get_connection.return_value = 'this is the connection'
objstore = ObjectStore(config='this is the config')
connection = objstore.get_connection()
mocked_get_connection.assert_called_with('this is the config')
self.assertEqual(connection, 'this is the connection')
def get_file(self, request, pk=None):
document_model = self.get_object()
container_path = ObjectStore.get_container_path(document_model.type)
filename = document_model.filename
objstore = ObjectStore(settings.OBJECTSTORE_CONNECTION_CONFIG)
connection = objstore.get_connection()
try:
store_object = objstore.get_document(connection, container_path, filename)
except ClientException as e:
return handle_swift_exception(container_path, filename, e)
content_type = store_object[0].get('content-type')
obj_data = store_object[1]
response = HttpResponse(content_type=content_type)
response['Content-Disposition'] = f'attachment; filename="{filename}"'
response.write(obj_data)
return response
def perform_import():
log.info('Clearing models')
clear_models()
objstore = ObjectStore(config=settings.OBJECTSTORE_CONNECTION_CONFIG)
log.info('Opening object store connection')
connection = objstore.get_connection()
log.info('Getting documents list')
document_list = objstore.get_wba_documents_list(connection)
log.info(f'document list size: {len(document_list)}')
log.info('Fetching xls file')
xls_path = objstore.fetch_spots(connection)
log.info('Importing xls file')
process_xls(xls_path, document_list)
log.info(f'Spot count: {Spot.objects.all().count()}')
log.info(f'Document count: {Document.objects.all().count()}')