Python基于BaseHTTPRequestHandler的HTTP代理V1.0

proxyHandler.py

#coding=utf8
from BaseHTTPServer import BaseHTTPRequestHandler,HTTPServer
import socket
import urllib
from zipdeal import GzipClass
class proxyHandler(BaseHTTPRequestHandler):

def do_POST(self):
#服务响应的链接uri
#类似http://xdcs-collector.ximalaya.com/api/v1/statistics
uri = self.path

#根据协议把url分成两部分，一部分是协议，例如http；
#另一部分是服务和接口：xdcs-collector.ximalaya.com/api/v1/statistics
#把另一部分内容放入到： hostAndInterface

hostAndInterface = urllib.splittype(uri)[-1]

#根据服务器拆分成两个部分，一部分是服务器：xdcs-collector.ximalaya.com
#另一部分是接口：/api/v1/statistics
host, interface = urllib.splithost(hostAndInterface)

#根据/拆分成几个部分，获取最后一个元素的值,例如：statistics

lastOfInterface=interface.split("/")[-1]

#判断接口最后一个字符是否是：statistics，如果是则执行if之后的语句
if lastOfInterface=='statistics':

#根据端口号拆分成两个部分：host、port
host, port = urllib.splitnport(host)

#如果端口号小于80，把端口号设置为80
if port < 0:
port = 80

#把服务域名转换为服务IP
host_ip = socket.gethostbyname(host)

#从headers中删除headers['Proxy-Connection']的引用
# 然后把headers['Connection']设置为'keep-alive'
del self.headers['Proxy-Connection']
self.headers['Connection'] = 'keep-alive'

#请求的第一行，把该行数据赋值给send_data变量
#requestline的格式如：POST http://xdcs-collector.ximalaya.com/api/v1/statistics HTTP/1.1
send_data = self.requestline

#重新创建个head用来保存新的heads值
head = ''
for key, val in self.headers.items():
head = head + "%s: %s\r\n" % (key, val)
#给请求的数据加上head和请求参数
send_data = send_data + head+'\r\n'

#创建一个客户端套接字链接
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#客户端发送连接请求
client.connect((host_ip, port))
#客户端发送请求数据
client.sendall(send_data)

#保存服务端返回的字段
data = ''
while True:
#把客户端从服务中接收的数据保存到一个临时变量tmp中
tmp = client.recv(8192)
#判断接收的数据是否为空，如果为空跳出循环
if not tmp:
break
#不为空，把tmp中的数据传递给data
data = data + tmp
#断开链接
client.close()

#把接收到的数据data写入到文件wfile中
self.wfile.write(data)

#self.rfile存放的是请求数据而且数据还被gzip
#通过print打印到屏幕的都是乱码
GzipClass().GZFile(self.rfile.read())

else:
pass

#do_CONNECT = do_POST
#do_GET=do_POST

def test():
host='192.168.62.33'
port=8888
try:
server = HTTPServer((host, port), proxyHandler)
print 'Welcome to the Server HTTP On %s Port %d...' %(host,port)
server.serve_forever()
except KeyboardInterrupt:
print '^C received, shutting down server'
server.socket.close()

if __name__ == '__main__':
test()

zipdeal.py

#coding=utf8
import os
import gzip

'''
创建一个全局变量FILE_PATH
并给变量赋值：gz文件所在路径
'''
class GzipClass():
def __init__(self):
self.FILE_PATH = 'D:\\DataProxy\\Pro_2.0\\GZfile\\request.gz'

#读取gz文件中的内容
def read_file(self,path):
try:
#判断路径是否存在，如果存在打开gz文件并读取
#不存在给出相应的提示信息
if os.path.exists(path):
with gzip.open(path, 'rb') as pf:
return pf.read()
else:
print 'the path % is not exist!' %(path)
finally:
pf.close()

#把内容写入gz文件
def write_file(self,path, content):
try:
#判断路径是否存在，如果存在打开gz文件并读取
#不存在给出相应的提示信息
if os.path.exists(path):
with gzip.open(path, 'wb') as f:
f.write(content)
else:
print 'the path % is not exist!' %(path)
finally:
f.close()

def GZFile(self,content):
self.write_file(self.FILE_PATH, content)
con =self.read_file(self.FILE_PATH)
print '#' * 50
print con

if __name__ == '__main__':
GzipClass().GZFile("hellO")