diff --git a/git_review/cmd.py b/git_review/cmd.py index 837bfa75..d3fce692 100644 --- a/git_review/cmd.py +++ b/git_review/cmd.py @@ -32,9 +32,16 @@ from urllib.parse import urlencode from urllib.parse import urljoin from urllib.parse import urlparse -import pkg_resources import requests +try: + import importlib.metadata as importlib_metadata + pkg_resources = None +except ImportError: + # Pre-py38 + importlib_metadata = None + import pkg_resources + VERBOSE = False UPDATE = False @@ -220,9 +227,12 @@ def run_http_exc(klazz, url, **env): def get_version(): - requirement = pkg_resources.Requirement.parse('git-review') - provider = pkg_resources.get_provider(requirement) - return provider.version + if importlib_metadata: + return importlib_metadata.version('git-review') + else: + requirement = pkg_resources.Requirement.parse('git-review') + provider = pkg_resources.get_provider(requirement) + return provider.version def get_git_version():