python: Introduce __version__

labapart · Mar 26, 2024 · a288685 · a288685
1 parent 6cea2d3
commit a288685
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/gattlib-py/gattlib/__init__.py b/gattlib-py/gattlib/__init__.py
@@ -8,6 +8,12 @@
 import logging
 import pathlib
 
+try:
+    # '_version.py' is generated by 'setup.py'
+    from ._version import __version__
+except:
+    pass
+
 logger = logging.getLogger(__name__)
 
 try:

diff --git a/gattlib-py/setup.py b/gattlib-py/setup.py
@@ -25,6 +25,13 @@
 stdout, stderr = git_version_command.communicate()
 git_version = stdout.decode('utf-8').strip()
 
+#
+# Create '_version.py'
+#
+package_version = os.environ.get('GATTLIB_PY_VERSION', git_version)
+with open(os.path.join("gattlib", "_version.py"), "w") as f:
+    f.write(f"__version__ = \"{package_version}\"\n")
+
 
 class CMakeExtension(Extension):
     """Custom extension class that allows to specify the root folder of the CMake project."""
@@ -149,7 +156,7 @@ def build_extension(self, ext: CMakeExtension) -> None:
 
 setup(
     name='gattlib-py',
-    version=os.environ.get('GATTLIB_PY_VERSION', git_version),
+    version=package_version,
     author="Olivier Martin",
     author_email="olivier@labapart.com",
     description="Python wrapper for gattlib library",