Let shortcut.py take more target args, use betterhelp.

master
voussoir 2021-02-25 19:42:35 -08:00
parent 51188b30ee
commit af9d797b0e
No known key found for this signature in database
GPG Key ID: 5F7554F8C26DACCB
1 changed files with 31 additions and 3 deletions

View File

@ -2,7 +2,9 @@ import argparse
import sys import sys
import winshell import winshell
from voussoirkit import betterhelp
from voussoirkit import pathclass from voussoirkit import pathclass
from voussoirkit import subproctools
def shortcut(lnk_name, target, start_in=None, icon=None): def shortcut(lnk_name, target, start_in=None, icon=None):
lnk = pathclass.Path(lnk_name) lnk = pathclass.Path(lnk_name)
@ -11,6 +13,7 @@ def shortcut(lnk_name, target, start_in=None, icon=None):
lnk.assert_not_exists() lnk.assert_not_exists()
(target, args) = (target[0], target[1:])
target = pathclass.Path(target) target = pathclass.Path(target)
target.assert_exists() target.assert_exists()
@ -24,6 +27,9 @@ def shortcut(lnk_name, target, start_in=None, icon=None):
shortcut = winshell.Shortcut(lnk.absolute_path) shortcut = winshell.Shortcut(lnk.absolute_path)
shortcut.path = target.absolute_path shortcut.path = target.absolute_path
if args:
shortcut.arguments = subproctools.format_command(args)
if start_in is not None: if start_in is not None:
shortcut.working_directory = start_in.absolute_path shortcut.working_directory = start_in.absolute_path
if icon is not None: if icon is not None:
@ -32,6 +38,29 @@ def shortcut(lnk_name, target, start_in=None, icon=None):
shortcut.write() shortcut.write()
return lnk return lnk
DOCSTRING = '''
shortcut
========
> shortcut lnk_path target <flags>
lnk_path:
The filepath of the lnk file you want to create.
target:
The filepath of the target file and any additional arguments separated
by spaces. If you want to include an argument that starts with hyphens,
consider putting this last and use `--` to indicate the end of named
arguments. For example:
> shortcut game.lnk --icon game.ico -- javaw.exe -jar game.jar
--start-in:
Directory to use as CWD for the program.
--icon:
Path to an .ico file.
'''
def shortcut_argparse(args): def shortcut_argparse(args):
try: try:
lnk = shortcut( lnk = shortcut(
@ -50,13 +79,12 @@ def main(argv):
parser = argparse.ArgumentParser(description=__doc__) parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('lnk_name') parser.add_argument('lnk_name')
parser.add_argument('target') parser.add_argument('target', nargs='+')
parser.add_argument('--start_in', '--start-in', '--startin', default=None) parser.add_argument('--start_in', '--start-in', '--startin', default=None)
parser.add_argument('--icon', default=None) parser.add_argument('--icon', default=None)
parser.set_defaults(func=shortcut_argparse) parser.set_defaults(func=shortcut_argparse)
args = parser.parse_args(argv) return betterhelp.single_main(argv, parser, DOCSTRING)
return args.func(args)
if __name__ == '__main__': if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:])) raise SystemExit(main(sys.argv[1:]))