Update code in endpoints/user.post.py
This commit is contained in:
parent
cc9fb0bc8d
commit
14ff14fe0d
83
endpoints/user.post.py
Normal file
83
endpoints/user.post.py
Normal file
@ -0,0 +1,83 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from core.database import fake_users_db
|
||||
from typing import List
|
||||
from pydantic import BaseModel
|
||||
import uuid
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
class UserBase(BaseModel):
|
||||
username: str
|
||||
email: str
|
||||
|
||||
class UserCreate(UserBase):
|
||||
password: str
|
||||
|
||||
class User(UserBase):
|
||||
id: str
|
||||
disabled: bool = False
|
||||
|
||||
@router.post("/users", response_model=User)
|
||||
async def create_user(user: UserCreate, db: Session = Depends(get_db)):
|
||||
"""Create a new user"""
|
||||
if user.username in fake_users_db:
|
||||
raise HTTPException(status_code=400, detail="Username already exists")
|
||||
|
||||
user_id = str(uuid.uuid4())
|
||||
new_user = User(
|
||||
id=user_id,
|
||||
username=user.username,
|
||||
email=user.email,
|
||||
disabled=False
|
||||
)
|
||||
fake_users_db[user_id] = new_user
|
||||
|
||||
return {
|
||||
"message": "User created successfully",
|
||||
"data": new_user
|
||||
}
|
||||
|
||||
@router.get("/users", response_model=List[User])
|
||||
async def read_users(db: Session = Depends(get_db)):
|
||||
"""Get all users"""
|
||||
return list(fake_users_db.values())
|
||||
|
||||
@router.get("/users/{user_id}", response_model=User)
|
||||
async def read_user(user_id: str, db: Session = Depends(get_db)):
|
||||
"""Get a specific user by ID"""
|
||||
user = fake_users_db.get(user_id)
|
||||
if not user:
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
return {
|
||||
"message": "User found",
|
||||
"data": user
|
||||
}
|
||||
|
||||
@router.put("/users/{user_id}", response_model=User)
|
||||
async def update_user(user_id: str, user: UserCreate, db: Session = Depends(get_db)):
|
||||
"""Update a user"""
|
||||
stored_user = fake_users_db.get(user_id)
|
||||
if not stored_user:
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
|
||||
updated_user = User(
|
||||
id=user_id,
|
||||
username=user.username,
|
||||
email=user.email,
|
||||
disabled=stored_user.disabled
|
||||
)
|
||||
fake_users_db[user_id] = updated_user
|
||||
|
||||
return {
|
||||
"message": "User updated successfully",
|
||||
"data": updated_user
|
||||
}
|
||||
|
||||
@router.delete("/users/{user_id}")
|
||||
async def delete_user(user_id: str, db: Session = Depends(get_db)):
|
||||
"""Delete a user"""
|
||||
if user_id not in fake_users_db:
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
|
||||
del fake_users_db[user_id]
|
||||
return {"message": "User deleted successfully"}
|
Loading…
x
Reference in New Issue
Block a user