홍쌈's

fastapi에는 spring의 Interceptor와 같이 api호출 후 전/후처리를 할 수 있는 기능이 있습니다.

middleware를 적용하면 되는데

먼저 middleware/middleware.py를 생성해줍니다

from fastapi import Request
from chatterbot.utils import get_logger
import time

logger = get_logger(__name__)


class Middleware:
    def __init__(
            self,
            some_attribute: str,
    ):
        self.some_attribute = some_attribute

    async def __call__(self, request: Request, call_next):
        logger.debug("")
        logger.debug(" ******************** start ******************** ")
        logger.debug(" ***** request path : {} ".format(request.url.path))

        start_time = time.time()
        response = await call_next(request)
        process_time = time.time() - start_time
        response.headers["X-Process-Time"] = str(process_time)

        logger.debug(" ***** process_time : {} ".format(process_time))
        logger.debug(" ******************** end ******************** ")
        logger.debug("")

        return response

소스코드를 보시면 __call__함수가 있고, 그 안에서 
1. start_time 찍기
2. call_nest(request)호출
3. process_time찍기
순으로 실행을 하는데 이 과정을거쳐서 호출된 요청의 process_time을 찍을 수 있습니다.

await call_nest(reques)로 사용자가 요청하기 전에 전처리를 하고
응답값을 받아온 후에 후처리를 로직을 넣어서 구서을 하시면 됩니다.

이렇게 작성한 middleware를 router에 add해야하는데
main.py 소스에 아래부분을 추가하면 됩니다.

from middleware.middleware import Middleware
from starlette.middleware.base import BaseHTTPMiddleware

app = FastAPI()

middleware = Middleware(some_attribute="some_attribute_here_if_needed")
app.add_middleware(BaseHTTPMiddleware, dispatch=middleware)

파이썬으로 개발을 하다보면 pip를 이용해서 여러가지 모듈을 설치해서 사용을 하게 됩니다.

보통 개발을 하면서 필요한 모듈이 있으면 아래와 같은식으로 다운을 받은뒤
import를 해서 사용을 합니다.

pip install fastapi

하지만 한두개일때는 상관이 없지만 개발을 하다보면 20~30개의 모듈이 필요로 할 수 있기때문에

보통 requirements.txt파일을 이용해서 관리를 하는데 다운받을때마다 매번 추가를 해줄수 없기때문에

freeze를 사용해서 requirements.txt파일을 생성합니다.

가상환경에 설치된 리스트나 가상환경을 사용 안한다면 현재 사용중인 파이썬 버전에 설치된 리스트를
아래 명령어를 사용해서 txt파일을 생성 할 수 있습니다. 

pip freeze > requirements.txt

그리고 다른곳에서 해당 프로젝트를 실행하기 위해선 requirements.txt파일 들어있는 모듈들을 설치해야하는데
일일히 pip install A, pip install B이런식으로 설치를 각각 해줄수 있지만 20~30개 모듈을 하나하나 설치하는것 또한 쉽지 않기 때문에
아래 명령어를 통해 한번에 쭉 설치를 할 수 있습니다.

pip install -r requirementx.txt