写更地道的python第三课笔记

这个课程是这个博客的博主发起的一个众筹项目,博主blog写得不错,也很喜欢教学,因此就有了这个项目,项目资金达到1w+美元之后,所有的视频都会免费发布。

第三课继续将讲解第二课的脚本,该脚本的功能是实现一个网络中转服务

  • 能使用更高级的库就不要使用低级的库
    用threading而不是thread

  • 字符串分割可以用partition
    比起用

贴上最后修改好的源码,全部源码在这里

"""An HTTP proxy that supports IPv6 as well as the HTTP CONNECT method, among
other things."""

# Standard libary imports
import socket
import threading
import select

__version__ = '0.1.0 Draft 1'
BUFFER_LENGTH = 8192
VERSION = 'Python Proxy/{}'.format(__version__)
HTTP_VERSION = 'HTTP/1.1'


class Proxy(object):
    """Handles connections between the HTTP client and HTTP server."""
    def __init__(self, connection):
        self.client = connection
        self.client_buffer = ''
        self.target = None
        method, path, protocol = self.get_base_header()
        if method == 'CONNECT':
            self.method_connect(path)
        else:
            self.method_others(method, path, protocol)

    def get_base_header(self):
        """Return a tuple of (method, path, protocol) from the recieved
        message."""
        while '\n' not in self.client_buffer:
            self.client_buffer += self.client.recv(BUFFER_LENGTH)
        (data, _, self.client_buffer) = self.client_buffer.partition('\n')
        return data.split()

    def method_connect(self, path):
        """Handle HTTP CONNECT messages."""
        self._connect_to_target(path)
        self.client.send('{http_version} 200 Connection established\n'
                         'Proxy-agent: {version}\n\n'.format(
                             http_version=HTTP_VERSION,
                             version=VERSION))
        self.client_buffer = ''
        self._read_write()

    def method_others(self, method, path, protocol):
        """Handle all non-HTTP CONNECT messages."""
        path = path[len('http://'):]
        host, _, path = path.partition('/')
        path = '/{}'.format(path)
        self._connect_to_target(host)
        self.target.send('{method} {path} {protocol}\n{client_buffer}'.format(
            method=method,
            path=path,
            protocol=protocol,
            client_buffer=self.client_buffer))
        self.client_buffer = ''
        self._read_write()

    def _connect_to_target(self, host):
        """Create a connection to the HTTP server specified by *host*."""
        port = 80
        if ':' in host:
            host, _, port = host.partition(':')
        (socket_family, _, _, _, address) = socket.getaddrinfo(host, port)[0]
        self.target = socket.socket(socket_family)
        self.target.connect(address)

    def _read_write(self):
        """Read data from client connection and forward to server
        connection."""
        sockets = [self.client, self.target]
        try:
            while 1:
                (receive, _, error) = select.select(sockets, [], sockets, 10)
                if error:
                    break
                elif receive:
                    for source in receive:
                        data = source.recv(BUFFER_LENGTH)
                        if not data:
                            return
                        if source is self.client:
                            destination = self.target
                        else:
                            destination = self.client
                        destination.sendall(data)
        finally:
            self.client.close()
            self.target.close()


def start_server():
    """Start the HTTP proxy server."""
    host = 'localhost'
    port = 8080
    listener = socket.socket(socket.AF_INET)
    listener.bind((host, port))
    print 'Serving on {0}:{1}.'.format(host, port)
    listener.listen(0)
    while 1:
        connection, address = listener.accept()
        print 'Got connection from {}'.format(address)
        threading.Thread(
            target=Proxy, args=(connection, )).run()

if __name__ == '__main__':
    start_server()
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,189评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,577评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,857评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,703评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,705评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,620评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,995评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,656评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,898评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,639评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,720评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,395评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,982评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,953评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,195评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,907评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,472评论 2 342

推荐阅读更多精彩内容