diff --git a/src/pdh/main.py b/src/pdh/main.py index 9ddc417..ef0b217 100644 --- a/src/pdh/main.py +++ b/src/pdh/main.py @@ -15,7 +15,7 @@ # along with this program. If not, see . # import click -import pkg_resources +import importlib.metadata import sys import re import os @@ -56,7 +56,7 @@ def config(config): @main.command(help="Print cloud tools version and exit") def version(): - click.echo(f"v{pkg_resources.get_distribution('pdh').version}") + click.echo(f"v{importlib.metadata.version('pdh')}") @main.group(help="Operater on Users") diff --git a/tests/test_main.py b/tests/test_main.py index 049f862..7d555da 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -18,7 +18,7 @@ from rich.console import Console from pdh import main from click import testing -import pkg_resources +import importlib.metadata def test_assert_true(): @@ -34,7 +34,7 @@ def test_main_version(): runner = testing.CliRunner() result = runner.invoke(main.main, "version") assert result.exit_code == 0 - assert result.stdout == f"v{pkg_resources.get_distribution('pdh').version}\n" + assert result.stdout == f"v{importlib.metadata.version('pdh')}\n" def test_print_items_table():