|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| import sys
|
| from argparse import ArgumentParser
|
|
|
| from .commands import get_commands
|
| from .commands.base import Command, CommandContext
|
|
|
|
|
| def _build_parser(commands: list[Command]) -> ArgumentParser:
|
| parser = ArgumentParser(prog="trl", allow_abbrev=False)
|
| subparsers = parser.add_subparsers(help="available commands", dest="command")
|
|
|
| for command in commands:
|
| command.register(subparsers)
|
|
|
| return parser
|
|
|
|
|
| def main(argv: list[str] | None = None) -> int:
|
| """Run the TRL CLI."""
|
| commands = get_commands()
|
| commands_by_name = {command.name: command for command in commands}
|
| parser = _build_parser(commands)
|
| argv = list(sys.argv[1:] if argv is None else argv)
|
|
|
| args, _ = parser.parse_known_args(argv)
|
| command_name = getattr(args, "command", None)
|
| if command_name is None:
|
| parser.print_help()
|
| return 0
|
|
|
| command = commands_by_name[command_name]
|
| context = CommandContext(argv=argv)
|
| return command.run(args, context)
|
|
|
|
|
| if __name__ == "__main__":
|
| raise SystemExit(main())
|
|
|
