Flask 에서 로그아웃 API 코드 작성하는 방법

Flask 에서 로그아웃 API 코드 작성하는 방법

 

Flask 에서 로그아웃 코드를 작성할 때는 아래처럼 코드를 작성하여준다.

jwt_blacklist = set()

class UserLogoutResource(Resource) :
    @jwt_required()
    def post(self) :
    
        jti = get_jwt()['jti']
        print(jti)
        jwt_blacklist.add(jti)

        return {'result' : 'success'}, 200

 

메인 app.py 에서는 로그아웃된 토큰으로 요청하는 경우 처리하는 코드를 작성해주어야 한다.

app = Flask(__name__)
# 환경변수 셋팅
app.config.from_object(Config)

# JWT 매니저 초기화
jwt = JWTManager(app)

# 로그아웃된 토큰으로 요청하는 경우 처리하는 코드작성
@jwt.token_in_blocklist_loader
def check_if_token_is_revoked(jwt_header, jwt_payload) :
    jti = jwt_payload['jti']
    return jti in jwt_blacklist