Added Repo.set_description method.

Signed-off-by: Jelmer Vernooij <jelmer@samba.org>

dulwich/repo.py

@@ -981,6 +981,13 @@ class BaseRepo(object):
         """
         raise NotImplementedError(self.get_description)
 
+    def set_description(self, description):
+        """Set the description for this repository.
+
+        :param description: Text to set as description for this repository.
+        """
+        raise NotImplementedError(self.set_description)
+
     def get_config_stack(self):
         """Return a config stack for this repository.
 
@@ -1473,6 +1480,20 @@ class Repo(BaseRepo):
     def __repr__(self):
         return "<Repo at %r>" % self.path
 
+    def set_description(self, description):
+        """Set the description for this repository.
+
+        :param description: Text to set as description for this repository.
+        """
+
+        path = os.path.join(self._controldir, 'description')
+        try:
+            with open(path, 'w') as f:
+                f.write(description)
+        except (IOError, OSError), e:
+            if e.errno != errno.ENOENT:
+                raise
+
     @classmethod
     def _init_maybe_bare(cls, path, bare):
         for d in BASE_DIRECTORIES:

dulwich/tests/test_repository.py

@@ -178,6 +178,13 @@ class RepositoryTests(TestCase):
             f.close()
         self.assertEquals("Some description", r.get_description())
 
+    def test_set_description(self):
+        r = self._repo = open_repo('a.git')
+        f = open(os.path.join(r.path, 'description'), 'w')
+        description = "Some description"
+        r.set_description(description)
+        self.assertEquals(description, r.get_description())
+
     def test_contains_missing(self):
         r = self._repo = open_repo('a.git')
         self.assertFalse("bar" in r)