Initial commit: SIBU 2.0 MISSION
This commit is contained in:
109
backend/app/api/favorites/__init__.py
Normal file
109
backend/app/api/favorites/__init__.py
Normal file
@ -0,0 +1,109 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from sqlmodel import Session, select
|
||||
from typing import List
|
||||
from app.core.database import get_session
|
||||
from app.models.favorite import Favorite
|
||||
from app.api.deps import get_current_user
|
||||
from app.models.user import User
|
||||
from pydantic import BaseModel
|
||||
|
||||
router = APIRouter(prefix="/api/favorites", tags=["favorites"])
|
||||
|
||||
|
||||
class FavoriteCreate(BaseModel):
|
||||
item_type: str # 'coupon', 'business', 'taxi', 'route'
|
||||
item_id: str
|
||||
item_name: str | None = None
|
||||
item_image: str | None = None
|
||||
|
||||
|
||||
@router.get("")
|
||||
async def get_favorites(
|
||||
item_type: str | None = None,
|
||||
session: Session = Depends(get_session),
|
||||
current_user: User = Depends(get_current_user)
|
||||
) -> List[Favorite]:
|
||||
"""Get all favorites for the current user, optionally filtered by type."""
|
||||
statement = select(Favorite).where(Favorite.user_id == current_user.id)
|
||||
|
||||
if item_type:
|
||||
statement = statement.where(Favorite.item_type == item_type)
|
||||
|
||||
statement = statement.order_by(Favorite.created_at.desc())
|
||||
favorites = session.exec(statement).all()
|
||||
return list(favorites)
|
||||
|
||||
|
||||
@router.post("")
|
||||
async def add_favorite(
|
||||
favorite_data: FavoriteCreate,
|
||||
session: Session = Depends(get_session),
|
||||
current_user: User = Depends(get_current_user)
|
||||
) -> Favorite:
|
||||
"""Add an item to favorites."""
|
||||
# Check if already favorited
|
||||
existing = session.exec(
|
||||
select(Favorite).where(
|
||||
Favorite.user_id == current_user.id,
|
||||
Favorite.item_type == favorite_data.item_type,
|
||||
Favorite.item_id == favorite_data.item_id
|
||||
)
|
||||
).first()
|
||||
|
||||
if existing:
|
||||
raise HTTPException(status_code=400, detail="Item already in favorites")
|
||||
|
||||
favorite = Favorite(
|
||||
user_id=current_user.id,
|
||||
item_type=favorite_data.item_type,
|
||||
item_id=favorite_data.item_id,
|
||||
item_name=favorite_data.item_name,
|
||||
item_image=favorite_data.item_image
|
||||
)
|
||||
session.add(favorite)
|
||||
session.commit()
|
||||
session.refresh(favorite)
|
||||
return favorite
|
||||
|
||||
|
||||
@router.delete("/{item_type}/{item_id}")
|
||||
async def remove_favorite(
|
||||
item_type: str,
|
||||
item_id: str,
|
||||
session: Session = Depends(get_session),
|
||||
current_user: User = Depends(get_current_user)
|
||||
):
|
||||
"""Remove an item from favorites."""
|
||||
favorite = session.exec(
|
||||
select(Favorite).where(
|
||||
Favorite.user_id == current_user.id,
|
||||
Favorite.item_type == item_type,
|
||||
Favorite.item_id == item_id
|
||||
)
|
||||
).first()
|
||||
|
||||
if not favorite:
|
||||
raise HTTPException(status_code=404, detail="Favorite not found")
|
||||
|
||||
session.delete(favorite)
|
||||
session.commit()
|
||||
return {"ok": True}
|
||||
|
||||
|
||||
@router.get("/check/{item_type}/{item_id}")
|
||||
async def check_favorite(
|
||||
item_type: str,
|
||||
item_id: str,
|
||||
session: Session = Depends(get_session),
|
||||
current_user: User = Depends(get_current_user)
|
||||
) -> dict:
|
||||
"""Check if an item is favorited."""
|
||||
favorite = session.exec(
|
||||
select(Favorite).where(
|
||||
Favorite.user_id == current_user.id,
|
||||
Favorite.item_type == item_type,
|
||||
Favorite.item_id == item_id
|
||||
)
|
||||
).first()
|
||||
|
||||
return {"is_favorite": favorite is not None}
|
||||
Reference in New Issue
Block a user