def __generate_manifest(self, vol_container, file_size): days = self.args.get('expires') or 30 timeout = days * 86400 # in seconds _, bucket, key = self.args['s3_service'].resolve_url_to_location( vol_container['image']['importManifestUrl']) key_prefix = key.rsplit('/', 1)[0] manifest = ImportManifest(loglevel=self.log.level) manifest.file_format = vol_container['image']['format'] delete_req = DeleteObject.from_other( self, service=self.args['s3_service'], auth=self.args['s3_auth'], path='/'.join((bucket, key))) manifest.self_destruct_url = delete_req.get_presigned_url2(timeout) manifest.image_size = int(vol_container['image']['size']) manifest.volume_size = int(vol_container['volume']['size']) part_size = (self.args.get('part_size') or 10) * 2 ** 20 # MiB for index, part_start in enumerate(six.moves.range(0, file_size, part_size)): part = ImportImagePart() part.index = index part.start = part_start part.end = min(part_start + part_size, file_size) - 1 part.key = '{0}/{1}.part.{2}'.format( key_prefix, os.path.basename(self.args['source']), index) part_path = '/'.join((bucket, part.key)) head_req = HeadObject.from_other(delete_req, path=part_path) get_req = GetObject.from_other(delete_req, source=part_path) delete_req = DeleteObject.from_other(delete_req, path=part_path) part.head_url = head_req.get_presigned_url2(timeout) part.get_url = get_req.get_presigned_url2(timeout) part.delete_url = delete_req.get_presigned_url2(timeout) manifest.image_parts.append(part) return manifest
def main(self): if self.args.get('dry_run'): return if self.args.get('show_progress', False): print 'Uploading image for task', self.args['task'] # Manifest desc_conv = DescribeConversionTasks.from_other( self, ConversionTaskId=[self.args['task']]) task = desc_conv.main()['conversionTasks'][0] assert task['conversionTaskId'] == self.args['task'] if task.get('importVolume'): vol_container = task['importVolume'] else: vol_container = task['importInstance']['volumes'][0] file_size = euca2ools.util.get_filesize(self.args['source']) manifest = self.__get_or_create_manifest(vol_container, file_size) file_size_from_manifest = manifest.image_parts[-1].end + 1 if file_size_from_manifest != file_size: raise ArgumentError( 'file "{0}" is not the same size as the file the import ' 'started with (expected: {1}, actual: {2})' .format(self.args['source'], file_size_from_manifest, file_size)) # Now we have a manifest; check to see what parts are already uploaded _, bucket, _ = self.args['s3_service'].resolve_url_to_location( vol_container['image']['importManifestUrl']) pbar_label_template = euca2ools.util.build_progressbar_label_template( [os.path.basename(part.key) for part in manifest.image_parts]) for part in manifest.image_parts: part_s3path = '/'.join((bucket, part.key)) head_req = HeadObject.from_other( self, service=self.args['s3_service'], auth=self.args['s3_auth'], path=part_s3path) try: head_req.main() except AWSError as err: if err.status_code == 404: self.__upload_part(part, part_s3path, pbar_label_template) else: raise