actions/setup-uv
1
0
Fork 0
mirror of https://github.com/astral-sh/setup-uv.git synced 2025-04-05 07:05:18 -04:00
Code Issues Projects Releases Packages Wiki Activity

Simplify setup-python

Browse source
This commit is contained in:
Julfried 2025-02-17 11:53:54 +01:00
parent 1d0b806c47
commit 5770883575
1 changed files with 62 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

89
src/setup-uv.ts
View file

@ -146,50 +146,85 @@ function setToolDir(): void {
}
async function setupPython(): Promise<void> {
let pythonVersionToUse = undefined; // Python version to use for setup
const options: exec.ExecOptions = {
silent: !core.isDebug(),
};
// Case (1): No Python version and no pyproject.toml file
if (pythonVersion === "" && pyProjectFile === "") {
core.info("No Python setup required.");
return;
}
// Case (2): Python version is provided and no pyproject.toml file
else if (pythonVersion !== "" && pyProjectFile === "") {
pythonVersionToUse = pythonVersion;
const execArgs = ["pin", "python", pythonVersion];
core.info(`Pinning Python version to ${pythonVersion}...`);
await exec.exec("uv", execArgs, options);
}
// Case (3): No Python version and pyproject.toml file
else if (pythonVersion === "" && pyProjectFile !== "") {
pythonVersionToUse = getPythonVersionFromPyProject(pyProjectFile);
if (!pythonVersionToUse) {
core.info(`No valid Python version found in ${pyProjectFile}. Not able to setup Python.`);
const extractedPythonVersion = getPythonVersionFromPyProject(pyProjectFile);
if (!extractedPythonVersion){
core.warning(
`Could not find python version in pyproject.toml. Won't setup python.`,
);
return;
}
}
// Case (4): Python version is provided and pyproject.toml file
else if (pythonVersion !== "" && pyProjectFile !== "") {
//TODO: Implement this case
}
// If a valid Python version is found, setup Python
if (pythonVersionToUse) {
core.exportVariable("UV_PYTHON", pythonVersionToUse);
core.info(`Setting UV_PYTHON to ${pythonVersionToUse}`);
const options: exec.ExecOptions = {
silent: !core.isDebug(),
};
const execArgs = ["venv", "--python", pythonVersionToUse];
core.info("Activating Python venv...");
const execArgs = ["pin", "python", extractedPythonVersion, "--project", pyProjectFile];
core.info(`Pinning Python version to ${extractedPythonVersion}...`);
await exec.exec("uv", execArgs, options);
}
let venvBinPath = ".venv/bin";
if (process.platform === "win32") {
venvBinPath = ".venv/Scripts";
}
core.addPath(path.resolve(venvBinPath));
core.exportVariable("VIRTUAL_ENV", path.resolve(".venv"));
// Case (4): Pin python version using uv pin if python version is provided and pyproject.toml file is present
if (pythonVersion !== "" && pyProjectFile !== "") {
const execArgs = ["pin", "python", pythonVersion, "--project", pyProjectFile];
core.info(`Pinning Python version to ${pythonVersion}...`);
await exec.exec("uv", execArgs, options);
}
// Extract the pinned python version
let pinnedPythonVersion = getPinnedPythonVersion();
if (!pinnedPythonVersion) {
core.setFailed("Failed to determine pinned Python version after uv pin.");
return;
}
// Setup and activate venv
// Set UV_PYHTON to the pinned python version
core.exportVariable("UV_PYTHON", pinnedPythonVersion);
core.info(`Setting UV_PYTHON to ${pinnedPythonVersion}`);
core.info("Activating Python venv...");
const execArgs = ["venv"];
await exec.exec("uv", execArgs, options);
let venvBinPath = ".venv/bin";
if (process.platform === "win32") {
venvBinPath = ".venv/Scripts";
}
core.addPath(path.resolve(venvBinPath));
core.exportVariable("VIRTUAL_ENV", path.resolve(".venv"));
}
function getPinnedPythonVersion(): string | undefined {
const pythonVersionFile = ".python-version";
if (!fs.existsSync(pythonVersionFile)) {
core.warning(`No .python-version file found after uv pin.`);
return undefined;
}
try {
const version = fs.readFileSync(pythonVersionFile, "utf-8").trim();
core.info(`Detected pinned Python version from .python-version: ${version}`);
return version;
} catch (error) {
core.warning(`Failed to read .python-version: ${error}`);
return undefined;
}
}