110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
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}
|