Преглед изворни кода

extract: small bugfix and refactoring for parent dir creation

make_parent(path) helper to reduce code duplication.
also use it for directories although makedirs can also do it.

bugfix: also create parent dir for device files, if needed.

(cherry picked from commit d4e27e2952888d547c54d5b0f4be22892aa9e2f1)
Thomas Waldmann пре 8 година
родитељ
комит
ff4f04e5f1
1 измењених фајлова са 13 додато и 8 уклоњено
  1. 13 8
      borg/archive.py

+ 13 - 8
borg/archive.py

@@ -389,11 +389,16 @@ Number of files: {0.stats.nfiles}'''.format(
             raise self.IncompatibleFilesystemEncodingError(path, sys.getfilesystemencoding()) from None
         except OSError:
             pass
+
+        def make_parent(path):
+            parent_dir = os.path.dirname(path)
+            if not os.path.exists(parent_dir):
+                os.makedirs(parent_dir)
+
         mode = item[b'mode']
         if stat.S_ISREG(mode):
-            if not os.path.exists(os.path.dirname(path)):
-                with backup_io():
-                    os.makedirs(os.path.dirname(path))
+            with backup_io():
+                make_parent(path)
             # Hard link?
             if b'source' in item:
                 source = os.path.join(dest, item[b'source'])
@@ -425,13 +430,13 @@ Number of files: {0.stats.nfiles}'''.format(
         with backup_io():
             # No repository access beyond this point.
             if stat.S_ISDIR(mode):
+                make_parent(path)
                 if not os.path.exists(path):
-                    os.makedirs(path)
+                    os.mkdir(path)
                 if restore_attrs:
                     self.restore_attrs(path, item)
             elif stat.S_ISLNK(mode):
-                if not os.path.exists(os.path.dirname(path)):
-                    os.makedirs(os.path.dirname(path))
+                make_parent(path)
                 source = item[b'source']
                 if os.path.exists(path):
                     os.unlink(path)
@@ -441,11 +446,11 @@ Number of files: {0.stats.nfiles}'''.format(
                     raise self.IncompatibleFilesystemEncodingError(source, sys.getfilesystemencoding()) from None
                 self.restore_attrs(path, item, symlink=True)
             elif stat.S_ISFIFO(mode):
-                if not os.path.exists(os.path.dirname(path)):
-                    os.makedirs(os.path.dirname(path))
+                make_parent(path)
                 os.mkfifo(path)
                 self.restore_attrs(path, item)
             elif stat.S_ISCHR(mode) or stat.S_ISBLK(mode):
+                make_parent(path)
                 os.mknod(path, item[b'mode'], item[b'rdev'])
                 self.restore_attrs(path, item)
             else: