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")