def restore_image(self): """ """ information = Information(self.image_path) information.load() image_name = information.get_image_name() compressor_level = information.get_image_compressor_level() total_bytes = information.get_image_total_bytes() total_blocks = long(math.ceil(total_bytes/float(BLOCK_SIZE))) device = Device(self.target_device) if device.is_disk() != \ information.get_image_is_disk(): raise ErrorRestoringImage("Invalid target dispositive") try: disk = Disk(device) except _ped.DiskLabelException: try: device.fix_disk_label() disk = Disk(device) except: raise ErrorRestoringImage("Unrecognized disk label") if information.get_image_is_disk(): mbr = Mbr(self.image_path) mbr.restore_from_file(self.target_device) dlm = DiskLayoutManager(self.image_path) dlm.restore_from_file(disk) self._print_informations(total_bytes, image_name) progress = Progress(total_blocks) progress.start() partitions = information.get_partitions() for part in partitions: if information.get_image_is_disk(): partition = disk.get_partition_by_number(part.number, part.type) else: parent_path = get_parent_path(self.target_device) parent_device = Device(parent_path) parent_disk = Disk(parent_device) partition = parent_disk.get_partition_by_path( self.target_device, part.type) if partition is None: raise ErrorRestoringImage("No valid partitions found") if hasattr(part, "uuid"): partition.filesystem.open_to_write(part.uuid) else: partition.filesystem.open_to_write() if partition.filesystem.is_swap(): continue current_volume = 1 while True: file_name = FILE_PATTERN % (image_name, part.number, current_volume) file_path = self.image_path + file_name reader = ReaderFactory(file_path, compressor_level) for data in reader: partition.filesystem.write(data) progress.increment(1) if hasattr(part, "volumes"): if current_volume < part.volumes: current_volume += 1 continue break partition.filesystem.close() progress.stop() print "completed."
def restore_image(self): """ """ information = Information(self.image_path) information.load() image_name = information.get_image_name() compressor_level = information.get_image_compressor_level() total_bytes = information.get_image_total_bytes() total_blocks = long(math.ceil(total_bytes / float(BLOCK_SIZE))) compressor = Compressor(compressor_level) device = Device(self.target_device) if device.is_disk() != \ information.get_image_is_disk(): raise ErrorRestoringImage("Invalid target dispositive") try: disk = Disk(device) except _ped.DiskLabelException: try: device.fix_disk_label() disk = Disk(device) except: raise ErrorRestoringImage("Unrecognized disk label") if information.get_image_is_disk(): mbr = Mbr(self.image_path) mbr.restore_from_file(self.target_device) dlm = DiskLayoutManager(self.image_path) dlm.restore_from_file(disk) self._print_informations(total_bytes, image_name) progress = Progress(total_blocks) progress.start() partitions = information.get_partitions() for p in partitions: if information.get_image_is_disk(): partition = disk.get_partition_by_number(p.number, p.type) else: parent_path = get_parent_path(self.target_device) parent_device = Device(parent_path) parent_disk = Disk(parent_device) partition = parent_disk.get_partition_by_path( self.target_device, p.type) if partition is None: raise ErrorRestoringImage("No valid partitions found") if p.uuid is not None: partition.filesystem.open_to_write(p.uuid) else: partition.filesystem.open_to_write() if partition.filesystem.is_swap(): continue file_name = FILE_PATTERN % (image_name, p.number) with open(self.image_path + file_name, 'rb') as f: # TODO: Work with parallelism while True: # Ugly header = f.read(compressor.get_header_size()) if not len(header): break size = compressor.read_block_header(header) cdata = f.read(size) data = compressor.extract(cdata) partition.filesystem.write(data) progress.increment(1) partition.filesystem.close() progress.stop() print "completed."
def restore_image(self): """ """ information = Information(self.image_path) information.load() image_name = information.get_image_name() compressor_level = information.get_image_compressor_level() total_bytes = information.get_image_total_bytes() total_blocks = long(math.ceil(total_bytes/float(BLOCK_SIZE))) compressor = Compressor(compressor_level) device = Device(self.target_device) if device.is_disk() != \ information.get_image_is_disk(): raise ErrorRestoringImage("Invalid target dispositive") try: disk = Disk(device) except _ped.DiskLabelException: try: device.fix_disk_label() disk = Disk(device) except: raise ErrorRestoringImage("Unrecognized disk label") if information.get_image_is_disk(): mbr = Mbr(self.image_path) mbr.restore_from_file(self.target_device) dlm = DiskLayoutManager(self.image_path) dlm.restore_from_file(disk) self._print_informations(total_bytes, image_name) progress = Progress(total_blocks) progress.start() partitions = information.get_partitions() for p in partitions: if information.get_image_is_disk(): partition = disk.get_partition_by_number(p.number, p.type) else: parent_path = get_parent_path(self.target_device) parent_device = Device(parent_path) parent_disk = Disk(parent_device) partition = parent_disk.get_partition_by_path( self.target_device, p.type) if partition is None: raise ErrorRestoringImage("No valid partitions found") if p.uuid is not None: partition.filesystem.open_to_write(p.uuid) else: partition.filesystem.open_to_write() if partition.filesystem.is_swap(): continue file_name = FILE_PATTERN % (image_name, p.number) with open(self.image_path + file_name, 'rb') as f: # TODO: Work with parallelism while True: # Ugly header = f.read(compressor.get_header_size()) if not len(header): break size = compressor.read_block_header(header) cdata = f.read(size) data = compressor.extract(cdata) partition.filesystem.write(data) progress.increment(1) partition.filesystem.close() progress.stop() print "completed."