feat: Enable header generation

theoremlp · May 21, 2024 · 7d7e95a · 7d7e95a
1 parent bbac989
commit 7d7e95a
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 3 deletions.
diff --git a/examples/typical/requirements.txt b/examples/typical/requirements.txt
@@ -1,3 +1,5 @@
+# This file was autogenerated by uv via the following command:
+#    bazel run @@//:generate_requirements_txt
 --index-url https://pypi.org/simple
 
 click==8.1.7 \

diff --git a/examples/typical/requirements_linux.txt b/examples/typical/requirements_linux.txt
@@ -1,3 +1,5 @@
+# This file was autogenerated by uv via the following command:
+#    bazel run @@//:generate_requirements_linux_txt
 --index-url https://pypi.org/simple
 
 click==8.1.7 \

diff --git a/uv/private/pip.bzl b/uv/private/pip.bzl
@@ -25,6 +25,7 @@ def _uv_pip_compile(ctx, template, executable):
             "{{requirements_txt}}": ctx.file.requirements_txt.short_path,
             "{{resolved_python}}": py_toolchain.py3_runtime.interpreter.short_path,
             "{{python_platform}}": _python_platform(ctx.attr.python_platform),
+            "{{label}}": str(ctx.label),
         },
     )
 

diff --git a/uv/private/pip_compile.sh b/uv/private/pip_compile.sh
@@ -7,6 +7,7 @@ PYTHON_PLATFORM="{{python_platform}}"
 RESOLVED_PYTHON="{{resolved_python}}"
 REQUIREMENTS_IN="{{requirements_in}}"
 REQUIREMENTS_TXT="{{requirements_txt}}"
+LABEL="{{label}}"
 
 RESOLVED_PYTHON_BIN="$(dirname "$RESOLVED_PYTHON")"
 
@@ -19,10 +20,10 @@ PYTHON_VERSION="$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.
 $UV pip compile \
     --generate-hashes \
     --emit-index-url \
-    --no-header \
     --no-strip-extras \
+    --custom-compile-command "bazel run $LABEL" \
     --python-version=$PYTHON_VERSION \
     $(echo $PYTHON_PLATFORM) \
     -o $REQUIREMENTS_TXT \
     $REQUIREMENTS_IN \
-    $@
+    "$@"
diff --git a/uv/private/pip_compile_test.sh b/uv/private/pip_compile_test.sh
@@ -7,6 +7,7 @@ PYTHON_PLATFORM="{{python_platform}}"
 RESOLVED_PYTHON="{{resolved_python}}"
 REQUIREMENTS_IN="{{requirements_in}}"
 REQUIREMENTS_TXT="{{requirements_txt}}"
+LABEL="{{label}}"
 
 RESOLVED_PYTHON_BIN="$(dirname "$RESOLVED_PYTHON")"
 
@@ -24,8 +25,8 @@ $UV pip compile \
     --no-cache \
     --generate-hashes \
     --emit-index-url \
-    --no-header \
     --no-strip-extras \
+    --custom-compile-command "bazel run $LABEL" \
     --python-version=$PYTHON_VERSION \
     $(echo $PYTHON_PLATFORM) \
     -o __updated__ \