You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

66 lines
2.3 KiB

8 months ago
11 months ago
8 months ago
11 months ago
8 months ago
10 months ago
11 months ago
8 months ago
11 months ago
10 months ago
8 months ago
10 months ago
8 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, HTTPException, status
  6. from dependencies import *
  7. from internal.models import Token
  8. router = APIRouter(
  9. prefix="/users",
  10. tags=["用户管理"]
  11. )
  12. # 登录获取token
  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"}
  29. # 获取用户
  30. @router.get("/me/", response_model=User)
  31. async def read_users_me(current_user: User = Depends(get_current_active_user)):
  32. return current_user
  33. # 注册新用户
  34. @router.post("/register/")
  35. async def register_user(user: UserInDB, _: User = Depends(get_current_active_user)):
  36. # 检查用户名是否已经存在
  37. existing_user = get_user(user.username)
  38. if existing_user:
  39. raise HTTPException(
  40. status_code=400, detail="Username already registered")
  41. if not user.hashed_password:
  42. raise HTTPException(status_code=400, detail="password cannot be empty")
  43. # 创建新用户并保存到数据库
  44. hashed_password = get_password_hash(user.hashed_password)
  45. insert_query = "INSERT INTO users (username, email, full_name, hashed_password, disabled) VALUES (%s, %s, %s, %s, %s)"
  46. user_data = (user.username, user.email, user.full_name,
  47. hashed_password, user.disabled)
  48. execute_query(insert_query, user_data)
  49. # 返回创建的用户信息
  50. return {"status": status.HTTP_200_OK, "message": "users create successfully!"}
  51. @router.get("/me/items/")
  52. async def read_own_items(current_user: User = Depends(get_current_active_user)):
  53. return [{"item_id": "Foo", "owner": current_user.username}]