allow to set a new_volume_handler in iterate_tar_path, but make it optional
authorEduardo Robles Elvira <edulix@wadobo.com>
Mon, 23 Jun 2014 12:11:35 +0000 (14:11 +0200)
committerEduardo Robles Elvira <edulix@wadobo.com>
Mon, 23 Jun 2014 12:11:35 +0000 (14:11 +0200)
deltatar/deltatar.py

index 560841a..82c85e4 100644 (file)
@@ -959,7 +959,7 @@ class DeltaTar(object):
 
         return IndexPathIterator(self, index_path)
 
-    def iterate_tar_path(self, tar_path):
+    def iterate_tar_path(self, tar_path, new_volume_handler=None):
         '''
         Returns a tar iterator that iterates jsonized member items that contain
         an additional "member" field, used by RestoreHelper.
@@ -970,7 +970,7 @@ class DeltaTar(object):
                 self.tar_path = tar_path
                 self.tar_obj = None
                 self.last_member = None
-                self.new_volume_handler = None
+                self.new_volume_handler = new_volume_handler
                 self.__enter__()
 
             def __iter__(self):
@@ -1038,7 +1038,7 @@ class DeltaTar(object):
                     u'member': tarinfo
                 }, 0
 
-        return TarPathIterator(self, tar_path)
+        return TarPathIterator(self, tar_path, new_volume_handler)
 
     def jsonize_path_iterator(self, iter, strip=0):
         '''