version number automated for python build

update_version.py

@@ -6,6 +6,10 @@ Script to update VERSION file with semantic versioning if provided as an argumen
 
 import sys
 import re
+import os
+import toml
+import yaml
+from jinja2 import Template, Undefined
 
 def get_version():
 
@@ -28,9 +32,25 @@ def write_version_to_file(version):
         version_file.write(version)
     print(f"Version {version} written to VERSION file.")
 
+def define_environment_variable(version):
+    os.environ["VERSION"] = version
+
+def update_pyproject_toml_file(version):
+    pyproject = toml.load("pyproject.toml")
+    pyproject["project"]["version"] = version
+    toml.dump(pyproject, open("pyproject.toml", "w")) #write back
+    print(f"Version in pyproject.toml set to {version}")
+
+class NullUndefined(Undefined):
+  def __getattr__(self, key):
+    return ''
+
+
 
 # Main script
 if __name__ == "__main__":
 
     version = get_version()
-    write_version_to_file(version)
+    write_version_to_file(version)
+    define_environment_variable(version)
+    update_pyproject_toml_file(version)