flags2_asyncio_executor.py closing http request

ramalho · ramalho · commit 3c25f1ed0fc9 · 2015-07-14T03:31:40.000-03:00
diff --git a/17-futures/countries/flags2_asyncio_executor.py b/17-futures/countries/flags2_asyncio_executor.py
@@ -6,6 +6,7 @@
 
 import asyncio
 import collections
+import contextlib
 
 import aiohttp
 from aiohttp import web
@@ -28,15 +29,17 @@ def __init__(self, country_code):
 def get_flag(base_url, cc):
     url = '{}/{cc}/{cc}.gif'.format(base_url, cc=cc.lower())
     resp = yield from aiohttp.request('GET', url)
-    if resp.status == 200:
-        image = yield from resp.read()
-        return image
-    elif resp.status == 404:
-        raise web.HTTPNotFound()
-    else:
-        raise aiohttp.HttpProcessingError(
-            code=resp.status, message=resp.reason,
-            headers=resp.headers)
+    with contextlib.closing(resp):
+        if resp.status == 200:
+            image = yield from resp.read()
+            return image
+        elif resp.status == 404:
+            raise web.HTTPNotFound()
+        else:
+            raise aiohttp.HttpProcessingError(
+                code=resp.status, message=resp.reason,
+                headers=resp.headers)
+
 
 # BEGIN FLAGS2_ASYNCIO_EXECUTOR
 @asyncio.coroutine
