AIOHTTP中非法请求方法的优雅处理及请求方法限制方案咨询

我懂你碰到的麻烦了——当收到非法HTTP方法的请求时，aiohttp直接抛出一堆回溯信息，而不是返回优雅的错误响应，之前用CORS中间件的路子也没走通。这是因为CORS中间件只处理已经解析成功的合法HTTP请求，但非法方法的请求在HTTP解析阶段就触发了BadStatusLine异常，根本没走到中间件的处理流程里。

下面给你一套完整的解决方案，既能搞定回溯问题，又能严格限制只允许POST请求：

1. 捕获解析阶段异常，避免回溯并返回优雅响应 #

首先得处理BadStatusLine这类解析阶段的异常，我们可以给aiohttp应用注册全局异常处理器，把报错转换成标准的400响应：

from aiohttp import web
from aiohttp.http_exceptions import BadStatusLine

async def handle_invalid_method_error(request, exc):
    return web.Response(
        status=400,
        text="Invalid HTTP request method",
        content_type="text/plain"
    )

然后把这个处理器绑定到对应的异常上：

app.exception(BadStatusLine)(handle_invalid_method_error)

2. 中间件拦截合法但非POST的请求 #

对于那些解析成功但不是POST的合法请求（比如GET、PUT），我们需要返回405 Method Not Allowed，并通过Allow头明确告知允许的方法：

@web.middleware
async def restrict_to_post_middleware(request, handler):
    if request.method != "POST":
        return web.Response(
            status=405,
            text="Only POST requests are allowed",
            headers={"Allow": "POST"}
        )
    return await handler(request)

把这个中间件加入应用的中间件列表，确保所有合法请求都先经过这个检查。

完整可运行代码 #

把上面的逻辑整合起来，完整代码如下：

from aiohttp import web
from aiohttp.http_exceptions import BadStatusLine

async def handle_invalid_method_error(request, exc):
    return web.Response(
        status=400,
        text="Invalid HTTP request method",
        content_type="text/plain"
    )

@web.middleware
async def restrict_to_post_middleware(request, handler):
    if request.method != "POST":
        return web.Response(
            status=405,
            text="Only POST requests are allowed",
            headers={"Allow": "POST"}
        )
    return await handler(request)

async def handle_post_request(request):
    # 这里写你的POST请求处理逻辑
    return web.Response(text="Success: POST request processed")

app = web.Application(middlewares=[restrict_to_post_middleware])
# 注册异常处理器
app.exception(BadStatusLine)(handle_invalid_method_error)
# 添加POST专属路由
app.add_routes([web.post('/', handle_post_request)])

if __name__ == '__main__':
    web.run_app(app)

测试验证 #

现在三种情况都能被正确处理：

• 执行curl -X incorrect_method http://localhost:8080：收到400响应，内容为Invalid HTTP request method，无回溯信息
• 执行curl http://localhost:8080（GET方法）：收到405响应，提示Only POST requests are allowed
• 执行curl -X POST http://localhost:8080：收到成功响应，正常处理请求

内容的提问来源于stack exchange，提问作者user3447228