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