added --version command for fedn package

scaleoutsystems · Jul 3, 2024 · ba333c3 · ba333c3
1 parent 2485fb4
commit ba333c3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/fedn/cli/main.py b/fedn/cli/main.py
@@ -1,12 +1,17 @@
 import click
-
+import importlib.metadata
 CONTEXT_SETTINGS = dict(
     # Support -h as a shortcut for --help
     help_option_names=["-h", "--help"],
 )
 
-
+# Dynamically get the version of the package
+try:
+    version = importlib.metadata.version('fedn')
+except importlib.metadata.PackageNotFoundError:
+    version = 'unknown'
 @click.group(context_settings=CONTEXT_SETTINGS)
+@click.version_option(version)
 @click.pass_context
 def main(ctx):
     """:param ctx: