在做Web开发时,一个很重要的概念就是服务端和应用程序之间的沟通协议,比如java中的servlet,由于servlet的存在,使得用java开发的web程序既可以跑在tomcat上,也可以是jetty。反之亦然。而在python中,对应的协议也就是WSGI协议,本文的目标就是实现一个可以支持python主流框架的web服务器,也帮助自己加强对WSGI协议的理解。
实验环境:
python3.5
一个简单的服务器实现
这一节并不会直接给出一个遵循WSGI协议规范的服务器,只是单纯从如何与客户端通信的角度来考虑实现。我们都知道,HTTP协议是建立在TCP协议的基础上,所以首先我们借助python标准库中的socket来实现TCP通信。下面是我的实现代码和解释:
# wsgi_a.py
import socket
HOST, PORT = '', 8888
listen_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
listen_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
listen_socket.bind((HOST, PORT))
listen_socket.listen(1)
print('Serving HTTP on port %s ...', PORT)
# run or not
flag = True
while flag:
try:
client_connection, client_address = listen_socket.accept()
request = client_connection.recv(1024)
print(request)
http_response = """\
HTTP/1.1 200 OK
Hello, World!
"""
client_connection.sendall(http_response.encode())
client_connection.close()
except KeyboardInterrupt:
flag = False
print('exit')
这里需要说明的是关于socket的标准库中的基本函数及常量:
socket.socket(socket.AF_INET, socket.SOCK_STREAM) 返回一个socket对象,其中第一个参数需指明IP地址类型(IPv4, IPv6, ...),第二个参数用来指明通信的协议,这里两个参数的意思分别为(IPv4, TCP)。
socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)设置套接字重用。
socket.bind((HOST,PORT))绑定端口
socket.listen(1) 设置客户端的连接个数
socket.accept() 阻塞监听
打开终端。命令行中运行该脚本并在浏览器中输入
http://127.0.0.1:8888
即可查看结果。
满足WSGI协议的服务器
简单版本的服务器仅仅只是实现了与客户端之间的通信,同时将请求处理也放在了服务器里,并没有将两者分开。也没有对现有的主流框架进行支持。因此为了实现一个通用的Web服务器,根据WSGI协议,我们需要添加两个关键的部分。一个传给应用端的上下文环境,并一个是需要给应用端调用的start_response函数。详情可以参照我之前的翻译PEP333。
#wsgi_b.py
import socket
import sys, io
from datetime import date
class WSGIServer(object):
"""docstring for WSGIServer"""
def __init__(self, host, port, application):
self.host = host
self.port = port
self.listen_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.listen_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.listen_socket.bind((self.host, self.port))
self.listen_socket.listen(1)
self.flag = True
self.application = application
print('Serving HTTP on port %s ...', self.port)
def get_environ(self, request_data):
env = {}
# CGI variables
path, server, *args = request_data
env['REQUEST_METHOD'], env['PATH_INFO'], _= path.split()
env['SERVER_NAME'], env['SERVER_PORT'] = self.host, str(self.port)
# WSGI variables
env['wsgi.version'] = (1, 0)
env['wsgi.url_scheme'] = 'http'
env['wsgi.input'] = io.StringIO(self.request_data.decode())
env['wsgi.errors'] = sys.stderr
env['wsgi.multithread'] = False
env['wsgi.multiprocess'] = False
env['wsgi.run_once'] = False
return env
def make_server(self):
while self.flag:
try:
self.client_connection, self.client_address = self.listen_socket.accept()
self.request_data = self.client_connection.recv(1024)
request_data = self.request_data.decode().splitlines()
env = self.get_environ(request_data)
result = self.application(env, self.start_response)
self.make_response(result)
except KeyboardInterrupt:
self.flag = False
print('exit')
def make_response(self, result):
try:
status, response_headers = self.headers_set
response = 'HTTP/1.0 {status}\r\n'.format(status=status)
for header in response_headers:
response += '{0}: {1}\r\n'.format(*header)
response += '\r\n'
for data in result:
response += data.decode()
self.client_connection.sendall(response.encode())
finally:
self.client_connection.close()
def start_response(self, status, response_headers, exc_info=None):
# Add necessary server headers
server_headers = [
('Date', date.today().strftime('%Y-%m-%d')),
('Server', 'WSGIServer 0.2'),
]
self.headers_set = [status, response_headers + server_headers]
这个版本的服务器实现了一个简单的WSGI规范,但并不是全部,不过已经可以实现与多个框架的通信。相关的解释如下:
WSGI的初始化参数分别为主机ip,端口及需要调用的应用程序。
get_environ函数,从request_data中获取相应的CGI变量及WSGI变量,并传给应用程序。
make_server,阻塞监听端口。并接收客户端传来的消息交由应用程序进行处理,最后再将响应的结果打包,转成响应格式,交付给客户端。
make_response。将应用程序处理结果及请求头打包成响应,并关闭连接。
start_response,这个函数交给应用程序调用,其参数分别为状态码,响应头以及错误消息处理。
常见框架测试
这里是我的测试代码,通过我们自己写的服务器,可以成功的跑起weppy,flask及一个简单的满足WSGI规范的application。
# test.py
from wsgi_b import WSGIServer
from flask import Flask
from weppy import App
weppy_application = App(__name__)
flask_application = Flask(__name__)
@weppy_application.route("/")
def hello():
return "Hello World! from weppy"
@flask_application.route("/")
def hello():
return "Hello World! from flask"
def application(environ, start_response):
"""Simplest possible application object"""
status = '200 OK'
response_headers = [('Content-type', 'text/plain')]
start_response(status, response_headers)
return [b'Hello world! from simple app \n']
if __name__ == '__main__':
server = WSGIServer('127.0.0.1',8888,weppy_application).make_server()
但是这个服务器仍然还有许多需要完善的地方。不过不妨碍其做为学习WSGI协议时的补充。