better ClientResponse.release method

aio-libs · Jul 4, 2014 · 2e6b6e7 · 2e6b6e7 · popravich · Jul 4, 2014
1 parent 750a095
commit 2e6b6e7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/aiohttp/client.py b/aiohttp/client.py
@@ -663,7 +663,14 @@ def close(self, force=False):
 
     @asyncio.coroutine
     def release(self):
-        yield from self.read()
+        force = False
+        try:
+            while True:
+                yield from self.content.read()
+        except Exception as exc:
+            force = not isinstance(exc, aiohttp.EofStream)
+        finally:
+            self.close(force=force)
 
     @asyncio.coroutine
     def wait_for_close(self):