Enable client V2 to download images

Added the CLI option image-download to download an image via API V2.
Added utility function to save an image.
Added common iterator to validate the checksum.

Related to bp glance-client-v2

Change-Id: I0247f5a3462142dc5e9f3dc16cbe00c8e3d42f42

1 files changed, 38 insertions, 0 deletions

diff --git a/glanceclient/common/utils.py b/glanceclient/common/utils.py
index 957f713..31b8530 100644
--- a/glanceclient/common/utils.py
+++ b/glanceclient/common/utils.py
@@ -13,6 +13,8 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
+import errno
+import hashlib
 import os
 import sys
 import uuid
@@ -130,3 +132,39 @@ def exit(msg=''):
     if msg:
         print >> sys.stderr, msg
     sys.exit(1)
+
+
+def save_image(data, path):
+    """
+    Save an image to the specified path.
+
+    :param data: binary data of the image
+    :param path: path to save the image to
+    """
+    if path is None:
+        image = sys.stdout
+    else:
+        image = open(path, 'wb')
+    try:
+        for chunk in data:
+            image.write(chunk)
+    finally:
+        if path is not None:
+            image.close()
+
+
+def integrity_iter(iter, checksum):
+    """
+    Check image data integrity.
+
+    :raises: IOError
+    """
+    md5sum = hashlib.md5()
+    for chunk in iter:
+        yield chunk
+        md5sum.update(chunk)
+    md5sum = md5sum.hexdigest()
+    if md5sum != checksum:
+        raise IOError(errno.EPIPE,
+                      'Corrupt image download. Checksum was %s expected %s' %
+                      (md5sum, checksum))