Add: Support for uv

n-thumann · greenbonebot · commit 329294f27786 · 2026-02-25T08:58:09.000+01:00
diff --git a/autohooks/cli/__init__.py b/autohooks/cli/__init__.py
@@ -50,9 +50,10 @@ def main():
             str(Mode.PYTHONPATH),
             str(Mode.PIPENV),
             str(Mode.POETRY),
+            str(Mode.UV)
         ],
         help="Mode for loading autohooks during hook execution. Either load "
-        "autohooks from the PYTHON_PATH, via pipenv or via poetry.",
+        "autohooks from the PYTHON_PATH, via pipenv, via poetry or via uv.",
     )
     activate_parser.set_defaults(func=install_hooks)
 
diff --git a/autohooks/config.py b/autohooks/config.py
@@ -83,10 +83,12 @@ def _gather_mode(mode_string: Optional[str]) -> Mode:
     Gather the mode from a mode string
     """
     mode = Mode.from_string(mode_string)
-    is_virtual_env = mode == Mode.PIPENV or mode == Mode.POETRY
+    is_virtual_env = mode == Mode.PIPENV or mode == Mode.POETRY or mode == Mode.UV
     if is_virtual_env and not is_split_env():
         if mode == Mode.POETRY:
             mode = Mode.POETRY_MULTILINE
+        elif mode == Mode.UV:
+            mode = Mode.UV_MULTILINE
         else:
             mode = Mode.PIPENV_MULTILINE
     return mode
diff --git a/autohooks/hooks.py b/autohooks/hooks.py
@@ -14,6 +14,8 @@
     POETRY_MULTILINE_SHEBANG,
     POETRY_SHEBANG,
     PYTHON3_SHEBANG,
+    UV_SHEBANG,
+    UV_MULTILINE_SHEBANG,
     TEMPLATE_VERSION,
     PreCommitTemplate,
 )
@@ -65,6 +67,8 @@ def read_mode(self) -> Mode:
             return Mode.POETRY
         if shebang == PIPENV_SHEBANG:
             return Mode.PIPENV
+        if shebang == UV_SHEBANG:
+            return Mode.UV
 
         shebang = f"{lines[0][2:]}\n"
         shebang += "\n".join(lines[1:5])
@@ -74,6 +78,9 @@ def read_mode(self) -> Mode:
         if shebang == PIPENV_MULTILINE_SHEBANG:
             return Mode.PIPENV_MULTILINE
 
+        if shebang == UV_MULTILINE_SHEBANG:
+            return Mode.UV_MULTILINE
+
         return Mode.UNKNOWN
 
     def read_version(self) -> int:
diff --git a/autohooks/settings.py b/autohooks/settings.py
@@ -15,8 +15,10 @@ class Mode(Enum):
     PIPENV = 1
     PYTHONPATH = 2
     POETRY = 3
-    PIPENV_MULTILINE = 4
-    POETRY_MULTILINE = 5
+    UV = 4
+    PIPENV_MULTILINE = 5
+    POETRY_MULTILINE = 6
+    UV_MULTILINE = 7
     UNDEFINED = -1
     UNKNOWN = -2
 
@@ -29,6 +31,10 @@ def get_effective_mode(self):
             return Mode.POETRY
         if self.value == Mode.POETRY_MULTILINE.value:
             return Mode.POETRY_MULTILINE
+        if self.value == Mode.UV.value:
+            return Mode.UV
+        if self.value == Mode.UV_MULTILINE.value:
+            return Mode.UV_MULTILINE
         return Mode.PYTHONPATH
 
     @staticmethod
diff --git a/autohooks/template.py b/autohooks/template.py
@@ -13,6 +13,7 @@
 PYTHON3_SHEBANG = "/usr/bin/env python3"
 PIPENV_SHEBANG = "/usr/bin/env -S pipenv run python3"
 POETRY_SHEBANG = "/usr/bin/env -S poetry run python"
+UV_SHEBANG = "/usr/bin/env -S uv run python"
 # For OS's that don't support '/usr/bin/env -S'.
 PIPENV_MULTILINE_SHEBANG = (
     "/bin/sh\n"
@@ -28,6 +29,13 @@
     'exit "$?"\n'
     "'''"
 )
+UV_MULTILINE_SHEBANG = (
+    "/bin/sh\n"
+    "\"true\" ''':'\n"
+    'uv run python "$0" "$@"\n'
+    'exit "$?"\n'
+    "'''"
+)
 
 TEMPLATE_VERSION = 1
 
@@ -55,10 +63,14 @@ def render(self, *, mode: Mode) -> str:
             params["SHEBANG"] = PIPENV_SHEBANG
         elif mode == Mode.POETRY:
             params["SHEBANG"] = POETRY_SHEBANG
+        elif mode == Mode.UV:
+            params["SHEBANG"] = UV_SHEBANG
         elif mode == Mode.PIPENV_MULTILINE:
             params["SHEBANG"] = PIPENV_MULTILINE_SHEBANG
         elif mode == Mode.POETRY_MULTILINE:
             params["SHEBANG"] = POETRY_MULTILINE_SHEBANG
+        elif mode == Mode.UV_MULTILINE:
+            params["SHEBANG"] = UV_MULTILINE_SHEBANG
         else:
             params["SHEBANG"] = PYTHON3_SHEBANG
 
