编写一个FastAPI_Login实现用户登录认证代码
时间: 2024-01-31 12:04:08 浏览: 286
用API实现windows用户登录
5星 · 资源好评率100%
好的,我理解了您的要求。以下是实现用户登录认证的FastAPI_Login代码示例:
```python
from fastapi import FastAPI, Depends, HTTPException
from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
from passlib.context import CryptContext
app = FastAPI()
# 用户名和密码为演示目的而编写的示例代码,实际使用需要更加安全的存储方式
fake_users_db = {
"john@example.com": {
"username": "john@example.com",
"full_name": "John Doe",
"email": "john@example.com",
"hashed_password": "$2b$12$i1EOb6hdOo6PFWo7f13UduyuMUUUH1uGd.eLlS5XC5Ws5YiYwarUC",
"disabled": False,
}
}
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
# 获取用户信息的方法
def fake_decode_token(token):
try:
email = token.encode('utf-8').decode('utf-8')
user = fake_users_db[email]
return user
except:
return None
# 使用Authorization头部中的token验证用户
async def get_current_user(token: str = Depends(oauth2_scheme)):
user = fake_decode_token(token)
if not user:
raise HTTPException(status_code=401, detail="Invalid authentication token")
return user
# 认证用户的方法
async def authenticate_user(email: str, password: str):
user = fake_users_db.get(email)
if not user:
return False
if not pwd_context.verify(password, user["hashed_password"]):
return False
return user
# 处理登录请求的路由方法
@app.post("/login")
async def login(form_data: OAuth2PasswordRequestForm = Depends()):
user = await authenticate_user(form_data.username, form_data.password)
if not user:
raise HTTPException(status_code=401, detail="Incorrect email or password")
token = user["email"]
return {"access_token": token, "token_type": "bearer"}
# 使用验证用户的方法获取保护的资源
@app.get("/protected_resource")
async def protected_resource(current_user: dict = Depends(get_current_user)):
return {"user": current_user}
```
这段代码使用了FastAPI框架,实现了一个/login的路由,该路由可以处理接收的用户名和密码,认证用户身份,并返回一个访问令牌。然后,通过使用装饰器的方式,用/authenticate_user方法验证用户身份,并使用/get_current_user方法获取当前用户的信息。在这个认证体系中,我们使用了一个基于bcrypt的哈希密码存储方案。
阅读全文