Исключение Flask restplus не перехватывается api.errorhandler

Я хочу определить пользовательскую обработку ошибок для Flask-restful API

Пользовательская ошибка фляги (FlaskGenericException) возникает, но не обрабатывается обработчиком ошибок restplus. он просто выдает ошибку 500 Internal Server.

Не уверен, почему dict, определенный внутри exceptions.py, не выдается как ответ API.

Ожидаемый результат

{'user': 'Lionel Messi', 'company': 'Soccer Brazil','message': 'A generic error'}

Вот мои файлы, которые я использую для проверки:

app.py

import json,os
import logging
import sys
import requests
import config
from os import environ
from flask import Flask, request, jsonify, Response
from flask_jwt_extended import (JWTManager, jwt_required)
from flask_restplus import Api, Resource, reqparse, fields, inputs
from cal_Test import calTest
from exceptions import FlaskGenericException




# Setup Flask Server
app = Flask(__name__,root_path=os.path.join(os.getcwd(), './'))


app.config.from_object(config.Config)
api = Api(app=app,
          version='0.1',
          title="Test Cal API",
          doc="/swagger/",
          description='TestCal = Cal Test')
ns = api.namespace('/', description='Cal Test Api')

@api.errorhandler(FlaskGenericException)
def handle_error(error):
    response = jsonify(error.to_dict())
    response.status_code = error.status_code
    return response


resource_parser = reqparse.RequestParser()
resource_parser.add_argument('test', type=str, required=True)


resource_flds = api.model('Resource', {
    'test': fields.String(required = True),
})

@ns.route('/test', methods=['POST'])
class TestCal(Resource):


    # @jwt_required
    @ns.doc(body=resource_flds)
    @ns.expect(resource_flds, validate=True)
    def post(self):
        request_data = resource_parser.parse_args()
        caltst = calTest()
        result = caltst.cal_test(request_data)
        response = app.response_class(
        response=json.dumps(result),
        status=200,
        mimetype='application/json'
    )
        return response


if __name__ == "__main__":
    app.run(debug=True)

cal_Test.py

import pandas as pd
import pyodbc
import scipy.stats as st
import numpy as np
import pandas as pd
from scipy.stats import norm
from flask import Flask, request, jsonify
import os
from exceptions import FlaskGenericException




class calTest:

def __init__(self):
    self.test_name = 'Hi! I am test'



def cal_test(self,reqdata):
    if self.test_name == reqdata['test']:
        return "Successfull"
    else:
        raise FlaskGenericException('A generic error', status_code=404, payload={'user': 'Lionel Messi', 'company': 'Soccer Brazil'})

исключения.py

class FlaskGenericException(Exception):
    status_code = 500   # default unless overridden
    def __init__(self, message, status_code=None, payload=None):
        Exception.__init__(self)
        self.message = message
        if status_code is not None:
            self.status_code = status_code
        self.payload = payload

    def to_dict(self):
        rv = dict(self.payload or ())
        rv['message'] = self.message
        return rv

person min2bro    schedule 18.05.2020    source источник
comment
Это существующая ошибка в flask-restplus: github.com/noirbizarre/flask-restplus/ issue/744 и будет исправлено в версии 0.14.0. Даты выхода пока нет.   -  person min2bro    schedule 18.05.2020


Ответы (1)


Столкнувшись с той же проблемой, я использовал метод abort из Flask. Я не знаю, подходит ли это для вашего варианта использования, но попробуйте.

from flask import abort 

def cal_test(self,reqdata):
    if self.test_name == reqdata['test']:
        return "Successfull"
    else:
        abort(404, "A generic error")
person Yankee Maharjan    schedule 29.05.2020