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
