def download(self, request, *args, **kwargs):
"""
Download the volume data.
"""
volume = self.get_object()
# Check whether project is corresponding to the user
user = request.user
if not check_member_in_project(volume.project, user):
raise PermissionDenied(
detail="User {} is not in project {}.".format(
user.username, volume.project.name))
logger.info("User {} download the data of volume {}-{}.".format(
request.user.username, volume.project.name, volume.name))
# Copy file to local first
volume_dir = get_volume_direction_on_nfs(volume)
filename = get_upload_volume_filename(volume, request.user)
client = NFSLocalClient()
client.tar_and_copy_to_local(volume_dir, filename)
response = StreamingHttpResponse(big_file_iterator(filename))
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="{}"'.format(
os.path.basename(filename))
return response
class NFSLocalClientTestCase(unittest.TestCase):
def setUp(self):
self.client = NFSLocalClient()
def test_makedir(self):
self.client.makedir('/home/wangtao/testdir')
def test_removedir(self):
self.client.removedir('/home/wangtao/testdir')
def test_copy_file_to_remote(self):
self.client.copy_file_to_remote("/home/wangtao/test/test.tar",
"/hummer/test/test.tar")
def test_copy_file_to_local(self):
self.client.copy_file_to_local("/hummer/test/test.tar",
"/home/wangtao/test.tar")
def test_tar_and_copy_to_local(self):
self.client.tar_and_copy_to_local("/hummer/test",
"/home/wangtao/test.tar")
def test_copy_file_to_remote_and_untar(self):
self.client.copy_file_to_remote_and_untar("/home/wangtao/test/test.tar",
"/hummer/test")
def download(self, request, *args, **kwargs):
"""
Download the volume data.
"""
volume = self.get_object()
# Check whether project is corresponding to the user
user = request.user
if not check_member_in_project(volume.project, user):
raise PermissionDenied(detail="User {} is not in project {}."
.format(user.username, volume.project.name))
logger.info("User {} download the data of volume {}-{}.".format(
request.user.username, volume.project.name, volume.name))
# Copy file to local first
volume_dir = get_volume_direction_on_nfs(volume)
filename = get_upload_volume_filename(volume, request.user)
client = NFSLocalClient()
client.tar_and_copy_to_local(volume_dir, filename)
response = StreamingHttpResponse(big_file_iterator(filename))
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="{}"'.format(
os.path.basename(filename))
return response
class NFSLocalClientTestCase(unittest.TestCase):
def setUp(self):
self.client = NFSLocalClient()
def test_makedir(self):
self.client.makedir('/home/wangtao/testdir')
def test_removedir(self):
self.client.removedir('/home/wangtao/testdir')
def test_copy_file_to_remote(self):
self.client.copy_file_to_remote("/home/wangtao/test/test.tar",
"/hummer/test/test.tar")
def test_copy_file_to_local(self):
self.client.copy_file_to_local("/hummer/test/test.tar",
"/home/wangtao/test.tar")
def test_tar_and_copy_to_local(self):
self.client.tar_and_copy_to_local("/hummer/test",
"/home/wangtao/test.tar")
def test_copy_file_to_remote_and_untar(self):
self.client.copy_file_to_remote_and_untar(
"/home/wangtao/test/test.tar", "/hummer/test")
