前言
我们之前分享安全校验,这次我们去分享如何获取当前用户
正文
在上一章节中,安全性向路径操作函数提供了一个 str 类型的 token,让我们来使它返回当前用户给我们。那么我们看看去如何去实现。
我们先去创建一个用户的模型
代码语言:javascript复制from pydantic import BaseModel
class User(BaseModel):
username: str
email: Optional[str] = None
full_name: Optional[str] = None
status: Optional[bool] = None
创建一个 get_current_user 依赖项,来获取当前用户
代码语言:javascript复制def fake_decode_token(token):
return User(
username=token , email="leizi@leizi.com", full_name="leizi"
)
def get_current_user(token: str = Depends(oauth2_scheme)):
user = fake_decode_token(token)
return user
我们看下最后所有结果。
代码语言:javascript复制from fastapi import FastAPI,Depends
from pydantic import BaseModel
from typing import Optional
from fastapi.security import OAuth2PasswordBearer
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
app = FastAPI()
class User(BaseModel):
username: str
email: Optional[str] = None
full_name: Optional[str] = None
disabled: Optional[bool] = None
def fake_decode_token(token):
return User(
username=token, email="leizi@leizi.com", full_name="leizi"
)
def get_current_user(token: str = Depends(oauth2_scheme)):
user = fake_decode_token(token)
return user
@app.get("/users/me")
async def read_users_me(current_user: User = Depends(get_current_user)):
return current_user
我们看下最后的结果
现在你可以直接在路径操作函数中获取当前用户。
我们已经进行到一半了。
我们只需要再为用户/客户端添加一个真正发送 username 和 password 的路径操作。
后记
发现问题,解决问题。遇到问题,慢慢解决问题即可。
欢迎关注雷子说测试开发,后续将会持续为大家分享更多的技术知识
如果你有问题可以留言或者加我微信:952943386。
2021,一起牛转钱坤,一起牛逼。
如果觉得这篇文章还不错,来个【分享、点赞、在看】三连吧,让更多的人也看到~