Add version script

Author: davidecaroselli

src/lara_sdk/__init__.py

@@ -3,4 +3,6 @@
 from ._errors import LaraApiError, LaraError
 from ._translator import Memory, MemoryImport, TextBlock, TextResult, Memories, Translator, TranslatePriority, UseCache
 
+# This constant is auto-generated by the build script.
+# Manual modifications will be overwritten and may cause unexpected behavior.
 __version__ = "1.2.4"

version

@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+import os
+import subprocess
+import sys
+from typing import List
+
+
+def die(msg: str):
+    print(msg, file=sys.stderr)
+    sys.exit(1)
+
+
+def cmd_exec(cmd: List[str]) -> str:
+    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    if result.returncode != 0:
+        die(f"Error: Failed to execute command: {' '.join(cmd)}\n{result.stderr}")
+    return result.stdout.strip()
+
+
+def check_git_status():
+    stdout = cmd_exec(["git", "status", "--porcelain"])
+    if len(stdout) > 0:
+        die("Error: There are uncommitted changes. Please commit or stash them before proceeding.")
+
+    stdout = cmd_exec(["git", "tag", "--points-at", "HEAD"])
+    if len(stdout) > 0:
+        die("Error: HEAD is already tagged.")
+
+    branch = cmd_exec(["git", "branch", "--show-current"])
+    if branch != "main":
+        die("Error: You must be on the main branch to update the version.")
+
+
+def update_version(part: str) -> str:
+    init_file = os.path.join(os.path.dirname(__file__), "src", "lara_sdk", "__init__.py")
+    with open(init_file, "r") as f:
+        lines = f.readlines()
+
+    new_version = None
+
+    for i, line in enumerate(lines):
+        if line.startswith("__version__"):
+            version = line.split("=")[1].strip().strip("\"'")
+
+            major, minor, patch = version.split(".")
+            if part == "major":
+                major = str(int(major) + 1)
+                minor = "0"
+                patch = "0"
+            elif part == "minor":
+                minor = str(int(minor) + 1)
+                patch = "0"
+            elif part == "patch":
+                patch = str(int(patch) + 1)
+            else:
+                die("Error: Invalid version part.")
+
+            new_version = f"{major}.{minor}.{patch}"
+            lines[i] = f"__version__ = \"{new_version}\"\n"
+            break
+
+    if new_version is None:
+        die("Error: Could not find version in __init__.py.")
+
+    with open(init_file, "w") as f:
+        f.writelines(lines)
+
+    return new_version
+
+
+def git_tag(version: str):
+    cmd_exec(["git", "add", "."])
+    cmd_exec(["git", "commit", "-m", f"v{version}"])
+    cmd_exec(["git", "tag", f"v{version}"])
+
+
+def main():
+    args = sys.argv[1:]
+
+    if len(args) != 1 or args[0] not in ["major", "minor", "patch"]:
+        print("Usage: python3 version [major|minor|patch]")
+        sys.exit(1)
+
+    check_git_status()
+    version = update_version(args[0])
+    git_tag(version)
+
+    print(f"Tag v{version} created.")
+
+
+if __name__ == '__main__':
+    main()