项目基础模块代码

app/application/app.py

@@ -8,6 +8,7 @@ from app.api.common.upload_file import upload_files
 from app.api.sys.login_api import login
 from app.api.sys.sys_user_api import user
 from app.api.business.project_api import project
+from app.api.common.view_img import view
 
 my_app = FastAPI()
 
@@ -29,8 +30,10 @@ my_app.add_middleware(
 my_app.add_middleware(LoggerMiddleware)
 my_app.add_middleware(TokenMiddleware)
 
-my_app.include_router(user, prefix="/user", tags=["用户管理API"])
 my_app.include_router(login, prefix="/login", tags=["用户登录接口"])
 my_app.include_router(upload_files, prefix="/upload", tags=["文件上传API"])
+my_app.include_router(view, prefix="/view_img", tags=["查看图片"])
+my_app.include_router(user, prefix="/user", tags=["用户管理API"])
 my_app.include_router(project, prefix="/proj", tags=["项目管理API"])
 
+

app/application/token_middleware.py

@@ -6,7 +6,6 @@ from app.common import reponse_code as rc
 from app.common import jwt_check as jc
 
 
-
 class TokenMiddleware(BaseHTTPMiddleware):
 
     def __init__(self, app):
@@ -21,7 +20,7 @@ class TokenMiddleware(BaseHTTPMiddleware):
         """
         token = request.headers.get('Authorization')
         path = request.url.path
-        if '/login' in path:
+        if check_green(path):
             response = await call_next(request)
             return response
         if not token:
@@ -30,4 +29,15 @@ class TokenMiddleware(BaseHTTPMiddleware):
             jc.check_token(token)
             return await call_next(request)
         except PyJWTError as error:
+            print(error)
             return rc.response_code_view(status.HTTP_401_UNAUTHORIZED, "Token错误或失效，请重新验证")
+
+
+green = ['/login', '/view_img']
+
+
+def check_green(s: str):
+    for url in green:
+        if url in s:
+            return True
+    return False