__init__.py

from typing import Union

from flask import Blueprint, Flask
from flask_restful import Api, output_json
from pydantic import BaseModel


def _pydantic_safe_output_json(data: Union[BaseModel, dict, list], code, headers=None):
    if isinstance(data, BaseModel):
        data = data.dict()

    return output_json(data, code, headers)


class _Api(Api):
    def __init__(self, *args, **kwargs):
        super(_Api, self).__init__(*args, **kwargs)

        self.representations = {'application/json': _pydantic_safe_output_json}


def route(flask_app: Flask):
    from app.views.sample.api import SampleAPI

    handle_exception_func = flask_app.handle_exception
    handle_user_exception_func = flask_app.handle_user_exception
    # register_blueprint 시 defer되었던 함수들이 호출되며, flask-restful.Api._init_app()이 호출되는데
    # 해당 메소드가 test_app 객체의 에러 핸들러를 오버라이딩해서, 별도로 적용한 handler의 HTTPException 관련 로직이 동작하지 않음
    # 따라서 두 함수를 임시 저장해 두고, register_blueprint 이후 함수를 재할당하도록 함

    # - blueprint, api object initialize
    api_blueprint = Blueprint("api", __name__)
    api = _Api(api_blueprint)

    # - route
    api.add_resource(SampleAPI, "/sample")

    # - register blueprint
    flask_app.register_blueprint(api_blueprint)

    flask_app.handle_exception = handle_exception_func
    flask_app.handle_user_exception = handle_user_exception_func