Add dulwich.porcelain.write_tree.

NEWS

@@ -1,5 +1,10 @@
 0.19.10	UNRELEASED
 
+ IMPROVEMENTS
+
+ * Add `dulwich.porcelain.write_tree`.
+  (Jelmer Vernooij)
+
 0.19.9	2018-11-17
 
  BUG FIXES

bin/dulwich

@@ -388,6 +388,14 @@ class cmd_web_daemon(Command):
                              port=options.port)
 
 
+class cmd_write_tree(Command):
+
+    def run(self, args):
+        parser = optparse.OptionParser()
+        options, args = parser.parse_args(args)
+        sys.stdout.write('%s\n' % porcelain.write_tree('.'))
+
+
 class cmd_receive_pack(Command):
 
     def run(self, args):
@@ -689,6 +697,7 @@ commands = {
     "update-server-info": cmd_update_server_info,
     "upload-pack": cmd_upload_pack,
     "web-daemon": cmd_web_daemon,
+    "write-tree": cmd_write_tree,
     }
 
 if len(sys.argv) < 2:

dulwich/porcelain.py

@@ -1411,3 +1411,13 @@ def get_object_by_path(repo, path, committish=None):
             r.object_store.__getitem__,
             base_tree, path)
         return r[sha]
+
+
+def write_tree(repo):
+    """Write a tree object from the index.
+
+    :param repo: Repository for which to write tree
+    :return: tree id for the tree that was written
+    """
+    with open_repo_closing(repo) as r:
+        return r.open_index().commit(r.object_store)

dulwich/tests/test_porcelain.py

@@ -1562,3 +1562,15 @@ class GetObjectBypathTests(PorcelainTestCase):
         self.assertRaises(
             KeyError,
             porcelain.get_object_by_path, self.repo, 'foo')
+
+
+class WriteTreeTests(PorcelainTestCase):
+
+    def test_simple(self):
+        fullpath = os.path.join(self.repo.path, 'foo')
+        with open(fullpath, 'w') as f:
+            f.write("BAR")
+        porcelain.add(repo=self.repo.path, paths=[fullpath])
+        self.assertEqual(
+            b'd2092c8a9f311f0311083bf8d177f2ca0ab5b241',
+            porcelain.write_tree(self.repo))