90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
from fastapi import APIRouter, HTTPException, Depends, status
|
|
from fastapi.security import OAuth2PasswordRequestForm
|
|
from ..models import User, UserCreate, UserInDB, Token
|
|
from ..database import get_db
|
|
from ..auth import get_current_user, get_password_hash, verify_password, create_access_token, oauth2_scheme, create_user, authenticate_user
|
|
from bson import ObjectId
|
|
from typing import List
|
|
from datetime import timedelta
|
|
|
|
router = APIRouter(
|
|
prefix="/users",
|
|
tags=["users"]
|
|
)
|
|
|
|
|
|
@router.post("/register", response_model=User)
|
|
async def register_user(user: UserCreate):
|
|
db = get_db()
|
|
existing_user = db.users.find_one({"username": user.username})
|
|
if existing_user:
|
|
raise HTTPException(status_code=400, detail="Username already registered")
|
|
hashed_password = get_password_hash(user.password)
|
|
user_in_db = UserInDB(
|
|
**user.model_dump(exclude={"password"}), hashed_password=hashed_password)
|
|
created_user = create_user(db, user_in_db)
|
|
return User(**created_user.model_dump(exclude={"hashed_password"}))
|
|
|
|
|
|
def get_user(username: str):
|
|
db = get_db()
|
|
user_dict = db.users.find_one({"username": username})
|
|
if user_dict:
|
|
return UserInDB(**user_dict)
|
|
|
|
|
|
@router.post("/token", response_model=Token)
|
|
async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends()):
|
|
db = get_db()
|
|
user = authenticate_user(db, form_data.username, form_data.password)
|
|
if not user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="Incorrect username or password",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
access_token_expires = timedelta(minutes=30) # You can adjust this value
|
|
access_token = create_access_token(
|
|
data={"sub": user.username}, expires_delta=access_token_expires
|
|
)
|
|
return {"access_token": access_token, "token_type": "bearer"}
|
|
|
|
|
|
@router.get("/me", response_model=User)
|
|
async def read_users_me(current_user: User = Depends(get_current_user)):
|
|
return current_user
|
|
|
|
|
|
@router.get("/", response_model=List[User])
|
|
async def read_users(skip: int = 0, limit: int = 10, current_user: User = Depends(get_current_user)):
|
|
db = get_db()
|
|
users = list(db.users.find().skip(skip).limit(limit))
|
|
return [User(**user) for user in users]
|
|
|
|
|
|
@router.put("/{user_id}", response_model=User)
|
|
async def update_user(user_id: str, user: UserInDB, current_user: User = Depends(get_current_user)):
|
|
db = get_db()
|
|
existing_user = db.users.find_one({"_id": ObjectId(user_id)})
|
|
if not existing_user:
|
|
raise HTTPException(status_code=404, detail="User not found")
|
|
update_data = user.dict(exclude_unset=True)
|
|
if "password" in update_data:
|
|
update_data["hashed_password"] = get_password_hash(update_data["password"])
|
|
del update_data["password"]
|
|
result = db.users.update_one(
|
|
{"_id": ObjectId(user_id)}, {"$set": update_data})
|
|
if result.modified_count == 1:
|
|
updated_user = db.users.find_one({"_id": ObjectId(user_id)})
|
|
return User(**updated_user)
|
|
else:
|
|
raise HTTPException(status_code=400, detail="User update failed")
|
|
|
|
|
|
@router.delete("/{user_id}", status_code=status.HTTP_204_NO_CONTENT)
|
|
async def delete_user(user_id: str, current_user: User = Depends(get_current_user)):
|
|
db = get_db()
|
|
result = db.users.delete_one({"_id": ObjectId(user_id)})
|
|
if result.deleted_count != 1:
|
|
raise HTTPException(status_code=404, detail="User not found")
|