Review Repository.write_archive

Author: jdavid

pygit2/repository.py

@@ -1166,18 +1166,17 @@ def write_archive(self, treeish, archive, timestamp=None, prefix=''):
         Example::
 
             >>> import tarfile, pygit2
-            >>>> with tarfile.open('foo.tar', 'w') as archive:
-            >>>>     repo = pygit2.Repository('.')
-            >>>>     repo.write_archive(repo.head.target, archive)
+            >>> with tarfile.open('foo.tar', 'w') as archive:
+            >>>     repo = pygit2.Repository('.')
+            >>>     repo.write_archive(repo.head.target, archive)
         """
 
         # Try to get a tree form whatever we got
-        if isinstance(treeish, Tree):
-            tree = treeish
-
         if isinstance(treeish, (str, Oid)):
             treeish = self[treeish]
 
+        tree = treeish.peel(Tree)
+
         # if we don't have a timestamp, try to get it from a commit
         if not timestamp:
             try:
@@ -1190,8 +1189,6 @@ def write_archive(self, treeish, archive, timestamp=None, prefix=''):
         if not timestamp:
             timestamp = int(time())
 
-        tree = treeish.peel(Tree)
-
         index = Index()
         index.read_tree(tree)