mirrors/dotdrop
1
0
Fork 0
mirror of https://github.com/deadc0de6/dotdrop.git synced 2026-02-09 19:19:15 +00:00
Code Activity

refuse to remove key dirs for #174

Browse Source
This commit is contained in:
deadc0de6
2019-07-02 23:58:37 +02:00
parent 49275d371d
commit d783911420
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dotdrop/utils.py
View File

@@ -19,6 +19,13 @@ from dotdrop.logger import Logger
LOG = Logger() LOG = Logger()
STAR = '*' STAR = '*'
# files dotdrop refuses to remove
DONOTDELETE = [
os.path.expanduser('~'),
os.path.expanduser('~/.config'),
]
NOREMOVE = [os.path.normpath(p) for p in DONOTDELETE]
def run(cmd, raw=True, debug=False, checkerr=False): def run(cmd, raw=True, debug=False, checkerr=False):
"""run a command (expects a list)""" """run a command (expects a list)"""
@@ -89,6 +96,10 @@ def remove(path):
"""remove a file/directory/symlink""" """remove a file/directory/symlink"""
if not os.path.lexists(path): if not os.path.lexists(path):
raise OSError("File not found: {}".format(path)) raise OSError("File not found: {}".format(path))
if os.path.normpath(os.path.expanduser(path)) in NOREMOVE:
err = 'Dotdrop refuses to remove {}'.format(path)
LOG.err(err)
raise OSError(err)
if os.path.islink(path) or os.path.isfile(path): if os.path.islink(path) or os.path.isfile(path):
os.unlink(path) os.unlink(path)
elif os.path.isdir(path): elif os.path.isdir(path):