assert_error_flask.py

assert_error_flask.py

import json
import flask
from decorator import decorator

def assert_error():
    @decorator
    def _fn(f: callable, *args, **kwargs):
        try:
            return f(*args, **kwargs)
        except Exception as exception:
            import traceback
            status_code = 400
            code = 400
            try:
                if hasattr(exception, 'code'):
                    code = getattr(exception, 'code')
                if hasattr(exception, 'status_code'):
                    status_code = getattr(exception, 'status_code')
                    if isinstance(status_code, str) and status_code.isdigit():
                        status_code = int(status_code)
                    if not isinstance(status_code, int):
                        status_code = 400
            except Exception:
                status_code = 400
                code = 400
            resp = dict(
                code=code,
                message=str(exception),
                traceback=traceback.format_exc()
            )
            return resp, status_code
    return _fn