fastapi \u5982\u4f55\u5728\u4e2d\u95f4\u4ef6\u4e2d\u629b\u51fa\u5f02\u5e38\uff1f<\/p>\n
chatGPT \u7ed9\u6211\u7684\u65b9\u6848\u662f\uff1a \u4f46\u662f\u5b9e\u9645\u6ca1\u7528\uff0c\u5ba2\u6237\u7aef\u6536\u5230\u7684\u8fd8\u662f 500<\/p>\n \u5b9e\u9645\u5e94\u8be5 return \u4e00\u4e2a response<\/p>\n \u53c2\u8003\u4ee3\u7801\uff1a<\/p>\nraise HTTPException(status_code=401, detail="Invalid token")<\/code><\/p>\n@app.middleware("http")\nasync def get_user_id(request: Request, call_next):\n from core.mysql.models import SessionTable\n from core.mysql.models import SessionTable\n from core.mysql.models import UserTable\n from apps.user.schemas import User\n from playhouse.shortcuts import model_to_dict\n\n session_key = request.cookies.get("session_key")\n if session_key:\n try:\n session: SessionTable | None = SessionTable.get_or_none(\n session_key=session_key)\n if session:\n user_id = session.user_id\n if user_id:\n request.state.user_id = user_id\n user: UserTable | None = UserTable.get_or_none(id=user_id)\n request.state.user = User(**model_to_dict(user))\n except Exception as error:\n logger.warning(error)\n if not session_key or not request.state.get('user_id', None):\n return Response(status_code=401, content='Invalid token')\n response = call_next(request)\n return response<\/code><\/pre>\n