diff --git a/src/distro/distro.py b/src/distro/distro.py index 0dd55f2..90ceb87 100755 --- a/src/distro/distro.py +++ b/src/distro/distro.py @@ -1455,6 +1455,9 @@ def main() -> None: logger.addHandler(logging.StreamHandler(sys.stdout)) parser = argparse.ArgumentParser(description="OS distro info tool") + parser.add_argument( + "--version", "-v", action="version", version=f"%(prog)s {__version__}" + ) parser.add_argument( "--json", "-j", help="Output in machine readable format", action="store_true" ) @@ -1470,6 +1473,10 @@ def main() -> None: args = parser.parse_args() if args.root_dir: + if not os.path.isdir(args.root_dir): + parser.error( + f"Root directory '{args.root_dir}' does not exist or is not a directory" + ) dist = LinuxDistribution( include_lsb=False, include_uname=False,