28 lines
1.1 KiB

11 months ago
  1. from fastapi import APIRouter,Depends
  2. from fastapi.security import OAuth2PasswordRequestForm
  3. from datetime import timedelta
  4. from fastapi.security import OAuth2PasswordRequestForm
  5. from fastapi import Depends, FastAPI, HTTPException, status
  6. from dependencies import *
  7. from internal.models import Token
  8. from fastapi.middleware.cors import CORSMiddleware
  9. router=APIRouter(
  10. tags=["用户管理"]
  11. )
  12. # 用户登录
  13. @router.post("/token", response_model=Token)
  14. async def login_for_access_token(
  15. form_data: OAuth2PasswordRequestForm = Depends(),
  16. ) -> Token:
  17. user = authenticate_user(form_data.username, form_data.password)
  18. if not user:
  19. raise HTTPException(
  20. status_code=status.HTTP_401_UNAUTHORIZED,
  21. detail="Incorrect username or password",
  22. headers={"WWW-Authenticate": "Bearer"},
  23. )
  24. access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
  25. access_token = create_access_token(
  26. data={"sub": user.username}, expires_delta=access_token_expires
  27. )
  28. return {"access_token": access_token, "token_type": "bearer"}