Context managers on all connections

Author: Lukasa

hyper/common/connection.py

@@ -107,6 +107,15 @@ def request(self, method, url, body=None, headers={}):
                 method=method, url=url, body=body, headers=headers
             )
 
+    # The following two methods are the implementation of the context manager
+    # protocol.
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, tb):
+        self.close()
+        return False  # Never swallow exceptions.
+
     # Can anyone say 'proxy object pattern'?
     def __getattr__(self, name):
         return getattr(self._conn, name)

hyper/http11/connection.py

@@ -289,3 +289,12 @@ def close(self):
         """
         self._sock.close()
         self._sock = None
+
+    # The following two methods are the implementation of the context manager
+    # protocol.
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, tb):
+        self.close()
+        return False  # Never swallow exceptions.