Add any pending exception to context in all __exit__ methods

[bup.git] / lib / bup / repo.py

diff --git a/lib/bup/repo.py b/lib/bup/repo.py
index dd6ef3f90674537ee2813a12d1530bf94e4c32ac..d52aa0bdc3734d32ab31ff00d3ba307759e6d2fb 100644 (file)
--- a/lib/bup/repo.py
+++ b/lib/bup/repo.py
@@ -4,7 +4,7 @@ from os.path import realpath
 from functools import partial
 
 from bup import client, git, vfs
-
+from bup.compat import pending_raise
 
 _next_repo_id = 0
 _repo_ids = {}
@@ -36,7 +36,8 @@ class LocalRepo:
         return self
 
     def __exit__(self, type, value, traceback):
-        self.close()
+        with pending_raise(value, rethrow=False):
+            self.close()
 
     def id(self):
         """Return an identifier that differs from any other repository that
@@ -106,7 +107,8 @@ class RemoteRepo:
         return self
 
     def __exit__(self, type, value, traceback):
-        self.close()
+        with pending_raise(value, rethrow=False):
+            self.close()
 
     def id(self):
         """Return an identifier that differs from any other repository that