verify_token_route.py

from fastapi.routing import APIRoute
from fastapi import Request
from functions_jwt import validate_token


class VerifyTokenRoute(APIRoute):
    def get_route_handler(self):
        original_route_handler = super().get_route_handler()

        async def verify_token_middleware(request: Request):
            token = request.headers.get("Authorization", None)
            if token:
                token = token.split(" ")[1]

            token = validate_token(token)
            if token:
                return token

            return await original_route_handler(request)

        return verify_token_middleware