FastAPI 学习之路（二十八）获取当前用户

雷子

发布于 2021-08-25 10:14:24

9870

发布于 2021-08-25 10:14:24

文章被收录于专栏：雷子说测试开发雷子说测试开发

前言

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

正文

在上一章节中，安全性向路径操作函数提供了一个 str 类型的 token，让我们来使它返回当前用户给我们。那么我们看看去如何去实现。

我们先去创建一个用户的模型

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 依赖项，来获取当前用户

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

我们看下最后所有结果。

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

我们看下最后的结果