Add --version option

muselab-d2x · Oct 30, 2024 · 9e273a9 · 9e273a9
1 parent a1cac13
commit 9e273a9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/d2x/cli/main.py b/d2x/cli/main.py
@@ -6,12 +6,18 @@
 import pdb
 from d2x.base.types import OutputFormat, OutputFormatType, CLIOptions
 from typing import Optional
+from importlib.metadata import version, PackageNotFoundError
 
 # Disable rich_click's syntax highlighting
 click.SHOW_ARGUMENTS = False
 click.SHOW_METAVARS_COLUMN = False
 click.SHOW_OPTIONS = False
 
+try:
+    VERSION = version("d2x")
+except PackageNotFoundError:
+    VERSION = "dev"
+
 
 def common_options(func):
     """Decorator to add common options to all commands."""
@@ -26,6 +32,7 @@ def common_options(func):
 
 
 @click.group(name="d2x")
+@click.version_option(version=VERSION, prog_name="d2x")
 def d2x_cli():
     """D2X CLI main command group"""
     pass

diff --git a/pyproject.toml b/pyproject.toml
@@ -23,6 +23,9 @@ pip-tools = "^7.4.1"
 [tool.poetry.scripts]
 d2x = "d2x.cli.main:d2x_cli"
 
+[project]
+name = "d2x"
+
 [project.scripts]
 d2x = "d2x.cli:d2x_cli"