61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
from fastapi import APIRouter, UploadFile, HTTPException
|
|
import httpx
|
|
|
|
from app.config import settings
|
|
from app.images.shemas import SImageUrl, SAvatarUrl
|
|
|
|
router = APIRouter(
|
|
prefix="/images",
|
|
tags=["Загрузка картинок"]
|
|
)
|
|
|
|
|
|
async def upload_file(file: UploadFile, remote_server_url: str) -> httpx.Response:
|
|
async with httpx.AsyncClient() as client:
|
|
try:
|
|
file_content = await file.read()
|
|
response = await client.post(remote_server_url, files={"file": file_content})
|
|
if response.status_code != 200:
|
|
raise HTTPException(status_code=response.status_code,
|
|
detail="Ошибка при загрузке файла на удаленный сервер")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
return response
|
|
|
|
|
|
async def upload_file_returning_str(file: UploadFile, sub_str: str) -> str:
|
|
remote_server_url = settings.IMAGE_UPLOAD_SERVER + sub_str
|
|
response = await upload_file(file, remote_server_url)
|
|
image_url = response.json()['image_url']
|
|
image_url = settings.IMAGE_UPLOAD_SERVER + image_url
|
|
return image_url
|
|
|
|
|
|
async def upload_avatar_returning_dict(file: UploadFile, sub_str: str) -> dict:
|
|
remote_server_url = settings.IMAGE_UPLOAD_SERVER + sub_str
|
|
response = await upload_file(file, remote_server_url)
|
|
image_url = response.json()['image_url']
|
|
hex_color = response.json()['hex_color']
|
|
image_url = settings.IMAGE_UPLOAD_SERVER + image_url
|
|
return {'image_url': image_url, 'hex_color': hex_color}
|
|
|
|
|
|
@router.post('/upload_avatar', response_model=SAvatarUrl)
|
|
async def upload_avatar(file: UploadFile):
|
|
image_data = await upload_avatar_returning_dict(file, "upload_avatar")
|
|
return image_data
|
|
|
|
|
|
@router.post('/upload_image', response_model=SImageUrl)
|
|
async def upload_image(file: UploadFile):
|
|
image_url = await upload_file_returning_str(file, "upload_image")
|
|
return {'image_url': image_url}
|
|
|
|
|
|
@router.post('/upload_background', response_model=list[SImageUrl])
|
|
async def upload_background(file: UploadFile):
|
|
remote_server_url = settings.IMAGE_UPLOAD_SERVER + "upload_background"
|
|
response = await upload_file(file, remote_server_url)
|
|
uploaded_backgrounds = response.json()
|
|
uploaded_backgrounds = [{'image_url': settings.IMAGE_UPLOAD_SERVER + background}for background in uploaded_backgrounds]
|
|
return uploaded_backgrounds
|