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()