[virt-tools-list] [virt-bootstrap] [PATCH v7 07/26] Get layer size if not provided

[Cedric Bosdonnat]

On Sat, 2017-08-26 at 21:41 +0100, Radostin Stoyanov wrote:
> Docker registry with Manifest v1 does not require the size of layers to
> be included. However, when this information is not provided we can use
> os.path.getsize() to get and show the size of tarball. We can also
> use this function for FileSource to provide consistent output messages.
> ---
>  src/virtBootstrap/sources/docker_source.py | 5 ++---
>  src/virtBootstrap/utils.py                 | 9 +++++----
>  2 files changed, 7 insertions(+), 7 deletions(-)
> 
> diff --git a/src/virtBootstrap/sources/docker_source.py b/src/virtBootstrap/sources/docker_source.py
> index 246356a..2dadb42 100644
> --- a/src/virtBootstrap/sources/docker_source.py
> +++ b/src/virtBootstrap/sources/docker_source.py
> @@ -92,15 +92,14 @@ class DockerSource(object):
>              raise ValueError('Unsupported manifest schema.')
>  
>          for layer in layers_list:
> -            layer_digest = layer[digest_field]
> -            layer_size = layer['size'] if 'size' in layer else None
> -
>              # Store checksums of layers
> +            layer_digest = layer[digest_field]
>              sum_type, layer_sum = layer_digest.split(':')
>              self.checksums.append([sum_type, layer_sum])
>  
>              # Store file path and size of each layer
>              file_path = os.path.join(self.images_dir, layer_sum + '.tar')
> +            layer_size = layer.get('size', None)
>              self.layers.append([file_path, layer_size])
>  
>      def gen_valid_uri(self, uri):
> diff --git a/src/virtBootstrap/utils.py b/src/virtBootstrap/utils.py
> index a04627e..267963f 100644
> --- a/src/virtBootstrap/utils.py
> +++ b/src/virtBootstrap/utils.py
> @@ -150,10 +150,11 @@ def log_layer_extract(tar_file, tar_size, current, total, progress):
>      """
>      Create log message on layer extract.
>      """
> -    msg = 'Extracting layer (%s/%s)' % (current, total)
> -
> -    if tar_size:
> -        msg += " with size: %s" % bytes_to_size(tar_size)
> +    msg = 'Extracting layer (%s/%s) with size: %s' % (
> +        current,
> +        total,
> +        bytes_to_size(tar_size or os.path.getsize(tar_file))
> +    )
>      progress(msg, logger=logger)
>      logger.debug('Untar layer: %s', tar_file)
>  

ACK

--
Cedric