#!/usr/bin/env python
-import errno, sys, stat, re
+import copy, errno, sys, stat, re
from bup import options, git, metadata, vfs
from bup.helpers import *
--
C,outdir= change to given outdir before extracting files
numeric-ids restore numeric IDs (user, group, etc.) rather than names
-exclude-rx= skip paths that match the unanchored regular expression
+exclude-rx= skip paths matching the unanchored regex (may be repeated)
+exclude-rx-from= skip --exclude-rx patterns in file (may be repeated)
v,verbose increase log output (can be used more than once)
+map-user= given OLD=NEW, restore OLD user as NEW user
+map-group= given OLD=NEW, restore OLD group as NEW group
+map-uid= given OLD=NEW, restore OLD uid as NEW uid
+map-gid= given OLD=NEW, restore OLD gid as NEW gid
q,quiet don't show progress meter
"""
elif stat.S_ISLNK(n.mode):
os.symlink(n.readlink(), fullname)
+
+def parse_owner_mappings(type, options, fatal):
+ """Traverse the options and parse all --map-TYPEs, or call Option.fatal()."""
+ opt_name = '--map-' + type
+ value_rx = r'^([^=]+)=([^=]*)$'
+ if type in ('uid', 'gid'):
+ value_rx = r'^(-?[0-9]+)=(-?[0-9]+)$'
+ owner_map = {}
+ for flag in options:
+ (option, parameter) = flag
+ if option != opt_name:
+ continue
+ match = re.match(value_rx, parameter)
+ if not match:
+ raise fatal("couldn't parse %s as %s mapping" % (parameter, type))
+ old_id, new_id = match.groups()
+ if type in ('uid', 'gid'):
+ old_id = int(old_id)
+ new_id = int(new_id)
+ owner_map[old_id] = new_id
+ return owner_map
+
+
+def apply_metadata(meta, name, restore_numeric_ids, owner_map):
+ m = copy.deepcopy(meta)
+ m.user = owner_map['user'].get(m.user, m.user)
+ m.group = owner_map['group'].get(m.group, m.group)
+ m.uid = owner_map['uid'].get(m.uid, m.uid)
+ m.gid = owner_map['gid'].get(m.gid, m.gid)
+ m.apply_to_path(name, restore_numeric_ids = restore_numeric_ids)
+
+
# Track a list of (restore_path, vfs_path, meta) triples for each path
# we've written for a given hardlink_target. This allows us to handle
# the case where we restore a set of hardlinks out of order (with
return False
target_node = top.lresolve(target_vfs_path)
if src_node.mode != target_node.mode \
- or src_node.atime != target_node.atime \
or src_node.mtime != target_node.mtime \
or src_node.ctime != target_node.ctime \
or src_node.hash != target_node.hash:
meta_stream.close()
-def do_root(n):
+def do_root(n, owner_map, restore_root_meta = True):
# Very similar to do_node(), except that this function doesn't
# create a path for n's destination directory (and so ignores
# n.fullname). It assumes the destination is '.', and restores
# Directory metadata is the first entry in any .bupm file in
# the directory. Get it.
mfile = n.metadata_file() # VFS file -- cannot close().
+ root_meta = None
if mfile:
meta_stream = mfile.open()
- meta = metadata.Metadata.read(meta_stream)
+ root_meta = metadata.Metadata.read(meta_stream)
print_info(n, '.')
total_restored += 1
plog('Restoring: %d\r' % total_restored)
# Don't get metadata if this is a dir -- handled in sub do_node().
if meta_stream and not stat.S_ISDIR(sub.mode):
m = metadata.Metadata.read(meta_stream)
- do_node(n, sub, m)
- if meta:
- meta.apply_to_path('.', restore_numeric_ids = opt.numeric_ids)
+ do_node(n, sub, owner_map, meta = m)
+ if root_meta and restore_root_meta:
+ apply_metadata(root_meta, '.', opt.numeric_ids, owner_map)
finally:
if meta_stream:
meta_stream.close()
-def do_node(top, n, meta=None):
+def do_node(top, n, owner_map, meta = None):
# Create n.fullname(), relative to the current directory, and
# restore all of its metadata, when available. The meta argument
# will be None for dirs, or when there is no .bupm (i.e. no
# Don't get metadata if this is a dir -- handled in sub do_node().
if meta_stream and not stat.S_ISDIR(sub.mode):
m = metadata.Metadata.read(meta_stream)
- do_node(top, sub, m)
+ do_node(top, sub, owner_map, meta = m)
if meta and not created_hardlink:
- meta.apply_to_path(fullname, restore_numeric_ids = opt.numeric_ids)
+ apply_metadata(meta, fullname, opt.numeric_ids, owner_map)
finally:
if meta_stream:
meta_stream.close()
exclude_rxs = parse_rx_excludes(flags, o.fatal)
+owner_map = {}
+for map_type in ('user', 'group', 'uid', 'gid'):
+ owner_map[map_type] = parse_owner_mappings(map_type, flags, o.fatal)
+
if opt.outdir:
mkdirp(opt.outdir)
os.chdir(opt.outdir)
if not isdir:
add_error('%r: not a directory' % d)
else:
- if name == '.':
- do_root(n)
- else:
- for sub in n:
- do_node(n, sub)
+ do_root(n, owner_map, restore_root_meta = (name == '.'))
else:
# Source is /foo/what/ever -- extract ./ever to cwd.
if isinstance(n, vfs.FakeSymlink):
target = n.dereference()
mkdirp(n.name)
os.chdir(n.name)
- do_root(target)
+ do_root(target, owner_map)
else: # Not a directory or fake symlink.
meta = find_dir_item_metadata_by_name(n.parent, n.name)
- do_node(n.parent, n, meta=meta)
+ do_node(n.parent, n, owner_map, meta = meta)
if not opt.quiet:
progress('Restoring: %d, done.\n' % total_restored)
projects / bup.git / blobdiff