From 25d9dce5f1c6b71d18997a12cb17c61ec563d431 Mon Sep 17 00:00:00 2001 From: martsforever Date: Mon, 24 Nov 2025 16:58:55 +0800 Subject: [PATCH] feat: optimize token verification --- app/middlewares/app_middlewares.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/app/middlewares/app_middlewares.py b/app/middlewares/app_middlewares.py index 7c0fc5c..7b84177 100644 --- a/app/middlewares/app_middlewares.py +++ b/app/middlewares/app_middlewares.py @@ -1,6 +1,7 @@ import time import traceback +import jwt from fastapi import FastAPI, HTTPException from passlib.exc import InvalidTokenError from starlette import status @@ -65,6 +66,12 @@ def add_app_middlewares(app: FastAPI): raise unauthorized_exception except InvalidTokenError: raise unauthorized_exception + except jwt.exceptions.InvalidSignatureError: + raise unauthorized_exception + except jwt.exceptions.ExpiredSignatureError: + raise unauthorized_exception + except jwt.exceptions.DecodeError: + raise unauthorized_exception async def get_default_cache_value(): async with async_session() as session: