diff --git a/broker/commands.py b/broker/commands.py index b4f165a6..a19ed981 100644 --- a/broker/commands.py +++ b/broker/commands.py @@ -156,12 +156,14 @@ def provider_cmd(ctx, *args, **kwargs): # the actual subcommand def cli(version): """Command-line interface for interacting with providers.""" if version: + from importlib.metadata import version + from packaging.version import Version - import pkg_resources import requests - broker_version = pkg_resources.get_distribution("broker").version - # check the latest version publish to PyPi + broker_version = version("broker") + + # Check against the latest version published to PyPi try: latest_version = Version( requests.get("https://pypi.org/pypi/broker/json", timeout=60).json()["info"][