voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix symlink pruning not checking for directory links.

Browse source
This commit is contained in:
voussoir 2021-01-28 17:05:20 -08:00
parent 1511852e94
commit 0d918efe4c
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 20 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
frontends/etiquette_cli.py
View file

@ -253,14 +253,18 @@ def export_symlinks_argparse(args):
) )
total_paths.update(export) total_paths.update(export)
if args.prune and not args.dry_run: if not args.prune or args.dry_run:
symlinks = set(file for file in spinal.walk_generator(destination) if file.is_link) return
symlinks = spinal.walk_generator(destination, yield_directories=True, yield_files=True)
symlinks = set(path for path in symlinks if path.is_link)
symlinks = symlinks.difference(total_paths) symlinks = symlinks.difference(total_paths)
for old_symlink in symlinks: for old_symlink in symlinks:
print(f'Pruning {old_symlink}.') print(f'Pruning {old_symlink}.')
os.remove(old_symlink.absolute_path) os.remove(old_symlink.absolute_path)
if not old_symlink.parent.listdir(): if not old_symlink.parent.listdir():
os.rmdir(old_symlink.parent.absolute_path) os.rmdir(old_symlink.parent.absolute_path)
checkdirs = set(spinal.walk_generator(destination, yield_directories=True, yield_files=False)) checkdirs = set(spinal.walk_generator(destination, yield_directories=True, yield_files=False))
while checkdirs: while checkdirs:
check = checkdirs.pop() check = checkdirs.pop()