urec/uploader_server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Аватарки теперь ресайзятся на 512x512

Browse source
This commit is contained in:
urec56 2024-03-09 16:21:29 +03:00
parent 6a8eed7a07
commit d2db021c95
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
app/main.py
View file

@ -2,11 +2,13 @@ import random
import shutil import shutil
import string import string
import os import os
from io import BytesIO
from fastapi import FastAPI, UploadFile, Query from PIL import Image
from fastapi import FastAPI, UploadFile
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
def generate_random_string(length=16): def generate_random_string(length=16):
@ -41,10 +43,14 @@ app.mount("/static", StaticFiles(directory="app/static"), name="static")
@app.post('/upload_avatar', response_model=dict[str, str]) @app.post('/upload_avatar', response_model=dict[str, str])
async def upload_avatar(file: UploadFile): async def upload_avatar(file: UploadFile):
content = await file.read()
image = Image.open(BytesIO(content))
new_size = (512, 512)
resized_image = image.resize(new_size)
name = generate_random_string() name = generate_random_string()
image_url = f'static/images/avatars/{name}_avatar.png' image_url = f'static/images/avatars/{name}_avatar.png'
with open('app/' + image_url, 'wb+') as file_object: with open('app/' + image_url, 'wb+') as file_object:
shutil.copyfileobj(file.file, file_object) resized_image.save(file_object, format="PNG")
return {'image_url': image_url} return {'image_url': image_url}