文档章节

Python针对特定服务定制的代理工具V2.0------(proxyHandler.py)

henni_719
 henni_719
发布于 2017/07/21 08:53
字数 643
阅读 3
收藏 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):
        try:
            #服务响应的链接uri
            uri = self.path
    
            #根据协议把url分成两部分,一部分是协议,例如http;
            #另一部分是服务和接口
            #把另一部分内容放入到:   hostAndInterface     
    
            hostAndInterface = urllib.splittype(uri)[-1]
    
            #根据服务器拆分成两个部分,一部分是服务器
            #另一部分是接口:/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://xxx.xxxx.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())
        except Exception,e:
            print e

    def do_CONNECT(self):
        pass
    def do_GET(self):
        pass
    
def test():
    host='192.168.xx.xx'
    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()
操作比较简单,基于BaseHTTPServer、socket、urllib实现对/api/v1/statistics接口数据进行抓取。每行都附有注释。如果有不理解可以私信我,改版本比较简介,后期会跟进优化代码

© 著作权归作者所有

henni_719
粉丝 2
博文 466
码字总数 343938
作品 0
信阳
QA/测试工程师
私信 提问
自动化运维工具—Ansible基本概念

Ansible: ansible是新出现的自动化运维工具,基于Python开发,集合了众多运维工具(puppet,cfengine,chef,func,fabric)的优点,实现了批量系统配置、批量程序部署、批量运行命令等功能...

Hito
2018/08/07
0
0
SQLAlchemy 1.2.9 为 Python 3.7 和 MySQL 8 提供改进

SQLAlchemy 1.2.9 已发布。 该版本包括针对 Core 和 ORM 组件的修复,其中包含了特定方言的组件的新问题。最重要的变化之一是,已经针对 Python 3.7 和 MySQL 8.0 改进了部分行为,以及允许在...

局长
2018/06/30
1K
0
智能会话引擎 - Icecaps

随着自然语言处理的迅速普及,越来越多的工具为公众提供了构建大型系统的工具。其中一些工具用于通用 NLP,而其他工具则侧重于特定领域,如语言建模和文本生成。然而,很少被设计为针对会话情...

匿名
09/04
318
0
Python 语言在企业应用方面遭遇的十大谬误

语言多元化是PayPal编程文化中一个重要的组成部分。在C++和Java长期流行的同时,更多的团队选择了JvaScript和Scala。同时,Braintree的收购也引入了一个久经世故的Ruby社区。 Python作为一门...

magicoding
2014/12/17
18.3K
64
SQLAlchemy 1.2.5 和 1.1.18,Python 的 ORM 框架

SQLAlchemy 1.2.5 和 1.1.18 已发布,1.2.5 引入了一系列针对 Core 和 ORM 组件的修复,包括 CTE 修复,针对 Python 3.7 以及关于 1.2 新功能的错误的调整。完整更新日志点此查看:https://w...

局长
2018/03/07
664
1

没有更多内容

加载失败,请刷新页面

加载更多

RxJava进行单元测试的方式

@Test public void completeTask_retrievedTaskIsComplete() { // Given a new task in the persistent repository final Task newTask = new Task(TITLE, ""); ......

SuShine
44分钟前
5
0
正则表达式大全

检验手机号码 # 要求:手机号码必须为11位数字,以1开头,第二位为1或5或8。import redef verify_mobile(): mob = input("请输入手机号码:") ret = re.match(r"1[358]\d{9}", m......

彩色泡泡糖
47分钟前
7
0
QT之border-image属性

一、border-image的兼容性 border-image可以说是CSS3中的一员大将,将来一定会大放光彩,其应用潜力真的是非常的惊人。可惜目前支持的浏览器有限,仅Firefox3.5,chrome浏览器,Safari3+支持...

shzwork
48分钟前
6
0
Kubernetes Operator简易教程

1. 安装operator-sdk //安装 operator-sdk$ apt-get install operator-sdk.....$ operator-sdk versionoperator-sdk version: v0.7.0$ go versiongo version go1.11.4 darwin/amd64 2......

Robotcl_Blog
48分钟前
5
0
再谈DAG任务分解和Shuffle RDD

1、DagScheduler分析 DagScheduler功能主要是负责RDD的各个stage的分解和任务提交。Stage分解是从触发任务调度过程的finalStage开始倒推寻找父stage,如果父stage没有提交任务则循环提交缺失...

守望者之父
54分钟前
4
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部