daylin/viv
1
0
Fork 0
mirror of https://github.com/daylinmorgan/viv.git synced 2024-11-13 20:47:53 -06:00
Code Issues Projects Releases Packages Wiki Activity

refactor: shim/freeze overlap

Browse source
This commit is contained in:
Daylin Morgan 2023-05-27 16:42:56 -05:00
parent 5308f69b4e
commit 57f025b735
Signed by: daylin
GPG key ID: C1E52E7DD81DF79F
1 changed files with 29 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
src/viv/viv.py
View file

@ -678,14 +678,14 @@ def generate_import(
) )
return return
if include_path == "absolute": if include_path == "abs":
sys.stdout.write( sys.stdout.write(
SYS_PATH_TEMPLATE.format( SYS_PATH_TEMPLATE.format(
path_to_viv=Path(__file__).resolve().absolute().parent.parent path_to_viv=Path(__file__).resolve().absolute().parent.parent
) )
+ "\n" + "\n"
) )
elif include_path == "relative": elif include_path == "rel":
sys.stdout.write( sys.stdout.write(
REL_SYS_PATH_TEMPLATE.format( REL_SYS_PATH_TEMPLATE.format(
path_to_viv=str( path_to_viv=str(
@ -1120,6 +1120,7 @@ class Viv:
), ),
): ):
self._install_local_src(sha256, args.src, args.cli) self._install_local_src(sha256, args.src, args.cli)
def shim(self, args): def shim(self, args):
"""generate viv-powered cli apps""" """generate viv-powered cli apps"""
echo("not implemented.") echo("not implemented.")
@ -1202,36 +1203,38 @@ 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, p_freeze_shim_shared = ArgumentParser(add_help=False)
"freeze",
) p_freeze_shim_shared.add_argument(
p_freeze.add_argument(
"-p", "-p",
"--path", "--path",
help="generate line to add viv to sys.path", help="generate line to add viv to sys.path",
choices=["absolute", "relative"], choices=["abs", "rel"],
) )
p_freeze.add_argument( p_freeze_shim_shared.add_argument(
"-r", "-r",
"--requirements", "--requirements",
help="path to requirements.txt file", help="path/to/requirements.txt file",
metavar="<path>", metavar="<path>",
) )
p_freeze.add_argument( p_freeze_shim_shared.add_argument(
"-k", "-k",
"--keep", "--keep",
help="preserve environment", help="preserve environment",
action="store_true", action="store_true",
) )
p_freeze.add_argument( p_freeze_shim_shared.add_argument(
"-s", "-s",
"--standalone", "--standalone",
help="generate standalone activation function", help="generate standalone activation function",
action="store_true", action="store_true",
) )
p_freeze.add_argument("reqs", help="requirements specifiers", nargs="*") p_freeze_shim_shared.add_argument(
"reqs", help="requirements specifiers", nargs="*"
)
self._get_subcmd_parser(subparsers, "freeze", parents=[p_freeze_shim_shared])
self._get_subcmd_parser( self._get_subcmd_parser(
subparsers, subparsers,
"info", "info",
@ -1284,11 +1287,20 @@ class Viv:
"-p", "--pythonpath", help="show the path/to/install", action="store_true" "-p", "--pythonpath", help="show the path/to/install", action="store_true"
) )
self._get_subcmd_parser( (
subparsers, p_manage_shim := self._get_subcmd_parser(
"shim", subparsers, "shim", parents=[p_freeze_shim_shared]
)
).set_defaults(func=self.shim, cmd="shim") ).set_defaults(func=self.shim, cmd="shim")
p_manage_shim.add_argument(
"-o",
"--output",
help="path/to/output file",
metavar="<path>",
)
p_manage_shim.add_argument(
"-b", "--bin", help="console_script/script to invoke"
)
args = parser.parse_args() args = parser.parse_args()
args.func(args) args.func(args)