修改查看图片的问题

app/application/app.py

@@ -7,7 +7,7 @@ from app.application.logger_middleware import LoggerMiddleware
 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.business.project_train_api import project
 from app.api.common.view_img import view
 
 my_app = FastAPI()
@@ -32,7 +32,7 @@ my_app.add_middleware(TokenMiddleware)
 
 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(view, tags=["查看图片"])
 my_app.include_router(user, prefix="/user", tags=["用户管理API"])
 my_app.include_router(project, prefix="/proj", tags=["项目管理API"])
 

app/application/token_middleware.py

@@ -5,6 +5,15 @@ from jwt import PyJWTError
 from app.common import reponse_code as rc
 from app.common import jwt_check as jc
 
+green = ['/login', '/view_img', 'view_thumb']
+
+
+def check_green(s: str):
+    for url in green:
+        if url in s:
+            return True
+    return False
+
 
 class TokenMiddleware(BaseHTTPMiddleware):
 
@@ -33,11 +42,3 @@ class TokenMiddleware(BaseHTTPMiddleware):
             return rc.response_code_view(status.HTTP_401_UNAUTHORIZED, "Token错误或失效，请重新验证")
 
 
-green = ['/login', '/view_img', 'test']
-
-
-def check_green(s: str):
-    for url in green:
-        if url in s:
-            return True
-    return False