如何重载Python中的内置模块？

我正在尝试将主机绑定到我的 python 程序中的指定 ip。只是让它在python程序中起作用，所以我不打算修改/etc/hosts file.

我尝试添加一些代码create_connection函数于socket.py对于主机 IP 转换，如下所示：

host, port = address  # the original code in socket.py
# My change here:
if host == "www.google.com":
    host = target_ip  
for res in getaddrinfo(host, port, 0, SOCK_STREAM): # the original code in socket.py

我发现效果很好。

现在我希望主机 IP 转换只能在这个 python 程序中起作用。

所以我的问题是：我怎样才能让我的python程序在使用时导入这个socket.py而不是内置的import socket?

为了清楚起见，这里有一个例子。假设“test”是我的工作目录：

test
|--- main.py
|--- socket.py

在这种情况下：

1. 如何使 main.py 使用 test/socket.pyimport socket?

2. 如何让其他模块使用 test/socket.py 使用import socket?

我认为更改模块查找路径顺序可能会有所帮助。但我发现即使当前路径（'') 位于第一位sys.path已经和import socket仍然导入内置的 scoket 模块。

你可以打猴子补丁sys.modules，放置您自己的模块而不是标准模块socket，在导入可能使用它的任何其他模块之前。

# myscript.py

from myproject import mysocket
import sys
sys.modules['socket'] = mysocket

# ... the rest of your code
import requests
...

为了那个原因，mysocket应该暴露标准的所有内容socket does.

# mysocket.py

import socket as _std_socket
from socket import *  # expose everything

def create_connection(address, *args, **kwargs):
    if address == ...:
       address = ...
    return _std_socket.create_connection(address, *args, **kwargs)

这可能过于简单化了mysocket.py应该看起来像。在将其用于生产之前，您可能需要添加一些定义，但您已经明白了。

另一种方法是猴子修补socket模块本身，即覆盖原始模块内的名称。

# myscript.py
import socket

def create_connection2(...):
    ...

socket.create_connection = create_connection2

# ... the rest of your code
import requests
...

我更喜欢前一种方法，因为它更干净，因为您不需要去inside模块，只是为了隐藏它并从外部覆盖其中的一些内容。