chore: update the release script

Author: leodevian

scripts/release.py

@@ -34,9 +34,13 @@
 def create_parser() -> argparse.ArgumentParser:
     """Create the argument parser."""
     parser = argparse.ArgumentParser(description="make a release")
-    parser.add_argument(
+    mutex = parser.add_mutually_exclusive_group(required=True)
+    mutex.add_argument(
+        "--version",
+        help="set the version",
+    )
+    mutex.add_argument(
         "--bump",
-        required=True,
         help="update the version using the given semantics",
     )
     parser.add_argument(
@@ -57,20 +61,18 @@ def check_repository() -> None:
         raise SystemExit(1) from None
 
 
-def bump_version(bump: str) -> str:
-    """Bump the version and return the new version."""
+def update_version(version: str | None = None, bump: str | None = None) -> str:
+    """Update the version and return the new version."""
+    args = ["uv", "version", "--no-sync", "--output-format", "json"]
+
+    if version:
+        args.append(version)
+
+    if bump:
+        args.extend(("--bump", bump))
+
     try:
-        version_json = subprocess.check_output(
-            (
-                "uv",
-                "version",
-                "--no-sync",
-                "--output-format",
-                "json",
-                "--bump",
-                bump,
-            )
-        )
+        version_json = subprocess.check_output(args)
 
     except subprocess.CalledProcessError:
         print_error("An error occurred while bumping the version.")
@@ -80,17 +82,16 @@ def bump_version(bump: str) -> str:
     return version["version"]
 
 
-@contextmanager
-def switch_to_branch(branch: str) -> Generator[None]:
-    """Create a new branch and switch to it.
-
-    It is removed on exit.
-    """
-    base_branch = subprocess.check_output(
+def get_branch() -> str:
+    """Get the current branch."""
+    return subprocess.check_output(
         ("git", "rev-parse", "--abbrev-ref", "HEAD"),
         text=True,
     ).rstrip()
 
+
+def create_branch(branch: str) -> None:
+    """Create a new branch."""
     try:
         run(("git", "branch", branch))
 
@@ -100,6 +101,16 @@ def switch_to_branch(branch: str) -> Generator[None]:
 
     print(f"Created new branch {branch!r}.")
 
+
+@contextmanager
+def switch_to_branch(branch: str) -> Generator[None]:
+    """Create a new branch and switch to it.
+
+    It is removed on exit.
+    """
+    base_branch = get_branch()
+    create_branch(branch)
+
     try:
         run(("git", "checkout", branch))
         print(f"Switched from branch {base_branch!r} to branch {branch!r}.")
@@ -150,7 +161,7 @@ def main(argv: Sequence[str] | None = None) -> None:
 
     check_repository()
 
-    version = bump_version(args.bump)
+    version = update_version(args.version, args.bump)
 
     release_branch = f"release/{version}"
 
@@ -182,4 +193,4 @@ def main(argv: Sequence[str] | None = None) -> None:
 
 
 if __name__ == "__main__":
-    raise SystemExit(main())
+    main()