FastAPI 学习之路(二十八)获取当前用户

2021-08-25 10:14:24 浏览数 (1)

前言

我们之前分享安全校验,这次我们去分享如何获取当前用户

正文

在上一章节中,安全性向路径操作函数提供了一个 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,一起牛转钱坤,一起牛逼。

如果觉得这篇文章还不错,来个【分享、点赞、在看】三连吧,让更多的人也看到~

0 人点赞