VSCode扩展中child_process调用Python提示graph.py未找到求助

问题描述 #

我使用以下Node.js代码调用Python脚本，该代码作为独立文件运行正常，但放入VSCode扩展的extension.js中时，提示“graph.py not found”。

const {spawn} = require('child_process')

const child = spawn('python3', ['graph.py'])

child.stdout.on('data', (data)=>{
    console.log(`stdout: ${data}`)
})

child.stderr.on('data', (data)=>{
    console.log(`stderr: ${data}`)
})

child.on('error', (error)=> console.log(`error: ${error.message}`))

child.on('exit', (code, signal)=>{
    if(code) console.log(`Process exit with code: ${code}`)
    if(signal) console.log(`Process killed with signal: ${signal}`)
    console.log(`DONE!!`)
})

原因分析 #

VSCode扩展运行时的**当前工作目录（cwd）**并非你的扩展代码所在目录，而是VSCode的启动目录（比如用户家目录或当前打开的工作区目录）。直接使用相对路径graph.py会导致系统在错误的目录下查找文件。

解决方案 #

通过获取扩展的绝对路径，确保Python脚本的路径正确，有两种实现方式：

方式1：直接使用脚本的绝对路径 #

借助VSCode扩展API提供的context.extensionPath获取扩展根目录的绝对路径，再用path模块拼接出graph.py的完整路径：

const { spawn } = require('child_process');
const path = require('path');

function activate(context) {
    // 拼接Python脚本的绝对路径
    const scriptAbsolutePath = path.join(context.extensionPath, 'graph.py');
    const child = spawn('python3', [scriptAbsolutePath]);

    child.stdout.on('data', (data) => {
        console.log(`stdout: ${data}`);
    });

    child.stderr.on('data', (data) => {
        console.log(`stderr: ${data}`);
    });

    child.on('error', (error) => console.log(`error: ${error.message}`));

    child.on('exit', (code, signal) => {
        if (code) console.log(`Process exit with code: ${code}`);
        if (signal) console.log(`Process killed with signal: ${signal}`);
        console.log(`DONE!!`);
    });
}

exports.activate = activate;

方式2：设置子进程的工作目录 #

通过spawn的cwd选项，将子进程的工作目录指定为扩展根目录，这样相对路径graph.py就能被正确识别：

const { spawn } = require('child_process');

function activate(context) {
    const child = spawn('python3', ['graph.py'], {
        cwd: context.extensionPath // 把工作目录设为扩展根目录
    });

    child.stdout.on('data', (data) => {
        console.log(`stdout: ${data}`);
    });

    child.stderr.on('data', (data) => {
        console.log(`stderr: ${data}`);
    });

    child.on('error', (error) => console.log(`error: ${error.message}`));

    child.on('exit', (code, signal) => {
        if (code) console.log(`Process exit with code: ${code}`);
        if (signal) console.log(`Process killed with signal: ${signal}`);
        console.log(`DONE!!`);
    });
}

exports.activate = activate;

注意事项 #

• 确保graph.py确实位于你的扩展根目录下，如果脚本在子目录中（比如scripts文件夹），需要调整路径拼接逻辑，例如：path.join(context.extensionPath, 'scripts', 'graph.py')。
• 开发时可以通过console.log(context.extensionPath)打印扩展路径，验证路径是否正确。

内容的提问来源于stack exchange，提问作者Amina Nasrin