azpsen/tailfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix add_images endpoint

Browse Source
This commit is contained in:
april
2024-01-15 10:48:45 -06:00
parent d6a0eb349a
commit 204a12145d
3 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/routes/img.py
View File

@@ -24,15 +24,12 @@ async def get_image(user: UserDisplaySchema = Depends(get_current_user),
:param image_id: ID of image to retrieve
:return: Stream associated with requested image
"""
stream, user_created = await img.retrieve_image(image_id)
stream, user_created, media_type = await img.retrieve_image(image_id)
if not user.id == user_created and not user.level == AuthLevel.ADMIN:
raise HTTPException(403, "Access denied")
file_extension = os.path.splitext(image_id)[1]
media_type = mimetypes.types_map.get(file_extension)
return StreamingResponse(stream, media_type=media_type)
return StreamingResponse(stream, headers={'Content-Type': media_type})
@router.post("/upload", description="Upload an image to the database")