diff --git a/setup.py b/setup.py index 5dd5ee3..6dcee1b 100644 --- a/setup.py +++ b/setup.py @@ -18,6 +18,7 @@ "Flask~=2.0", "gunicorn", "redis", + "importlib_metadata; python_version<'3.8'", ], description="A simple TAK server and COT router", long_description=long_description, diff --git a/taky/__init__.py b/taky/__init__.py index b347ca8..dc5e796 100644 --- a/taky/__init__.py +++ b/taky/__init__.py @@ -1,8 +1,12 @@ -from pkg_resources import get_distribution, DistributionNotFound +try: + from importlib.metadata import version, PackageNotFoundError +except ImportError: + # Python < 3.8 + from importlib_metadata import version, PackageNotFoundError from . import cot try: - __version__ = get_distribution(__name__).version -except DistributionNotFound: + __version__ = version(__name__) +except PackageNotFoundError: __version__ = "unknown"