diff options
author | Louis Taylor <kragniz@gmail.com> | 2014-11-01 20:49:03 +0000 |
---|---|---|
committer | Louis Taylor <kragniz@gmail.com> | 2015-01-04 20:52:00 +0000 |
commit | df02ee8e2a10d2f37a9c013dec157c88b8dce49d (patch) | |
tree | 6adf239757b8880c0d6d0e5b70e110d5625442ae /glanceclient/common/utils.py | |
parent | 7ce1ff9325be702384d728ad6a94f597680e1707 (diff) | |
download | python-glanceclient-df02ee8e2a10d2f37a9c013dec157c88b8dce49d.tar.gz |
Fix Requests breaking download progress bar
The move to the requests library (dbb242b) broke the progress bar during
downloading an image with --progress enabled, due to requests returning
a generator, which has no __len__ function. This patch adds an iterable
wrapper which provides the length, sourced from Content-Length headers.
Closes-Bug: #1384664
Change-Id: I48598a824396bc6e7cba69eb3ce32e7c8f30a18a
Diffstat (limited to 'glanceclient/common/utils.py')
-rw-r--r-- | glanceclient/common/utils.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/glanceclient/common/utils.py b/glanceclient/common/utils.py index fe7a51a..61c526e 100644 --- a/glanceclient/common/utils.py +++ b/glanceclient/common/utils.py @@ -421,3 +421,18 @@ def safe_header(name, value): return name, "{SHA1}%s" % d else: return name, value + + +class IterableWithLength(object): + def __init__(self, iterable, length): + self.iterable = iterable + self.length = length + + def __iter__(self): + return self.iterable + + def next(self): + return next(self.iterable) + + def __len__(self): + return self.length |