Python“请求”库-定义特定的DNS?
在我的项目中,我正在使用pythonrequests
库处理所有HTTP请求。
现在,我需要使用特定的DNS查询http服务器-有两种环境,每种环境都使用自己的DNS,并且更改是独立进行的。
因此,当代码运行时,它应使用特定于环境的DNS,而不是在我的Internet连接中指定的DNS。
-
该
urllib3.connection
模块httplib.HTTPConnection
以相同的名称子类化,已用.connect()
调用的方法替换了该方法self._new_conn
。反过来,这委托给urllib3.util.connection.create_connection()
。修补
该 功能可能是最简单的:from urllib3.util import connection _orig_create_connection = connection.create_connection def patched_create_connection(address, *args, **kwargs): """Wrap urllib3's create_connection to resolve the name elsewhere""" # resolve hostname to an ip address; use your own # resolver here, as otherwise the system resolver will be used. host, port = address hostname = your_dns_resolver(host) return _orig_create_connection((hostname, port), *args, **kwargs) connection.create_connection = patched_create_connection
并且您将提供自己的代码来
host
将地址的一部分解析为ip地址,而不是依靠connection.create_connection()
调用(自动包装socket.create_connection()
)为您解析主机名。像所有的monkeypatching一样,请注意,在以后的版本中代码没有显着更改;该补丁是针对
urllib3
1.21.1版创建的。但应该适用于1.9之前的版本。
请注意,此答案已重新编写以与较新的
urllib3
版本一起使用,这些新版本添加了更方便的修补位置。请参阅适用于版本<1.9的旧方法的编辑历史记录,以作为供应商urllib3
版本的补丁程序,而不是独立安装。