From 33aa0c06392c3389b81f3dad435927f4b553cb35 Mon Sep 17 00:00:00 2001
From: Bruno Alla <alla.brunoo@gmail.com>
Date: Sat, 11 Jan 2025 11:33:30 +0000
Subject: [PATCH] Add script to automatically keep ruff versions up-to-date

---
 scripts/ruff_version.py | 53 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 scripts/ruff_version.py

diff --git a/scripts/ruff_version.py b/scripts/ruff_version.py
new file mode 100644
index 00000000..acc6beb1
--- /dev/null
+++ b/scripts/ruff_version.py
@@ -0,0 +1,53 @@
+import subprocess
+import tomllib
+from pathlib import Path
+
+ROOT = Path(__file__).parent.parent
+TEMPLATED_ROOT = ROOT / "{{cookiecutter.project_slug}}"
+REQUIREMENTS_LOCAL_TXT = TEMPLATED_ROOT / "requirements" / "local.txt"
+PRE_COMMIT_CONFIG = TEMPLATED_ROOT / ".pre-commit-config.yaml"
+PYPROJECT_TOML = ROOT / "pyproject.toml"
+
+
+def main():
+    new_version = get_requirements_txt_version()
+    old_version = get_pyproject_toml_version()
+    if old_version == new_version:
+        return
+
+    update_ruff_version(old_version, new_version)
+    subprocess.run(["uv", "lock", "--no-upgrade"], cwd=ROOT)
+
+
+def get_requirements_txt_version():
+    content = REQUIREMENTS_LOCAL_TXT.read_text()
+    for line in content.split("\n"):
+        if line.startswith("ruff"):
+            return line.split(" ")[0].split("==")[1]
+    return None
+
+
+def get_pyproject_toml_version():
+    data = tomllib.loads(PYPROJECT_TOML.read_text())
+    for dependency in data["project"]["dependencies"]:
+        if dependency.startswith("ruff=="):
+            return dependency.split("==")[1]
+
+
+def update_ruff_version(old_version, new_version):
+    # Update pyproject.toml
+    new_content = PYPROJECT_TOML.read_text().replace(
+        f"ruff=={old_version}",
+        f"ruff=={new_version}",
+    )
+    PYPROJECT_TOML.write_text(new_content)
+    # Update pre-commit config
+    new_content = PRE_COMMIT_CONFIG.read_text().replace(
+        f"repo: https://github.com/astral-sh/ruff-pre-commit\n    rev: v{old_version}",
+        f"repo: https://github.com/astral-sh/ruff-pre-commit\n    rev: v{new_version}",
+    )
+    PRE_COMMIT_CONFIG.write_text(new_content)
+
+
+if __name__ == "__main__":
+    main()