如何使用os.execv()在python中继承stdin和stdout

时间:2011-12-14 05:54:13

标签: python fork stdout stdin execv

首先,我写了一个c ++代码如下:

#include <cstdio>
int main()
{
    int a,b;
    while(scanf("%d %d",&a,&b) == 2)
        printf("%d\n",a+b);
    return 0;
}

我使用g++ -o a a.cpp来编译它。

之后,我编写了如下的python代码:

import os,sys
sys.stdin = open("./data.in","r")
sys.stdout = open("./data.out","w")
pid = os.fork()
if pid == 0:
    cmd = ["./a","./a"]
    os.execv(cmd[0],cmd)

但是,data.out文件不包含任何内容。也就是说,子进程没有从父进程继承stdin和stdout。但是当我按如下方式编写c ++代码时:

#include<unistd.h>
#include<cstdio>
int main()
{
    freopen("data.in","r",stdin);a
    freopen("data.out","w",stdout);
    int pid = fork();
    if(pid == 0)
    {
        char* cmd[]= {"./a","./a"};
        execv(cmd[0],cmd);
    }
    return 0;
}

我在data.out得到了正确的答案,也就是说execv在c ++代码中有效。

那么,我应该怎么做才能让execv也能在python中运行?我真的需要这个功能才能工作,任何人都可以告诉我吗?非常感谢!

data.in包含以下内容:

1 1

1 个答案:

答案 0 :(得分:5)

在您的python代码中,您已打开新IO流并设置sys.stdinsys.stdout Python 名称以指向这些。他们将拥有自己的文件描述符(很可能是3和4)。但是,标准文件描述符01将保持不变,并继续指向其原始位置。这些是在{C ++程序中exec进入stdin和stdout时继承的描述符。

您需要确保在执行前调整实际的文件描述符。

os.dup2(sys.stdin.fileno(), 0)
os.dup2(sys.stdout.fileno(), 1)