Add input python-version

astral-sh · Nov 28, 2024 · f5fe5c1 · f5fe5c1
1 parent 9839fa9
commit f5fe5c1
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 0 deletions.
diff --git a/dist/setup/index.js b/dist/setup/index.js
diff --git a/src/download/download-uv-python.ts b/src/download/download-uv-python.ts
@@ -0,0 +1,9 @@
+import * as exec from "@actions/exec";
+
+export async function downloadUvPython(pythonVersion: string) {
+  const options: exec.ExecOptions = {
+    silent: false,
+  };
+  const execArgs = ["python", "install", pythonVersion];
+  await exec.exec("uv", execArgs, options);
+}
diff --git a/src/setup-uv.ts b/src/setup-uv.ts
@@ -22,6 +22,7 @@ import {
   toolDir,
   version,
 } from "./utils/inputs";
+import { downloadUvPython } from "./download/download-uv-python";
 
 async function run(): Promise<void> {
   const platform = getPlatform();
@@ -45,6 +46,7 @@ async function run(): Promise<void> {
     addUvToPath(setupResult.uvDir);
     addToolBinToPath();
     setToolDir();
+    setupPython();
     core.setOutput("uv-version", setupResult.version);
     core.info(`Successfully installed uv version ${setupResult.version}`);
 
@@ -133,6 +135,12 @@ function setToolDir(): void {
   }
 }
 
+async function setupPython(): Promise<void> {
+  await downloadUvPython("3.12");
+  core.exportVariable("UV_PYTHON", "3.12");
+  core.info("Set UV_PYTHON to 3.12");
+}
+
 function setCacheDir(cacheLocalPath: string): void {
   core.exportVariable("UV_CACHE_DIR", cacheLocalPath);
   core.info(`Set UV_CACHE_DIR to ${cacheLocalPath}`);