Make version-discovery more robust

Uninett · Apr 25, 2024 · 066b3b1 · 066b3b1
1 parent 0fbeb47
commit 066b3b1
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/argus/site/views.py b/src/argus/site/views.py
@@ -109,11 +109,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__,