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

Add argument --recurse.

Browse source
This commit is contained in:
voussoir 2020-08-20 11:29:10 -07:00
parent 3d61b3b1ec
commit 2e9d8cf82a
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
prune_shortcuts.py
View file

@ -10,10 +10,14 @@ import winshell
from voussoirkit import getpermission from voussoirkit import getpermission
from voussoirkit import pathclass from voussoirkit import pathclass
from voussoirkit import spinal
def prune_shortcuts(autoyes=False): def prune_shortcuts(recurse=False, autoyes=False):
if recurse:
lnks = [file for file in spinal.walk_generator('.') if file.extension == 'lnk']
else:
lnks = pathclass.Path('.').glob('*.lnk') lnks = pathclass.Path('.').glob('*.lnk')
stale = [lnk for lnk in lnks if not os.path.exists(winshell.Shortcut(lnk.absolute_path).path)]
stale = [] stale = []
for lnk in lnks: for lnk in lnks:
shortcut = winshell.Shortcut(lnk.absolute_path) shortcut = winshell.Shortcut(lnk.absolute_path)
@ -40,11 +44,12 @@ def prune_shortcuts(autoyes=False):
send2trash.send2trash(lnk.absolute_path) send2trash.send2trash(lnk.absolute_path)
def prune_shortcuts_argparse(args): def prune_shortcuts_argparse(args):
return prune_shortcuts(autoyes=args.autoyes) return prune_shortcuts(recurse=args.recurse, autoyes=args.autoyes)
def main(argv): def main(argv):
parser = argparse.ArgumentParser(description=__doc__) parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('--recurse', dest='recurse', action='store_true')
parser.add_argument('--yes', dest='autoyes', action='store_true') parser.add_argument('--yes', dest='autoyes', action='store_true')
parser.set_defaults(func=prune_shortcuts_argparse) parser.set_defaults(func=prune_shortcuts_argparse)