Add `Repo.close()`.

dulwich/object_store.py

@@ -248,6 +248,12 @@ class BaseObjectStore(object):
         """Close any files opened by this object store."""
         # Default implementation is a NO-OP
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
 
 class PackBasedObjectStore(BaseObjectStore):
 

dulwich/repo.py

@@ -916,6 +916,14 @@ class Repo(BaseRepo):
 
     create = init_bare
 
+    def close(self):
+        self.object_store.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
 
 class MemoryRepo(BaseRepo):
     """Repo that stores refs, objects, and named files in memory.

dulwich/tests/compat/server_utils.py

@@ -55,6 +55,9 @@ class _StubRepo(object):
 
         os.mkdir(self.path)
 
+    def close(self):
+        pass
+
 
 def _get_shallow(repo):
     shallow_file = repo.get_named_file('shallow')