Python中如何使用Requests库设置自定义DNS解析

在Python的requests库中设置自定义DNS解析，最直接的方法是结合requests和httpx库的底层能力，或者使用aiohttp。这里给你一个最实用的方案。

核心原理：修改操作系统的hosts文件映射，或者使用支持自定义DNS解析器的HTTP客户端。

方法一：使用requests + urllib3的底层控制（推荐）

requests库基于urllib3。你可以自定义一个urllib3的连接池，为其绑定一个能解析特定域名的自定义Resolver。但更常见的做法是直接劫持HTTP请求的URL，或者使用hosts文件覆盖。

下面是一个通过安装requests-toolbelt和修改HTTPAdapter来实现的示例：

import requests
from requests_toolbelt.adapters.source import SourceAddressAdapter
from urllib3 import PoolManager
import socket

class CustomDNSResolver(PoolManager):
    """自定义DNS解析的PoolManager"""
    def __init__(self, host_map, *args, **kwargs):
        self.host_map = host_map  # 映射字典，如 {'example.com': '93.184.216.34'}
        super().__init__(*args, **kwargs)
    
    def _new_conn(self):
        """创建新连接时，如果host在映射中，则使用映射的IP"""
        # 这里简化处理，实际需重写更底层的方法
        # 更完整的实现需要修改urllib3的connectionpool
        pass

# 更实用的方法：使用requests的hooks或直接修改URL
def dns_hook(response, *args, **kwargs):
    """一个hook示例，但无法在请求发送前修改DNS"""
    pass

# 最直接的方法：使用IP地址并设置Host头
def request_with_custom_dns():
    target_ip = '93.184.216.34'  # example.com的真实IP
    url = f'http://{target_ip}/path'
    headers = {'Host': 'example.com'}  # 关键：设置Host头
    
    response = requests.get(url, headers=headers)
    print(response.status_code)
    print(response.text[:200])

if __name__ == '__main__':
    request_with_custom_dns()

方法二：使用httpx库（支持异步，更现代）

httpx库原生支持自定义DNS解析，通过transport参数：

import httpx
import asyncio

async def custom_dns_request():
    # 定义DNS覆盖映射
    transport = httpx.AsyncHTTPTransport(
        limits=httpx.Limits(max_connections=100),
        retries=2,
        # 使用自定义的DNS解析函数
        # 注意：这里需要自己实现一个resolver
    )
    
    # 更简单的方法：使用httpx的client mount，覆盖特定域名
    async with httpx.AsyncClient() as client:
        # 方法：直接替换URL中的域名为IP，并设置Host头
        url = 'http://93.184.216.34/path'
        headers = {'Host': 'example.com'}
        resp = await client.get(url, headers=headers)
        print(resp.status_code)

# asyncio.run(custom_dns_request())

方法三：修改系统hosts（临时方案）

对于临时测试，可以直接修改hosts文件，但这不是编程解决方案：

import subprocess
import platform

def add_hosts_entry(hostname, ip):
    """临时添加hosts条目（需要管理员权限）"""
    if platform.system() == 'Windows':
        hosts_path = r'C:\Windows\System32\drivers\etc\hosts'
    else:
        hosts_path = '/etc/hosts'
    
    entry = f'{ip}\t{hostname}\n'
    with open(hosts_path, 'a') as f:
        f.write(entry)
    print(f'Added: {hostname} -> {ip}')

总结

最可靠的方法还是用IP发起请求并设置Host头。如果必须用requests库，可以这样封装：

import requests

class CustomDNSRequest:
    def __init__(self, dns_map=None):
        self.dns_map = dns_map or {}
    
    def get(self, url, **kwargs):
        # 解析URL，替换域名
        from urllib.parse import urlparse
        parsed = urlparse(url)
        if parsed.hostname in self.dns_map:
            new_ip = self.dns_map[parsed.hostname]
            # 重建URL使用IP
            new_netloc = new_ip if parsed.port is None else f'{new_ip}:{parsed.port}'
            new_url = parsed._replace(netloc=new_netloc).geturl()
            # 设置Host头
            headers = kwargs.get('headers', {})
            headers['Host'] = parsed.hostname
            kwargs['headers'] = headers
            url = new_url
        return requests.get(url, **kwargs)

# 使用示例
client = CustomDNSRequest({'example.com': '93.184.216.34'})
resp = client.get('http://example.com/path')

一句话建议：用IP请求并设置Host头是最简单直接的方法。