Update code in endpoints\user.post.py

endpoints/user.post.py

@@ -4,69 +4,37 @@ import uuid
 
 router = APIRouter()
 
-@router.get("/users/{id}")
-async def get_user_handler(
-    id: str = Path(..., description="User ID"),
-    db: Session = Depends(get_db)
-):
-    """Get user by ID"""
-    user = fake_users_db.get(id)
+@router.get("/users/{user_id}")
+async def get_user(user_id: str, db: Session = Depends(get_db)):
+    """Get user profile"""
+    user = db.query(User).filter(User.id == user_id).first()
     if not user:
         raise HTTPException(status_code=404, detail="User not found")
-    return {
-        "message": "User found",
-        "data": user,
-        "metadata": {
-            "source": "demo_db",
-            "result_count": 1
-        }
-    }
+    return {"message": "User found", "data": user}
 
-@router.put("/users/{id}")
-async def update_user_handler(
-    id: str = Path(..., description="User ID"),
-    username: str = Body(..., description="Updated username"),
-    email: str = Body(..., description="Updated email"),
-    password: str = Body(..., description="Updated password"),
-    db: Session = Depends(get_db)
-):
-    """Update user by ID"""
-    user = fake_users_db.get(id)
+@router.put("/users/{user_id}")
+async def update_user(user_id: str, user_data: UserUpdate, db: Session = Depends(get_db), token: str = Depends(oauth2_scheme)):
+    """Update user profile"""
+    user = db.query(User).filter(User.id == user_id).first()
     if not user:
         raise HTTPException(status_code=404, detail="User not found")
-    user["username"] = username
-    user["email"] = email
-    user["password"] = password
-    return {
-        "message": "User updated successfully",
-        "data": user
-    }
+    for field, value in user_data:
+        setattr(user, field, value)
+    db.commit()
+    return {"message": "User updated successfully"}
 
-@router.delete("/users/{id}")
-async def delete_user_handler(
-    id: str = Path(..., description="User ID"),
-    db: Session = Depends(get_db)
-):
-    """Delete user by ID"""
-    user = fake_users_db.pop(id, None)
+@router.delete("/users/{user_id}")
+async def delete_user(user_id: str, db: Session = Depends(get_db), token: str = Depends(oauth2_scheme)):
+    """Delete user"""
+    user = db.query(User).filter(User.id == user_id).first()
     if not user:
         raise HTTPException(status_code=404, detail="User not found")
-    return {
-        "message": "User deleted successfully",
-        "data": user
-    }
+    db.delete(user)
+    db.commit()
+    return {"message": "User deleted successfully"}
 
-@router.get("/users")
-async def get_all_users_handler(
-    db: Session = Depends(get_db)
-):
+@router.get("/users/")
+async def get_users(db: Session = Depends(get_db)):
     """Get all users"""
-    users = list(fake_users_db.values())
-    return {
-        "message": "Users retrieved successfully",
-        "data": users,
-        "metadata": {
-            "source": "demo_db",
-            "result_count": len(users)
-        }
-    }
+    users = db.query(User).all()
+    return {"message": "Users found", "data": users}