如何在 Tornado 中实现 Middleware

最新推荐文章于 2024-11-16 23:52:49 发布

转载最新推荐文章于 2024-11-16 23:52:49 发布 · 326 阅读

2 ·

CC 4.0 BY-SA版权

原文链接：https://segmentfault.com/a/1190000004351569

本文介绍如何在Tornado Web框架中实现和使用中间件来增强应用程序的功能，包括进行登录检查等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

定义 Middleware

    class MiddleWare(object):
        def process_request(self, request):
            #request是一个RequestHandler的实例
            pass
    
        def process_response(self, request):
            pass

实现登录检查 Middleware


class AuthorizationException(Exception):
    def __init__(self, msg):
        super(AuthorizationException, self).__init__(403, msg)


class CheckLogin(MiddleWare):

    def is_login(self, requesthandler):
        return True
        
    def process_request(self, request):
        if must_login(request):
            if not self.is_login(request):
                raise AuthenticationException(403, "login required")

    def is_login(self, request):
        return request.get_session() != None

找地方注册 Middleware

可以在 Tornado 初始化的时候将Middleware注册到 Application 中

调用 Middlware

class MiddleRequestHandler(RequestHandler):
    def prepare(self):
        
        for middleware in self.app.middleware:
            middleware.process_request(self)
        
    def finish(self):
        for middleware in self.app.middleware:
            middleware.process_response(self)

        super(MiddleRequestHandler, self).finish()