Mercurial > public > mercurial-scm > hg-stable
diff mercurial/sshpeer.py @ 27798:8953e963ce8c
sshpeer: make remotelock a context manager
author | Bryan O'Sullivan <bryano@fb.com> |
---|---|
date | Fri, 15 Jan 2016 13:14:50 -0800 |
parents | 1aee2ab0f902 |
children | 48fd02dac1d4 |
line wrap: on
line diff
--- a/mercurial/sshpeer.py Fri Jan 15 13:14:45 2016 -0800 +++ b/mercurial/sshpeer.py Fri Jan 15 13:14:50 2016 -0800 @@ -22,6 +22,11 @@ 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()