Context manage connections (via BaseConn)

[bup.git] / lib / bup / cmd / on.py

diff --git a/lib/bup/cmd/on.py b/lib/bup/cmd/on.py
index bb5e3f3117a41eded776545426b66672bc63dd42..983283cebf45730406cb23034c0cb05a660bea26 100755 (executable)
--- a/lib/bup/cmd/on.py
+++ b/lib/bup/cmd/on.py
@@ -56,9 +56,9 @@ def main(argv):
             p.stdin.close()
             p.stdout.close()
             # Demultiplex remote client's stderr (back to stdout/stderr).
-            dmc = DemuxConn(p.stderr.fileno(), open(os.devnull, "wb"))
-            for line in iter(dmc.readline, b''):
-                out.write(line)
+            with DemuxConn(p.stderr.fileno(), open(os.devnull, "wb")) as dmc:
+                for line in iter(dmc.readline, b''):
+                    out.write(line)
         finally:
             while 1:
                 # if we get a signal while waiting, we have to keep waiting, just