From 0174f0de6fa45015cd9cb5a3d55a00685cb893d4 Mon Sep 17 00:00:00 2001
From: benvilliere <22590566+benvilliere@users.noreply.github.com>
Date: Sun, 8 Oct 2023 22:37:48 +0100
Subject: [PATCH] Making bump version script update version in pyproject.toml
 as well

---
 scripts/bump_version.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/scripts/bump_version.py b/scripts/bump_version.py
index 97b51c8..cc83083 100644
--- a/scripts/bump_version.py
+++ b/scripts/bump_version.py
@@ -64,6 +64,30 @@ def validate_version(version):
     return bool(pattern.match(version))
 
 
+def read_version_from_toml() -> str:
+    with open('./pyproject.toml', 'r') as f:
+        lines = f.readlines()
+        for line in lines:
+            if "version =" in line:
+                return line.split('=')[1].strip().strip('"')
+    raise ValueError("Version not found in pyproject.toml")
+
+
+def write_version_to_toml(new_version: str, dry_run=False):
+    with open('./pyproject.toml', 'r') as f:
+        contents = f.read()
+
+    new_contents = contents.replace(read_version_from_toml(), new_version)
+
+    if not dry_run:
+        with open('./pyproject.toml', 'w') as f:
+            f.write(new_contents)
+    else:
+        print("--Dry Run for pyproject.toml--")
+        print("File would have been updated to:")
+        print(new_contents)
+
+
 if __name__ == '__main__':
     try:
         parser = argparse.ArgumentParser(description="Bump version for Slimer project.")
@@ -85,6 +109,7 @@ def validate_version(version):
             new_version = bump_version(major=args.major, minor=args.minor, patch=args.patch)
 
         write_version(new_version, dry_run=args.dry_run)
+        write_version_to_toml(new_version, dry_run=args.dry_run)
 
         if args.dry_run:
             print(f"Version would be set to: {new_version}")