fix hanging on chunked encoding

tsloughter · Jun 10, 2012 · be779cb · be779cb
1 parent fe5ac05
commit be779cb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/disco/comm.py b/lib/disco/comm.py
@@ -161,6 +161,8 @@ def __iter__(self):
     def __len__(self):
         if 'content-range' in self.headers:
             return int(self.headers['content-range'].split('/')[1])
+        elif self.headers.get('transfer-encoding') == 'chunked':
+            return len(self.buf)
         return int(self.headers.get('content-length', 0))
 
     def close(self):