动手写一个简单的Web框架(Werkzeug路由问题)

动手写一个简单的Web框架(Werkzeug路由问题)

继承上一篇博客,实现了HelloWorld,但是这并不是一个Web框架,只是自己手写的一个程序,别人是无法通过自己定义路由和返回文本,来使用的,所以在这篇博客中,将实现一个简单的路由自定义功能

首先引入werkzeug中的两个工具,分别是Map,和Rule,需要通过以下代码引入

from werkzeug.routing import Map, Rule

这两个可以完成路由定义和匹配的基本实现,其次,还引入到werkzeug中的Request对象来处理environ

首先,我们看一下Map和Rule的使用:

from werkzeug.routing import Rule, Map
url_map = Map([
Rule('/', endpoint='index', methods=['GET']),
Rule('/about', endpoint='about', methods=['GET']),
])
print(url_map)

上面这段代码定义了一个url_map的对象,它是Map的实例,传入参数是一个列表,在列表里面,存储着路由,我们一个一个来看,Rule中首先传入的是路由参数,代表着访问的路由,然后是methods,用于标记请求的方法,再接着是endpoint,这个参数其实只是给这个路由起一个名字,但是关系确是重大的,我们下一步进行的视图函数的定义以及和路由绑定在一起就离不开endpoint这个参数,它可由使用者自行定义,也可默认为视图函数的函数名。可以看到打印出来的url_map,是路由和endpoint以及methods一一对应的

动手写一个简单的Web框架(Werkzeug路由问题)

Map中有一个add方法,可以用于添加新的路由

from werkzeug.routing import Rule, Map
url_map = Map([
Rule('/', endpoint='index', methods=['GET']),
Rule('/about', endpoint='about', methods=['GET']),
])
url_map.add(Rule('/hello', endpoint='hello', methods=['GET']))
print(url_map)

可以看到运行结果

动手写一个简单的Web框架(Werkzeug路由问题)

关于路由的匹配问题,可以使用Map的match来进行匹配,用法只需传入路由,即可获取endpoint,在运行前,需要使用Map中的bind,来实例化一个urls,它必需的参数为hostname,还有其他的参数可自行研究,我们需要根据实例化出的urls中的match来实现路由的匹配:

from werkzeug.routing import Rule, Map
url_map = Map([
Rule('/', endpoint='index', methods=['GET']),
Rule('/about', endpoint='about', methods=['GET']),
])
url_map.add(Rule('/hello', endpoint='hello', methods=['GET']))
print(url_map)
urls = url_map.bind('127.0.0.1:5000')
url = '/'
endpoint = urls.match(path_info=url)
print(endpoint)

运行结果

动手写一个简单的Web框架(Werkzeug路由问题)

看到,匹配到了路由的endpoint

那么,endpoint要怎么使用呢?

我们在添加视图函数的时候,只是添加了路由,我们对于视图函数的存储还没有实现,其实,这里需要自己定义一个字典,用于存储endpoint和视图函数的键值对,这样,就可以通过获取和存入的endpoint来实现获取和存入视图函数,这部分代码就整合到所有代码中:

from werkzeug.wrappers import Response, Request
from werkzeug.serving import run_simple
from werkzeug.routing import Map, Rule class Jlask(object):
# 初始化url_map和存储endpoint对应视图函数
# 这里是一个小坑,不能再__init__中初始化,这样会使得在类被调用的时候,执行了初始化的代码,导致原先创建的路由都丢失
url_map = Map([])
endpoint_dict = {} def dispatch_request(self, request):
# 获取请求的路由
url = request.path
urls = self.url_map.bind('127.0.0.1:5000')
# 匹配得到endpoint
endpoint = urls.match(path_info=url)[0]
# 获取到视图函数处理得结果
view_func = self.endpoint_dict[endpoint](request)
return Response(view_func) def wsgi_app(self, environ, start_response):
# 启动
# 实例化request,存入environ,这里的request就是werkzeug中对于environ的处理,可以通过request来获取各种请求信息
request = Request(environ)
# 调用dispatch_request解析url,得到对应的视图函数处理的结果
response = self.dispatch_request(request)
return response(environ, start_response) def __call__(self, environ, start_response):
# 调用到wsgi_app,来执行url对应视图函数
return self. wsgi_app(environ, start_response) def add_url_rule(self, rule, endpoint=None, view_func=None, **options):
# 获取请求方法
methods = options.pop('methods', None)
# 如果没有定义endpoint,就用试图函数的名字
if endpoint is None:
endpoint = view_func.__name__
# 生成rule
# rule: 定义的路由
self.url_map.add(Rule(rule, endpoint=endpoint, methods=methods))
# 存储endpoint对应的视图函数
self.endpoint_dict[endpoint] = view_func
# print(self.endpoint_dict)
# print(self.url_map) app = Jlask() def Hello(request):
return "Hello World"
app.add_url_rule(rule='/', view_func=Hello, methods=['GET']) if __name__ == '__main__':
run_simple('127.0.0.1', 5000, Shortly())

到此就完成了简单得包含路由匹配功能的“Web框架”


更新:

对于自定义的视图函数,需要考虑使用者是否自己构造Response对象,所以需要在dispatch_request中对调用的视图函数返回值进行判断,如果是字符串对象,就构造Response,否则就直接返回

def dispatch_request(self, request):
url = request.path
urls = self.url_map.bind('127.0.0.1:5000')
endpoint = urls.match(path_info=url)[0]
view_func = self.endpoint_dict[endpoint](request)
if isinstance(view_func, str):
return Response(view_func)
else:
return view_func
上一篇:动手写一个简单的Web框架(模板渲染)


下一篇:用Gson解析没有数据头的纯数组json字符串