尝试屏蔽f2py进程输出时出现WinError 6（无效句柄）问题

问题背景 #

要屏蔽f2py扩展模块的控制台输出时，Python标准的redirect_stdout方法无效——因为Fortran的输出不经过Python的print逻辑，直接写入系统标准输出。之前用一段基于文件句柄重定向的代码在Linux上正常运行，但在Windows环境下会报错：

OSError: [WinError 6] Invalid handle

复现错误的测试代码：

with LoggerLowLevelMuted(filename="testout.txt"):
   print("test")

原实现代码如下：

class LoggerLowLevelMuted():
    """A logger to mute low level output form Fortran modules"""
    def __init__(self, filename=None):
        self.filename = filename
        if filename is None:
            self.filename = os.devnull
        self.stdchannel = sys.__stdout__
    
    def __enter__(self):
        self.dest_file = open(self.filename, 'a')
        self.oldstdchannel = os.dup(self.stdchannel.fileno())
        os.dup2(self.dest_file.fileno(), self.stdchannel.fileno())
        os.close(self.stdchannel.fileno())

    def __exit__(self, exc_type, exc_val, exc_tb):
        os.dup2(self.oldstdchannel, self.stdchannel.fileno())
        self.dest_file.close()
        os.close(self.oldstdchannel)

问题原因 #

Windows和Linux的系统句柄机制存在核心差异：

1. 原代码中os.close(self.stdchannel.fileno())直接关闭了sys.__stdout__对应的控制台句柄，Windows中控制台句柄不允许被随意关闭，关闭后后续的dup2恢复操作会因句柄无效触发报错。
2. Linux下sys.__stdout__.fileno()对应普通文件描述符，而Windows中该句柄指向控制台设备，dup/dup2的行为逻辑与Linux不完全一致。

修复方案 #

修改后的代码兼容Windows和Linux，核心是避免关闭原始标准输出句柄，同时调整句柄重定向逻辑：

import os
import sys

class LoggerLowLevelMuted():
    """跨平台屏蔽Fortran模块底层输出的工具类"""
    def __init__(self, filename=None):
        self.filename = filename if filename is not None else os.devnull
        self._old_stdout = None
        self._dest_fd = None

    def __enter__(self):
        # 保存原始标准输出的文件描述符副本
        self._old_stdout = os.dup(sys.stdout.fileno())
        # 打开目标文件并获取其文件描述符
        self._dest_fd = open(self.filename, 'a').fileno()
        # 将标准输出重定向到目标文件
        os.dup2(self._dest_fd, sys.stdout.fileno())
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 恢复原始标准输出
        os.dup2(self._old_stdout, sys.stdout.fileno())
        # 释放资源
        os.close(self._old_stdout)
        os.close(self._dest_fd)

修改要点 #

• 不关闭sys.stdout的原始句柄，仅保存其副本，避免Windows下控制台句柄失效问题。
• 直接操作sys.stdout.fileno()而非sys.__stdout__，sys.stdout是Python当前的标准输出流，跨平台兼容性更好。
• 单独管理目标文件的文件描述符，确保退出时正确清理资源。

调用示例 #

with LoggerLowLevelMuted(filename="testout.txt"):
    call_my_f2py_module()  # Fortran输出会被写入指定文件或/dev/null

内容的提问来源于stack exchange，提问作者Amos Egel