ve-compiler-2.0/compilers/pyCompiler/pythonCompiler.js

Upgraded version of this package - https://www.npmjs.com/package/ve-compiler

const { spawn } = require("child_process");
const fs = require("fs");
const path = require("path");

const executePython = async (filepath, input = "") => {
  try {
    const jobId = path.basename(filepath).split(".")[0];
    const codebasePath = path.join(__dirname, "../../codebase");

    if (!fs.existsSync(codebasePath)) {
      fs.mkdirSync(codebasePath, { recursive: true });
    }

    const { stdout, stderr } = await runPythonWithInput(filepath, input);
    return { stdout, stderr };
  } catch (error) {
    throw new Error(`Execution error: ${error}`);
  }
};

const runPythonWithInput = (filepath, input = "") => {
  return new Promise((resolve, reject) => {
    const process = spawn("python3", [filepath]);

    let stdout = "";
    let stderr = "";

    process.stdout.on("data", (data) => {
      stdout += data.toString();
    });

    process.stderr.on("data", (data) => {
      stderr += data.toString();
    });

    process.on("error", reject);

    process.on("close", (code) => {
      if (code !== 0) {
        return reject(new Error(`Python process exited with code ${code}, stderr: ${stderr}`));
      }
      resolve({ stdout, stderr });
    });


    if (input) {
      const lines = Array.isArray(input) ? input : [input];
      for (const line of lines) {
        process.stdin.write(line + '\n');
      }
    }

    process.stdin.end();
  });
};

module.exports = {
  executePython,
};