ishangsf/DeepCodeGeniusWeb-vscode
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #593 from devchat-ai/config_conda_forge

Browse Source 
feat: Add configurable conda-forge URL for Python installation
This commit is contained in:
Tim 2024-09-20 16:56:35 +08:00 committed by GitHub
commit 4e5283556c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 78 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

141
src/util/python_installer/python_install.ts
View File

@ -2,6 +2,7 @@ import { exec, spawn } from 'child_process';
import * as path from 'path';
import * as os from 'os';
import { logger } from '../logger';
import { DevChatConfig } from '../config';
const fs = require('fs');
// Check if the environment already exists
@ -96,69 +97,83 @@ function canCreateSubdirectory(dirPath: string): boolean {
export async function installPythonMicromamba(mambaCommandPath: string, envName: string, pythonVersion: string): Promise<string> {
// Set the installation directory for conda
// Set the installation directory for conda
let userHome = process.platform === 'win32' ? fs.realpathSync(process.env.USERPROFILE || '') : process.env.HOME;
if (os.platform() === 'win32' && /[^\x00-\xFF]/.test(userHome)) {
if (fs.existsSync('C:/Program Files') && canCreateSubdirectory('C:/Program Files')) {
userHome = 'C:/Program Files';
} else if (fs.existsSync('D:/Program Files') && canCreateSubdirectory('D:/Program Files')) {
userHome = 'D:/Program Files';
} else if (fs.existsSync('E:/Program Files') && canCreateSubdirectory('E:/Program Files')) {
userHome = 'E:/Program Files';
}
}
if (os.platform() === 'win32' && /[^\x00-\xFF]/.test(userHome)) {
if (fs.existsSync('C:/Program Files') && canCreateSubdirectory('C:/Program Files')) {
userHome = 'C:/Program Files';
} else if (fs.existsSync('D:/Program Files') && canCreateSubdirectory('D:/Program Files')) {
userHome = 'D:/Program Files';
} else if (fs.existsSync('E:/Program Files') && canCreateSubdirectory('E:/Program Files')) {
userHome = 'E:/Program Files';
}
}
const pathToMamba = `${userHome}/.chat/mamba`;
const envPath = path.resolve(pathToMamba, 'envs', envName);
let pythonPath;
let pythonPath2;
if (os.platform() === 'win32') {
pythonPath = path.join(envPath, 'Scripts', 'python.exe');
pythonPath2 = path.join(envPath, 'python.exe');
} else {
pythonPath = path.join(envPath, 'bin', 'python');
}
if (fs.existsSync(pythonPath)) {
return pythonPath;
} else if (pythonPath2 && fs.existsSync(pythonPath2)) {
return pythonPath2;
}
return new Promise<string>((resolve, reject) => {
const cmd = mambaCommandPath;
const args = ['create', '-n', envName, '-c', 'conda-forge', '-r', pathToMamba, `python=${pythonVersion}`, '--yes'];
// output command and args in line
// args to "create -n xx -c conda-forge ..."
logger.channel()?.info(`cmd: ${cmd} ${args.join(' ')}`);
const child = spawn(cmd, args);
child.stdout.on('data', (data) => {
logger.channel()?.info(`${data}`);
});
child.stderr.on('data', (data) => {
console.error(`stderr: ${data}`);
});
child.on('error', (error) => {
logger.channel()?.error(`Error installing python ${pythonVersion} in env ${envName}`);
logger.channel()?.show();
reject('');
});
child.on('close', (code) => {
if (code !== 0) {
reject(new Error(`Command exited with code ${code}`));
} else {
if (fs.existsSync(pythonPath)) {
resolve(pythonPath);
} else if (pythonPath2 && fs.existsSync(pythonPath2)) {
resolve(pythonPath2);
} else {
reject(new Error(`No Python found`));
}
}
});
});
}
const envPath = path.resolve(pathToMamba, 'envs', envName);
let pythonPath;
let pythonPath2;
if (os.platform() === 'win32') {
pythonPath = path.join(envPath, 'Scripts', 'python.exe');
pythonPath2 = path.join(envPath, 'python.exe');
} else {
pythonPath = path.join(envPath, 'bin', 'python');
}
if (fs.existsSync(pythonPath)) {
return pythonPath;
} else if (pythonPath2 && fs.existsSync(pythonPath2)) {
return pythonPath2;
}
// Get conda-forge URL from config file
const condaForgeUrl = getCondaForgeUrl();
return new Promise<string>((resolve, reject) => {
const cmd = mambaCommandPath;
const args = ['create', '-n', envName, '-c', condaForgeUrl, '-r', pathToMamba, `python=${pythonVersion}`, '--yes'];
logger.channel()?.info(`cmd: ${cmd} ${args.join(' ')}`);
const child = spawn(cmd, args);
child.stdout.on('data', (data) => {
logger.channel()?.info(`${data}`);
});
child.stderr.on('data', (data) => {
console.error(`stderr: ${data}`);
});
child.on('error', (error) => {
logger.channel()?.error(`Error installing python ${pythonVersion} in env ${envName}`);
logger.channel()?.show();
reject('');
});
child.on('close', (code) => {
if (code !== 0) {
reject(new Error(`Command exited with code ${code}`));
} else {
if (fs.existsSync(pythonPath)) {
resolve(pythonPath);
} else if (pythonPath2 && fs.existsSync(pythonPath2)) {
resolve(pythonPath2);
} else {
reject(new Error(`No Python found`));
}
}
});
});
}
export function getCondaForgeUrl(): string {
const defaultUrl = "https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/";
try {
const config = DevChatConfig.getInstance();
const url = config.get("conda-forge-url", defaultUrl);
return url || defaultUrl; // 如果 url 是 undefined 或空字符串，返回默认 URL
} catch (error) {
logger.channel()?.error(`Error reading conda-forge URL from config: ${error}`);
logger.channel()?.show();
return defaultUrl;
}
}