Flask单POST路由能否按JSON结构分发处理单个对象和数组请求

结论 #

Flask原生不支持像Rust Rocket框架那样基于请求体JSON结构自动匹配同路径同请求方法的多个路由，你示例中直接给两个同路径/test的POST路由加@app.route装饰器的写法不会生效，Flask只会调用第一个注册的处理函数。

不用拆分不同路由也能实现该需求，有两种常用实现方式： #

方式1：统一入口分发（最简单，无额外依赖） #

直接写一个统一的入口路由，先判断请求体JSON的类型，再转发到对应处理函数即可：

from flask import Flask, request, jsonify

app = Flask(__name__)

# 单个对象处理逻辑
def handle_single_object(data):
    print(data)
    return jsonify(data)

# 对象数组处理逻辑
def handle_array_object(data):
    for item in data:
        print(item)
    return jsonify(data)

# 统一路由入口
@app.route('/test', methods=['POST'])
def test_dispatcher():
    # 注意：Flask的request.json是属性不是方法，request.json()写法会报错
    data = request.get_json(silent=True)
    if data is None:
        return jsonify({"error": "非法JSON格式"}), 400
    
    if isinstance(data, dict):
        return handle_single_object(data)
    elif isinstance(data, list):
        return handle_array_object(data)
    else:
        return jsonify({"error": "仅支持JSON对象或对象数组格式"}), 400

方式2：封装装饰器模拟Rocket风格注册 #

如果要更接近你预期的按处理函数分开注册的写法，可以自行封装注册逻辑，不用修改路由分发规则也能实现类似效果：

from flask import Flask, request, jsonify

app = Flask(__name__)
# 存储同路由的处理函数映射
route_handler_map = {}

def register_json_route(path: str, methods: list, expect_type: type, handler):
    route_key = (path, tuple(sorted(methods)))
    # 首次注册该路由时初始化统一分发入口
    if route_key not in route_handler_map:
        route_handler_map[route_key] = []
        @app.route(path, methods=methods)
        def dispatcher(*args, **kwargs):
            data = request.get_json(silent=True)
            if data is None:
                return jsonify({"error": "非法JSON格式"}), 400
            # 匹配对应类型的处理函数
            for expected_t, func in route_handler_map[route_key]:
                if isinstance(data, expected_t):
                    return func(data, *args, **kwargs)
            return jsonify({"error": "无匹配的处理函数"}), 400
    # 注册对应类型的处理函数
    route_handler_map[route_key].append((expect_type, handler))

# 按你的需求分别注册处理函数即可
def handle_single_object(data):
    print(data)
    return jsonify(data)
register_json_route('/test', ['POST'], dict, handle_single_object)

def handle_array_object(data):
    for item in data:
        print(item)
    return jsonify(data)
register_json_route('/test', ['POST'], list, handle_array_object)

注意事项 #

如果需要更复杂的JSON结构匹配（比如不是只判断根类型，还要判断内部字段），只需要修改类型校验的逻辑即可，无需调整整体分发框架。

内容的提问来源于stack exchange，提问作者Tim B.