Howto:close_fds = True的解决方法,并在Windows上重定向stdout / stderr

时间:2018-02-07 18:50:06

标签: python windows subprocess

我遇到了一个问题:使用Python 2.7,无法使用

创建子进程
subprocess.Popen([.......], close_fds=True, stdout=subprocess.PIPE, ...)

在Windows上,因为有限制。在我的情况下需要使用close_fds,因为我不希望子进程从已经打开的文件继承文件描述符。

这是known bug,已在Python 3.4+

上修复

我遇到的问题是:如何在不获取

的情况下使用子流程
  如果重定向stdin / stdout / stderr,则Windows平台不支持

close_fds

下面的答案

1 个答案:

答案 0 :(得分:2)

这绝对是一个棘手的黑客:答案是在使用subprocess模块之前迭代已打开的文件描述符。

def _hack_windows_subprocess():
    """HACK: python 2.7 file descriptors.
    This magic hack fixes https://bugs.python.org/issue19575
    by adding HANDLE_FLAG_INHERIT to all already opened file descriptors.
    """
    # See https://github.com/secdev/scapy/issues/1136
    import stat
    from ctypes import windll, wintypes
    from msvcrt import get_osfhandle

    HANDLE_FLAG_INHERIT = 0x00000001

    for fd in range(100):
        try:
            s = os.fstat(fd)
        except:
            continue
        if stat.S_ISREG(s.st_mode):
            handle = wintypes.HANDLE(get_osfhandle(fd))
            mask   = wintypes.DWORD(HANDLE_FLAG_INHERIT)
            flags  = wintypes.DWORD(0)
            windll.kernel32.SetHandleInformation(handle, mask, flags)

这是一个没有它会崩溃的样本:

import os, subprocess
f = open("a.txt", "w")
subprocess.Popen(["cmd"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
f.close()
os.remove(f.name)
  

追踪(最近一次呼叫最后一次):

     

文件“stdin”,第1行,在模块

中      

WindowsError:[错误32] Le processus ne peutpasactÚderau   fichier car ce fichierestutilisÚp​​arun autre processus:'a.txt'

现在有了修复:

import os, subprocess
f = open("a.txt", "w")
_hack_windows_subprocess()
subprocess.Popen(["cmd"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
f.close()
os.remove(f.name)

作品。

希望我帮助