mirror of
https://github.com/mealie-recipes/mealie.git
synced 2026-02-07 00:13:12 -05:00
84c23765cd
* fix strict optional errors * fix typing in repository * fix backup db files location * update workspace settings
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from pydantic import UUID4
|
|
|
|
from mealie.routes._base import BaseUserController, controller
|
|
from mealie.routes._base.routers import UserAPIRouter
|
|
from mealie.routes.users._helpers import assert_user_change_allowed
|
|
from mealie.schema.user import UserFavorites
|
|
|
|
router = UserAPIRouter()
|
|
|
|
|
|
@controller(router)
|
|
class UserFavoritesController(BaseUserController):
|
|
@router.get("/{id}/favorites", response_model=UserFavorites)
|
|
async def get_favorites(self, id: UUID4):
|
|
"""Get user's favorite recipes"""
|
|
return self.repos.users.get_one(id, override_schema=UserFavorites)
|
|
|
|
@router.post("/{id}/favorites/{slug}")
|
|
def add_favorite(self, id: UUID4, slug: str):
|
|
"""Adds a Recipe to the users favorites"""
|
|
assert_user_change_allowed(id, self.user)
|
|
|
|
if not self.user.favorite_recipes:
|
|
self.user.favorite_recipes = []
|
|
|
|
self.user.favorite_recipes.append(slug)
|
|
self.repos.users.update(self.user.id, self.user)
|
|
|
|
@router.delete("/{id}/favorites/{slug}")
|
|
def remove_favorite(self, id: UUID4, slug: str):
|
|
"""Adds a Recipe to the users favorites"""
|
|
assert_user_change_allowed(id, self.user)
|
|
|
|
if not self.user.favorite_recipes:
|
|
self.user.favorite_recipes = []
|
|
|
|
self.user.favorite_recipes = [x for x in self.user.favorite_recipes if x != slug]
|
|
self.repos.users.update(self.user.id, self.user)
|
|
return
|