daylin/viv
1
0
Fork 0
mirror of https://github.com/daylinmorgan/viv.git synced 2024-09-19 19:47:18 -05:00
Code Issues Projects Releases Packages Wiki Activity

refactor: make aliases default to first letter

Browse source
This commit is contained in:
Daylin Morgan 2022-12-26 08:20:13 -06:00
parent a9cd2a4de1
commit a5803b087e
1 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/viv/viv.py
View file

@ -782,10 +782,11 @@ class Viv:
vivenv.dump_info() vivenv.dump_info()
def _get_subcmd_parser(self, subparsers, name: str, **kwargs): def _get_subcmd_parser(self, subparsers, name: str, **kwargs) -> ArgumentParser:
aliases = kwargs.pop("aliases", [name[0]])
cmd = getattr(self, name) cmd = getattr(self, name)
parser = subparsers.add_parser( parser = subparsers.add_parser(
name, help=cmd.__doc__, description=cmd.__doc__, **kwargs name, help=cmd.__doc__, description=cmd.__doc__, aliases=aliases, **kwargs
) )
parser.set_defaults(func=cmd) parser.set_defaults(func=cmd)
@ -807,7 +808,10 @@ class Viv:
) )
p_vivenv_arg = ArgumentParser(add_help=False) p_vivenv_arg = ArgumentParser(add_help=False)
p_vivenv_arg.add_argument("vivenv", help="name/hash of vivenv") p_vivenv_arg.add_argument("vivenv", help="name/hash of vivenv")
p_list = self._get_subcmd_parser(subparsers, "list", aliases=["l"]) p_list = self._get_subcmd_parser(
subparsers,
"list",
)
p_list.add_argument( p_list.add_argument(
"-q", "-q",
@ -817,7 +821,10 @@ class Viv:
default=False, default=False,
) )
p_exe = self._get_subcmd_parser(subparsers, "exe", aliases=["e"]) p_exe = self._get_subcmd_parser(
subparsers,
"exe",
)
p_exe_sub = p_exe.add_subparsers( p_exe_sub = p_exe.add_subparsers(
title="subcommand", metavar="<sub-cmd>", required=True title="subcommand", metavar="<sub-cmd>", required=True
) )
@ -847,7 +854,10 @@ class Viv:
) )
p_remove.add_argument("vivenv", help="name/hash of vivenv", nargs="*") p_remove.add_argument("vivenv", help="name/hash of vivenv", nargs="*")
p_freeze = self._get_subcmd_parser(subparsers, "freeze", aliases=["f"]) p_freeze = self._get_subcmd_parser(
subparsers,
"freeze",
)
p_freeze.add_argument( p_freeze.add_argument(
"-p", "-p",
"--path", "--path",
@ -871,7 +881,6 @@ class Viv:
self._get_subcmd_parser( self._get_subcmd_parser(
subparsers, subparsers,
"info", "info",
aliases=["i"],
parents=[p_vivenv_arg], parents=[p_vivenv_arg],
) )