Python Flask 自定义返回实体

一、定义返回结果封装函数

def response(code=200, message='', data=None):
    """
    自定义返回结果的封装函数
    :param code: 状态码，默认为 200
    :param message: 提示信息，默认为空字符串
    :param data: 返回数据，默认为 None
    :return: Response 对象
    """
    response_data = {
        'code': code,
        'message': message,
        'data': data
    }
    return jsonify(response_data)

二、在接口处调用并返回

@app.route('/test', methods=['GET'])
def test():
    if request.method == 'GET':
        try:
            result_data = func_main()
            return response(data=result_data, message='模型调用成功！')
        except Exception as e:
            return jsonify({"error": str(e)}), 500