47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from fastapi import APIRouter, UploadFile, HTTPException, status
|
|
import httpx
|
|
|
|
from app.config import settings
|
|
from app.images.shemas import SImageUrl
|
|
|
|
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=response.json()["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
|
|
|
|
|
|
@router.post(
|
|
"/upload_avatar",
|
|
status_code=status.HTTP_200_OK,
|
|
response_model=SImageUrl,
|
|
)
|
|
async def upload_avatar(file: UploadFile):
|
|
image_url = await upload_file_returning_str(file, "upload_avatar")
|
|
return {"image_url": image_url}
|
|
|
|
|
|
@router.post(
|
|
"/upload_image",
|
|
status_code=status.HTTP_200_OK,
|
|
response_model=SImageUrl,
|
|
)
|
|
async def upload_image(file: UploadFile):
|
|
image_url = await upload_file_returning_str(file, "upload_image")
|
|
return {"image_url": image_url}
|