在导入的模块中模拟 python 的原生 `open` 函数

Question

我想覆盖 python 的本机 open 函数。 这是我尝试过的。我的目录中有以下文件：

main.py
mock.py
test.txt
test_main.py

main.py 包含以下内容：

fs = open('test.txt', 'r')

mock.py 包含以下内容：

def open():
  print("hello")

test.txt 包含以下内容：

abc

test_main.py 包含以下内容：

import pathlib
import mock

if __name__=="__main__":
  file_path = pathlib.Path('test.txt')  
  open = mock.open
  import main
  file_path.unlink()

在 test_main.py 的第 6 行中，我尝试通过设置自己的内容来覆盖 将 mock.py 中的函数设置为关键字 open。在 test_main.py 中 看起来 python 将使用我的定义。然而，在 导入的 main.py 看起来 python 仍然求助于本机定义，因此 python test_main.py 抛出以下错误：

PermissionError: [WinError 32] 该进程无法访问该文件，因为该文件正在被另一个进程使用：“test.txt”

如何重新定义 open 以便我的定义适用于所有 导入的命名空间？

Answer 1

我刚刚找到它，感谢 这个问题。需要导入builtins：

import pathlib
import mock
import builtins

if __name__=="__main__":
  file_path = pathlib.Path('test.txt')  
  builtins.open = mock.open
  import main
  file_path.unlink()