Python wsgiref 模块源码浅析

def demo_app(environ,start_response):
    from StringIO import StringIO
    stdout = StringIO()
    print >>stdout, "Hello world!"
    print >>stdout
    h = environ.items(); h.sort()
    for k,v in h:
        print >>stdout, k,'=', repr(v)
    start_response("200 OK", [('Content-Type','text/plain')])
    return [stdout.getvalue()]

Python的web框架，都是一个wsgi app。通过自己构造wsgi应用，很容易写出一个框架的骨架。python定义了wsgi，让web框架几乎可以大一统了。下面就来分析，python是如何实现这个协议的。

WSGIServer

python使用WSGIServer和WSGIRequestHandler构建wsgi协议的服务。

class WSGIServer(HTTPServer):

    application = None

    def server_bind(self):
        HTTPServer.server_bind(self)
        self.setup_environ()

    def setup_environ(self):
        # 甚至环境变量
        env = self.base_environ = {}
        env['SERVER_NAME'] = self.server_name
        env['GATEWAY_INTERFACE'] = 'CGI/1.1'
        env['SERVER_PORT'] = str(self.server_port)
        env['REMOTE_HOST']=''
        env['CONTENT_LENGTH']=''
        env['SCRIPT_NAME'] = ''

    def get_app(self):
        return self.application

    def set_app(self,application):
        self.application = application

WSGIServer继承HTTPServer，重写了server_bind仿佛，设置了一些专用的环境变量。比较简单，我们也知道，Server只是处理socket连接相关的逻辑，RequestHandler才是处理客户端请求逻辑。

WSGIRequestHandler

WSGIRequestHandler 也不复杂，只有3个方法，get_environ用来设置并返回环境变量的字典，get_stderr用于获取标准错误输出。handle则是重写基类BaseRequestHandler的方法。前文我们也提到，handle用于不同协议处理客户端的入口。

handler

def handle(self):
    self.raw_requestline = self.rfile.readline()
    if not self.parse_request(): 
        return

    handler = ServerHandler(
        self.rfile, self.wfile, self.get_stderr(), self.get_environ()
    )
    handler.request_handler = self     
    handler.run(self.server.get_app())

BaseHandler 和 ServerHandler

ServerHandler 来自wsgiref的handlers模块，它继承 BaseHandler类。又一个带base类。BaseHandler主要用于操作WSGI app。run方法就是在该类定义的。

def run(self, application):
    try:
        self.setup_environ()
        self.result = application(self.environ, self.start_response)
        self.finish_response()
    except:
        try:
            self.handle_error()
        except:
            self.close()
            raise   

run方法最重要的就是调用自定义的wsgi app，并把在finish_reponse方法中把结果send给客户端。

finish_response

def finish_response(self):
    try:
        if not self.result_is_file() or not self.sendfile():
            for data in self.result:
                self.write(data)
            self.finish_content()
    finally:
        self.close()

把可迭代对象数据返回。通过write方法把数据写入缓冲可写文件，然后发送给客户端。

send_preamble

def send_preamble(self):

    if self.origin_server:
        if self.client_is_modern():
            self._write('HTTP/%s %s\r\n' % (self.http_version,self.status))
            if 'Date' not in self.headers:
                self._write(
                    'Date: %s\r\n' % format_date_time(time.time())
                )
            if self.server_software and 'Server' not in self.headers:
                self._write('Server: %s\r\n' % self.server_software)
    else:
        self._write('Status: %s\r\n' % self.status)

wsgiref模块还包含了另外几个子模块，上面的的内容主要来自 simple_server 和 handler模块，此外还有headers喝validata模块，用于包装headers和做验证，暂且忽略了。

wsgi是 python web框架中的标准，实现了wsgi协议就能写出一个web框架。python的世界里不少知名的框架。接下来将会探究一下werkzeug和flask的源码以及tornado的源码。