mirror of
https://github.com/mealie-recipes/mealie.git
synced 2026-02-19 14:23:12 -05:00
c4540f1395
* add dependency injection for get_repositories * convert events api to controller * update generic typing * add abstract controllers * update test naming * migrate admin services to controllers * add additional admin route tests * remove print * add public shared dependencies * add types * fix typo * add static variables for recipe json keys * add coverage gutters config * update controller routers * add generic success response * add category/tag/tool tests * add token refresh test * add coverage utilities * covert comments to controller * add todo * add helper properties * delete old service * update test notes * add unit test for pretty_stats * remove dead code from post_webhooks * update group routes to use controllers * add additional group test coverage * abstract common permission checks * convert ingredient parser to controller * update recipe crud to use controller * remove dead-code * add class lifespan tracker for debugging * convert bulk export to controller * migrate tools router to controller * update recipe share to controller * move customer router to _base * ignore prints in flake8 * convert units and foods to new controllers * migrate user routes to controllers * centralize error handling * fix invalid ref * reorder fields * update routers to share common handling * update tests * remove prints * fix cookbooks delete * fix cookbook get * add controller for mealplanner * cover report routes to controller * remove __future__ imports * remove dead code * remove all base_http children and remove dead code
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
from shutil import copyfileobj
|
|
|
|
from fastapi import Depends, File, Form, HTTPException, status
|
|
from fastapi.datastructures import UploadFile
|
|
from pydantic import BaseModel
|
|
from slugify import slugify
|
|
from sqlalchemy.orm.session import Session
|
|
|
|
from mealie.db.db_setup import generate_session
|
|
from mealie.repos.all_repositories import get_repositories
|
|
from mealie.routes._base.routers import UserAPIRouter
|
|
from mealie.schema.recipe import CreateRecipeByUrl, Recipe, RecipeAsset
|
|
from mealie.services.image.image import scrape_image, write_image
|
|
|
|
router = UserAPIRouter()
|
|
|
|
|
|
class UpdateImageResponse(BaseModel):
|
|
image: str
|
|
|
|
|
|
@router.post("/{slug}/image")
|
|
def scrape_image_url(slug: str, url: CreateRecipeByUrl):
|
|
"""Removes an existing image and replaces it with the incoming file."""
|
|
scrape_image(url.url, slug)
|
|
|
|
|
|
@router.put("/{slug}/image", response_model=UpdateImageResponse)
|
|
def update_recipe_image(
|
|
slug: str,
|
|
image: bytes = File(...),
|
|
extension: str = Form(...),
|
|
session: Session = Depends(generate_session),
|
|
):
|
|
"""Removes an existing image and replaces it with the incoming file."""
|
|
db = get_repositories(session)
|
|
write_image(slug, image, extension)
|
|
new_version = db.recipes.update_image(slug, extension)
|
|
|
|
return UpdateImageResponse(image=new_version)
|
|
|
|
|
|
@router.post("/{slug}/assets", response_model=RecipeAsset)
|
|
def upload_recipe_asset(
|
|
slug: str,
|
|
name: str = Form(...),
|
|
icon: str = Form(...),
|
|
extension: str = Form(...),
|
|
file: UploadFile = File(...),
|
|
session: Session = Depends(generate_session),
|
|
):
|
|
"""Upload a file to store as a recipe asset"""
|
|
file_name = slugify(name) + "." + extension
|
|
asset_in = RecipeAsset(name=name, icon=icon, file_name=file_name)
|
|
dest = Recipe(slug=slug).asset_dir.joinpath(file_name)
|
|
|
|
with dest.open("wb") as buffer:
|
|
copyfileobj(file.file, buffer)
|
|
|
|
if not dest.is_file():
|
|
raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR)
|
|
|
|
db = get_repositories(session)
|
|
|
|
recipe: Recipe = db.recipes.get(slug)
|
|
recipe.assets.append(asset_in)
|
|
db.recipes.update(slug, recipe.dict())
|
|
return asset_in
|