index.Writer.__del__: replace with context management

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

diff --git a/lib/bup/index.py b/lib/bup/index.py
index e7214d1daca24c79a7598b6822d10c9f050a4541..b9a4013be032bfd75d68e350250a43d4b9571d85 100644 (file)
--- a/lib/bup/index.py
+++ b/lib/bup/index.py
@@ -537,8 +537,12 @@ class Writer:
         self.f = os.fdopen(ffd, 'wb', 65536)
         self.f.write(INDEX_HDR)
 
-    def __del__(self):
-        self.abort()
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        with pending_raise(value, rethrow=False):
+            self.abort()
 
     def abort(self):
         f = self.f