mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-11-22 11:52:20 -05:00
83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm.session import Session
|
|
|
|
from mealie.core.dependencies import is_logged_in
|
|
from mealie.core.root_logger import get_logger
|
|
from mealie.db.database import get_database
|
|
from mealie.db.db_setup import generate_session
|
|
from mealie.routes.routers import AdminAPIRouter, UserAPIRouter
|
|
from mealie.schema.recipe import CategoryIn, RecipeCategoryResponse
|
|
|
|
public_router = APIRouter()
|
|
user_router = UserAPIRouter()
|
|
admin_router = AdminAPIRouter()
|
|
logger = get_logger()
|
|
|
|
|
|
@public_router.get("")
|
|
async def get_all_recipe_categories(session: Session = Depends(generate_session)):
|
|
"""Returns a list of available categories in the database"""
|
|
db = get_database(session)
|
|
return db.categories.get_all_limit_columns(fields=["slug", "name"])
|
|
|
|
|
|
@public_router.get("/empty")
|
|
def get_empty_categories(session: Session = Depends(generate_session)):
|
|
"""Returns a list of categories that do not contain any recipes"""
|
|
db = get_database(session)
|
|
return db.categories.get_empty()
|
|
|
|
|
|
@public_router.get("/{category}", response_model=RecipeCategoryResponse)
|
|
def get_all_recipes_by_category(
|
|
category: str, session: Session = Depends(generate_session), is_user: bool = Depends(is_logged_in)
|
|
):
|
|
"""Returns a list of recipes associated with the provided category."""
|
|
db = get_database(session)
|
|
|
|
category_obj = db.categories.get(category)
|
|
category_obj = RecipeCategoryResponse.from_orm(category_obj)
|
|
|
|
if not is_user:
|
|
category_obj.recipes = [x for x in category_obj.recipes if x.settings.public]
|
|
|
|
return category_obj
|
|
|
|
|
|
@user_router.post("")
|
|
async def create_recipe_category(category: CategoryIn, session: Session = Depends(generate_session)):
|
|
"""Creates a Category in the database"""
|
|
db = get_database(session)
|
|
|
|
try:
|
|
return db.categories.create(category.dict())
|
|
except Exception:
|
|
raise HTTPException(status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
@admin_router.put("/{category}", response_model=RecipeCategoryResponse)
|
|
async def update_recipe_category(category: str, new_category: CategoryIn, session: Session = Depends(generate_session)):
|
|
"""Updates an existing Tag in the database"""
|
|
db = get_database(session)
|
|
|
|
try:
|
|
return db.categories.update(category, new_category.dict())
|
|
except Exception:
|
|
logger.exception("Failed to update category")
|
|
raise HTTPException(status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
@admin_router.delete("/{category}")
|
|
async def delete_recipe_category(category: str, session: Session = Depends(generate_session)):
|
|
"""
|
|
Removes a recipe category from the database. Deleting a
|
|
category does not impact a recipe. The category will be removed
|
|
from any recipes that contain it
|
|
"""
|
|
db = get_database(session)
|
|
|
|
try:
|
|
db.categories.delete(category)
|
|
except Exception:
|
|
raise HTTPException(status.HTTP_400_BAD_REQUEST)
|