commit d03bb2294fcd1cdd79e147033f2a793f15f71fac
author Craig Citro <craigcitro@gmail.com> Tue Feb 17 15:20:54 2015 -0800
committer Craig Citro <craigcitro@gmail.com> Tue Feb 17 15:20:54 2015 -0800
tree 61039eda2915f0b3efa89f9f4b3d02f5140fc076
parent bbad4f0e89c7662a79a35bc3d88391086ca86f24
parent ac8df21304d366fffb4e312da02575f389252aa1

Merge pull request #51 from jacktasia/master

MediaIoBaseDownload hangs forever when object to download is smaller than DEFAULT_CHUNK_SIZE

googleapiclient/http.py

diff --git a/googleapiclient/http.py b/googleapiclient/http.py
index 3959d81..cdc9f58 100644
--- a/googleapiclient/http.py
+++ b/googleapiclient/http.py
@@ -559,6 +559,10 @@
         content_range = resp['content-range']
         length = content_range.rsplit('/', 1)[1]
         self._total_size = int(length)
+      elif 'content-length' in resp:
+        self._total_size = int(resp['content-length'])
+      else:
+        raise HttpError(resp, content, uri=self._uri)
 
       if self._progress == self._total_size:
         self._done = True