Connection as a context manager

diff --git a/remoto/connection.py b/remoto/connection.py
index 93e18663a90bb4d3a5e67b9d8d54941894ac9296..9d9370ef607274a1a8ef0006eab722454e9fcdea 100644 (file)
--- a/remoto/connection.py
+++ b/remoto/connection.py
@@ -13,6 +13,13 @@ class Connection(object):
         self.logger = logger or FakeRemoteLogger()
         self.sudo = sudo
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.exit()
+        return False
+
     def execute(self, function, **kw):
         return self.gateway.remote_exec(function, **kw)