# HG changeset patch # User Bryan O'Sullivan # Date 2016-01-15 21:14:50 # Node ID 8953e963ce8c8858c39a64db02a2d1dd97d85129 # Parent 054abf2377e806d9ba51130dd3f9db2670d8bdf5 sshpeer: make remotelock a context manager diff --git a/mercurial/sshpeer.py b/mercurial/sshpeer.py --- a/mercurial/sshpeer.py +++ b/mercurial/sshpeer.py @@ -22,6 +22,11 @@ class remotelock(object): def release(self): self.repo.unlock() self.repo = None + def __enter__(self): + return self + def __exit__(self, exc_type, exc_val, exc_tb): + if self.repo: + self.release() def __del__(self): if self.repo: self.release()