1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
"""The cli implementation for hatch_jupyter_builder."""
import argparse
import sys
from typing import Optional
from .compare_migrated import cli as compare_cli
from .migrate import cli as migrate_cli
def make_parser(
parser: Optional[argparse.ArgumentParser] = None, prog: Optional[str] = None
) -> argparse.ArgumentParser:
"""Make an arg parser."""
if parser is None:
parser = argparse.ArgumentParser(prog=prog)
parsers = parser.add_subparsers()
migrate_parser = parsers.add_parser("migrate")
migrate_cli.make_parser(migrate_parser)
migrate_parser.set_defaults(func=migrate_cli.run)
compare_parser = parsers.add_parser("compare-migrated")
compare_cli.make_parser(compare_parser)
compare_parser.set_defaults(func=compare_cli.run)
return parser
def run(args: Optional[argparse.Namespace] = None) -> None:
"""Run the main script."""
if args is None:
prog = (
f"{sys.executable} -m hatch_jupyter_builder"
if sys.argv[0].endswith("__main__.py")
else None
)
parser = make_parser(prog=prog)
args = parser.parse_args()
args.func()
|
