Make PackWriter a "with" context manager

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

diff --git a/lib/bup/git.py b/lib/bup/git.py
index db704485961cd38961d638a6506a23ca14482461..35bf838fe42ca875c8d707013771e74c8b29d946 100644 (file)
--- a/lib/bup/git.py
+++ b/lib/bup/git.py
@@ -637,6 +637,12 @@ class PackWriter:
     def __del__(self):
         self.close()
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        self.close()
+
     def _open(self):
         if not self.file:
             objdir = dir = os.path.join(self.repo_dir, 'objects')