psi-jack/dotbot
1
0
Fork 0
mirror of synced 2024-06-26 02:41:10 -04:00
Code Issues Releases Wiki Activity

Add backup option

Browse source
This commit is contained in:
Jonas De 2019-08-16 07:59:40 +02:00
parent 9a8d292681
commit 457b38326c
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
dotbot/plugins/link.py
View file

@ -27,6 +27,7 @@ class Link(dotbot.Plugin):
destination = os.path.expandvars(destination) destination = os.path.expandvars(destination)
relative = defaults.get('relative', False) relative = defaults.get('relative', False)
force = defaults.get('force', False) force = defaults.get('force', False)
backup = defaults.get('backup', False)
relink = defaults.get('relink', False) relink = defaults.get('relink', False)
create = defaults.get('create', False) create = defaults.get('create', False)
use_glob = defaults.get('glob', False) use_glob = defaults.get('glob', False)
@ -36,6 +37,7 @@ class Link(dotbot.Plugin):
test = source.get('if', test) test = source.get('if', test)
relative = source.get('relative', relative) relative = source.get('relative', relative)
force = source.get('force', force) force = source.get('force', force)
backup = defaults.get('backup', force)
relink = source.get('relink', relink) relink = source.get('relink', relink)
create = source.get('create', create) create = source.get('create', create)
use_glob = source.get('glob', use_glob) use_glob = source.get('glob', use_glob)
@ -66,7 +68,7 @@ class Link(dotbot.Plugin):
if create: if create:
success &= self._create(destination) success &= self._create(destination)
if force or relink: if force or relink:
success &= self._delete(path, destination, relative, force) success &= self._delete(path, destination, relative, force, backup)
success &= self._link(path, destination, relative) success &= self._link(path, destination, relative)
else: else:
self._log.lowinfo("Globs from '" + path + "': " + str(glob_results)) self._log.lowinfo("Globs from '" + path + "': " + str(glob_results))
@ -77,7 +79,7 @@ class Link(dotbot.Plugin):
if create: if create:
success &= self._create(glob_link_destination) success &= self._create(glob_link_destination)
if force or relink: if force or relink:
success &= self._delete(glob_full_item, glob_link_destination, relative, force) success &= self._delete(glob_full_item, glob_link_destination, relative, force, backup)
success &= self._link(glob_full_item, glob_link_destination, relative) success &= self._link(glob_full_item, glob_link_destination, relative)
else: else:
if create: if create:
@ -88,7 +90,7 @@ class Link(dotbot.Plugin):
(destination, path)) (destination, path))
continue continue
if force or relink: if force or relink:
success &= self._delete(path, destination, relative, force) success &= self._delete(path, destination, relative, force, backup)
success &= self._link(path, destination, relative) success &= self._link(path, destination, relative)
if success: if success:
self._log.info('All links have been set up') self._log.info('All links have been set up')
@ -153,7 +155,7 @@ class Link(dotbot.Plugin):
self._log.lowinfo('Creating directory %s' % parent) self._log.lowinfo('Creating directory %s' % parent)
return success return success
def _delete(self, source, path, relative, force): def _delete(self, source, path, relative, force, backup):
success = True success = True
source = os.path.join(self._context.base_directory(), source) source = os.path.join(self._context.base_directory(), source)
fullpath = os.path.expanduser(path) fullpath = os.path.expanduser(path)
@ -168,9 +170,13 @@ class Link(dotbot.Plugin):
removed = True removed = True
elif force: elif force:
if os.path.isdir(fullpath): if os.path.isdir(fullpath):
if backup:
shutil.copytree(fullpath, backup)
shutil.rmtree(fullpath) shutil.rmtree(fullpath)
removed = True removed = True
else: else:
if backup:
shutil.copyfile(fullpath, backup)
os.remove(fullpath) os.remove(fullpath)
removed = True removed = True
except OSError: except OSError: