mirror of
https://github.com/mealie-recipes/mealie.git
synced 2026-02-10 09:53:14 -05:00
ea7c4771ee
* add default assets for user profile * add recipe avatar * change user_id to UUID * add profile image upload * setup image cache keys * cleanup tests and add image tests * purge user data on delete * new user repository tests * add user_id validator for int -> UUID conversion * delete depreciated route * force set content type * refactor tests to use temp directory * validate parent exists before createing * set user_id to correct type * update instruction id * reset primary key on migration
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import shutil
|
|
from pathlib import Path
|
|
|
|
from fastapi import Depends, File, HTTPException, UploadFile, status
|
|
from fastapi.routing import APIRouter
|
|
from pydantic import UUID4
|
|
from sqlalchemy.orm.session import Session
|
|
|
|
from mealie import utils
|
|
from mealie.core.dependencies import get_current_user
|
|
from mealie.core.dependencies.dependencies import temporary_dir
|
|
from mealie.db.database import get_database
|
|
from mealie.db.db_setup import generate_session
|
|
from mealie.routes.routers import UserAPIRouter
|
|
from mealie.routes.users._helpers import assert_user_change_allowed
|
|
from mealie.schema.user import PrivateUser
|
|
from mealie.services.image import minify
|
|
|
|
public_router = APIRouter(prefix="", tags=["Users: Images"])
|
|
user_router = UserAPIRouter(prefix="", tags=["Users: Images"])
|
|
|
|
|
|
@user_router.post("/{id}/image")
|
|
def update_user_image(
|
|
id: UUID4,
|
|
profile: UploadFile = File(...),
|
|
temp_dir: Path = Depends(temporary_dir),
|
|
current_user: PrivateUser = Depends(get_current_user),
|
|
session: Session = Depends(generate_session),
|
|
):
|
|
"""Updates a User Image"""
|
|
assert_user_change_allowed(id, current_user)
|
|
|
|
temp_img = temp_dir.joinpath(profile.filename)
|
|
|
|
with temp_img.open("wb") as buffer:
|
|
shutil.copyfileobj(profile.file, buffer)
|
|
|
|
image = minify.to_webp(temp_img)
|
|
dest = PrivateUser.get_directory(id) / "profile.webp"
|
|
|
|
shutil.copyfile(image, dest)
|
|
|
|
db = get_database(session)
|
|
|
|
db.users.patch(id, {"cache_key": utils.new_cache_key()})
|
|
|
|
if not dest.is_file:
|
|
raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR)
|