关于Java调用python脚本

1.使用Runtime.getRuntime().exec调用

脚本调用工具类

// python虚拟环境路径
private static final String PYTHON_PATH = (String) PropertiesUtil.get("python-virtual-address");
// 日志类
private static final Logger LOGGER = LoggerFactory.getLogger(CallPyUtils.class);
/**
 * 调用python文件
 *
 * @param filePath 文件绝对路径
 * @param args     要传递给python程序的参数
 * @return python打印的json数据
 */
public static List<String> callPy(String filePath, String[] args) {
    List<String> result = new ArrayList<>();
    try {
        // 前面为python虚拟环境路径和python程序路径
        List<String> args1 = Stream.of(PYTHON_PATH, filePath).collect(Collectors.toList());
        // 将要传递给python文件的参数添加到字符串列表
        args1.addAll(Stream.of(args).collect(Collectors.toList()));
        String allArgs = String.join(" ", args1);
        // 执行py文件
        LOGGER.info("开始执行python文件: 文件路径{}, 执行参数{}", filePath, allArgs);
        // allArgs的格式为(虚拟环境路径 python文件路径 传递的参数)
        Process proc = Runtime.getRuntime().exec(allArgs);

        // 输入流和错误流的读取顺序不能变,否则会导致程序一直等待
        // 用输入输出流来截取结果
        BufferedReader in = new BufferedReader(new InputStreamReader(proc.getInputStream(), StandardCharsets.UTF_8));
        // 获取数据并将其实例化
        result = in.lines().collect(Collectors.toList());

        // 获取错误流
        BufferedReader error = new BufferedReader(new InputStreamReader(proc.getErrorStream(), StandardCharsets.UTF_8));
        String err;
        while ((err = error.readLine()) != null) {
            LOGGER.error("调用python脚本错误, 路径为{}, 错误信息为: {}", filePath, err);
        }
        LOGGER.info("python文件执行完成: 文件路径{}", filePath);
        in.close();
        proc.waitFor();
    } catch (Exception e) {
        LOGGER.error("调用python脚本错误", e);
    }
    return result;
}

python脚本示例

import sys
# 从命令行获取输入的参数,并将其打印到输出流,以供Java读取
print(sys.argv[1], sys.argv[2], sys.argv[3])
# 手动输出错误流的方法,python发生错误时会自动将报错打印到错误流中
# sys.stderr.write("这是错误流")

调用脚本代码示例

// py文件地址
private static final String filePath = "python文件路径";
// 设置需要传入的参数
String[] args = new String[]{"需要传入的参数"};
// 调用python脚本,获取输出流
List<String> strings = callPy(filePath, args);

 

上一篇:学习 Python 编程的三种困境


下一篇:上传图片/视频含有大量重复的解决办法