48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from ...core.security import get_current_user
|
|
from ...schemas.user import User
|
|
from Backend.services.store_service import StoreService
|
|
from ...services.room_service import RoomService
|
|
|
|
router = APIRouter(prefix="/store", tags=["store"])
|
|
store_service = StoreService()
|
|
room_service = RoomService()
|
|
|
|
|
|
@router.get("")
|
|
async def get_dotory(
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
try:
|
|
dotory = await store_service.get_dotory_by_id(current_user.id)
|
|
return {"dotory": dotory}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/{product_id}")
|
|
async def buy_product(
|
|
product_id: int,
|
|
product_name: str,
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
try:
|
|
response = await store_service.buy_product(product_id, current_user.id)
|
|
if response["isSuccess"]:
|
|
await room_service.add_furniture(current_user.id, product_name)
|
|
return {"dotory": response["dotory"]}
|
|
raise HTTPException(status_code=404, detail="Product not found")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.put("")
|
|
async def update_dotory(
|
|
dotory_num: int,
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
try:
|
|
dotory = await store_service.update_user_dotory(current_user.id, dotory_num)
|
|
return {"dotory": dotory}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|