diff --git a/pytest_abra/cli.py b/pytest_abra/cli.py index 0f21f30..dfcf90a 100644 --- a/pytest_abra/cli.py +++ b/pytest_abra/cli.py @@ -2,6 +2,7 @@ import argparse import os from pathlib import Path +import pkg_resources # type: ignore from loguru import logger from pytest_abra import Coordinator @@ -9,8 +10,13 @@ from pytest_abra.dir_manager import DirManager from pytest_abra.utils import get_session_id +def get_version(): + return pkg_resources.get_distribution("pytest_abra").version + + def run(): parser = argparse.ArgumentParser() + parser.add_argument("--version", "-V", action="version", version=get_version(), help="output the version number") parser.add_argument("--env_paths", type=str, help="List of loaded env files separated with ;", required=True) parser.add_argument("--recipes_dir", type=Path, help="List of loaded env files separated with ;", required=True) parser.add_argument("--output_dir", type=Path, help="List of loaded env files separated with ;", required=True)