Add argument --recurse.
This commit is contained in:
parent
3d61b3b1ec
commit
2e9d8cf82a
1 changed files with 9 additions and 4 deletions
|
@ -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(recurse=False, autoyes=False):
|
||||||
|
if recurse:
|
||||||
|
lnks = [file for file in spinal.walk_generator('.') if file.extension == 'lnk']
|
||||||
|
else:
|
||||||
|
lnks = pathclass.Path('.').glob('*.lnk')
|
||||||
|
|
||||||
def prune_shortcuts(autoyes=False):
|
|
||||||
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)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue