version.py 588 B

123456789101112131415161718192021222324
  1. import argparse
  2. from gpustack import __version__, __git_commit__
  3. def setup_version_cmd(subparsers: argparse._SubParsersAction):
  4. parser: argparse.ArgumentParser = subparsers.add_parser(
  5. "version",
  6. help="Print version.",
  7. description="Print version.",
  8. )
  9. parser.add_argument(
  10. "--short",
  11. action="store_true",
  12. help="Print without commit hash.",
  13. default=False,
  14. )
  15. parser.set_defaults(func=run)
  16. def run(args):
  17. if args.short:
  18. print(__version__)
  19. else:
  20. print(f"{__version__} ({__git_commit__})")