Make version-discovery more robust

Uninett · Aug 12, 2021 · 259a55e · 259a55e
1 parent fef4304
commit 259a55e
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/argus/site/views.py b/src/argus/site/views.py
@@ -104,11 +104,21 @@ class MetadataView(APIView):
  )
  def get(self, request, format=None):
  try:
- from argus.version import __version__
- except (ModuleNotFoundError, ImportError):
- import pkg_resources
+ # Run from source-dir
+ import argus.version
 
- __version__ = pkg_resources.get_distribution("argus-server").version
+ __version__ = argus.version.version
+ except (AttributeError, ModuleNotFoundError, ImportError):
+ # Installed from package
+ try:
+ from importlib.metadata import version
+
+ __version__ = version("wheel")
+ except ImportError:
+ # Python < 3.8
+ import pkg_resources
+
+ __version__ = pkg_resources.get_distribution("argus-server").version
 
  metadata = {
  "server-version": __version__,