Remove usage of deprecated pkg_resources

mbovo · Jun 4, 2024 · bcafeb8 · bcafeb8
1 parent 7317d5d
commit bcafeb8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/pdh/main.py b/src/pdh/main.py
@@ -15,7 +15,7 @@
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
 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
@@ -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():