### 使用Flask中间件实现身份验证 在 Flask 中，可以通过覆盖 `wsgi_app` 方法或者直接替换 `app.wsgi_app` 来实现自定义中间件[^1]。这种中间件可以在请求到达视图函数之前执行特定的逻辑，比如身份验证。 以下是通过中间件实现身份验证的一个完整示例： #### 自定义中间件类 创建一个中间件类，在其中实现身份验证逻辑。该类需要继承自 WSGI 的接口，并提供 `__call__` 方法来拦截请求和响应。 ```python class AuthenticationMiddleware: def __init__(self, app): self.app = app # 原始的 WSGI 应用程序 def __call__(self, environ, start_response): from urllib.parse import urlparse path_info = environ['PATH_INFO'] # 如果路径不需要身份验证，则跳过检查 if path_info == '/login': return self.app(environ, start_response) # 获取请求头中的 Authorization 字段 authorization_header = environ.get('HTTP_AUTHORIZATION') if not authorization_header or not authorization_header.startswith('Bearer '): response_body = b'Unauthorized' status = '401 Unauthorized' headers = [('Content-Type', 'text/plain'), ('Content-Length', str(len(response_body)))] start_response(status, headers) return [response_body] token = authorization_header.split(' ')[1] if token != "valid_token": response_body = b'Invalid Token' status = '403 Forbidden' headers = [('Content-Type', 'text/plain'), ('Content-Length', str(len(response_body)))] start_response(status, headers) return [response_body] # 验证通过，继续处理原始应用 return self.app(environ, start_response) ``` #### 将中间件应用于 Flask 应用 将上述中间件应用到 Flask 应用中，如下所示： ```python from flask import Flask app = Flask(__name__) # 替换默认的 wsgi_app app.wsgi_app = AuthenticationMiddleware(app.wsgi_app) @app.route('/protected') def protected(): return "This is a protected route" @app.route('/login') def login(): return "Login page" ``` 在这个例子中，任何访问 `/protected` 路由的请求都会经过 `AuthenticationMiddleware` 进行身份验证。只有当请求头中包含有效的 Bearer Token 时，才会允许访问受保护的资源[^2]。 #### 结合模板的身份验证场景 如果希望结合前端页面进行身份验证，可以参考以下 HTML 页面设计[^5]： ```html {% extends 'bootstrap/base.html' %} {% block content %} <div class="container"> <div class="page-header"> <h1>Index</h1> </div> <form action="/auth" method="post"> <input type="password" name="token" placeholder="Enter your token"> <button type="submit">Submit</button> </form> </div> {% endblock content %} ``` 在此基础上，服务器端可以进一步扩展逻辑以支持基于表单的身份验证[^3]。 --- ### 总结 通过覆盖 `wsgi_app` 或者直接替换 `app.wsgi_app`，可以轻松地在 Flask 中引入中间件来实现身份验证功能。这种方式不仅灵活而且高效，能够满足大多数应用场景下的需求[^1]。 ---