Python脚本内自定义域名解析的IP

发表于 分类于 Disqus：

今天因为网站上有一个小功能，原先单台服务器时执行一条url即可完成功能，但目前执行URL只会作用在某一台服务器。遂想着用Python写一段脚本然后放在集中管理平台上对所有服务器下发。其实这种情形，用shell脚本执行curl也可实现相同效果。

由于服务器走公网解析我们域名的话，同样只有一台服务器完成功能。所以我想着是不是在不修改本机hosts的情况下，在Python脚本内实现对域名的解析。带着这个想法，终于在stackoverflow上找到了可行的答案。

大致思路是，首先定义一个函数，记录你需要自定义IP的域名，并建立一个httplib.HTTPConnection的子类，在参数传递给socket.create_connection前修改self.host以封装connection方法。然后继承HTTPHandler这个类，用上面的HTTPConnection替换httplib内自带的HTTPConnection，并重写http_open方法；最后把HTTPHandler放到自定义的opener。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#!/usr/bin/env python

import urllib2
import httplib
import socket

def req_host(host):
        if host == 'www.***.cn':
                return '8.8.8.8'

class MyHTTPConnection(httplib.HTTPConnection):
        def connect(self):
                self.sock = socket.create_connection((req_host(self.host),self.port),self.timeout)

class MyHTTPHandler(urllib2.HTTPHandler):
        def http_open(self,req):
                return self.do_open(MyHTTPConnection,req)

opener = urllib2.build_opener(MyHTTPHandler)
urllib2.install_opener(opener)

url = 'http://'
f = urllib2.urlopen(url)
req = f.read()
print req