61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
from utils.loggers import logger
|
|
|
|
# def saveImage(imageData, fileName: str):
|
|
# try:
|
|
# imageFormat = imageData.split(';')[0].split('/')[1]
|
|
# if imageFormat != 'png':
|
|
# logger.error(f"Неподдерживаемый формат изображения: {imageFormat}")
|
|
# return False
|
|
# imageData = imageData.split(';base64,')[1]
|
|
# with open(f"static/images/{fileName}", "wb") as f:
|
|
# f.write(base64.b64decode(imageData))
|
|
# logger.info(f"Изображение {fileName} успешно сохранено")
|
|
# return True
|
|
# except Exception as e:
|
|
# logger.error(f"Ошибка сохранения изображения: {str(e)}")
|
|
# return False
|
|
|
|
# UPLOAD_DIR = "uploads"
|
|
# os.makedirs(UPLOAD_DIR, exist_ok=True)
|
|
|
|
|
|
def saveImage(file_bytes, fileName: str):
|
|
from PIL import Image
|
|
import io
|
|
|
|
# Загружаем изображение через Pillow
|
|
try:
|
|
img = Image.open(io.BytesIO(file_bytes))
|
|
except Exception:
|
|
logger.error("Неподдерживаемый формат изображения")
|
|
return False
|
|
|
|
# Конвертация (если нужно)
|
|
if img.mode not in ("RGB", "RGBA", "P"):
|
|
img = img.convert("RGB")
|
|
|
|
# Сохранение в выбранный формат
|
|
try:
|
|
logger.info(f"Сохраняем изображение {fileName}")
|
|
img.save(fileName, "PNG")
|
|
except Exception as e:
|
|
logger.error(f"Ошибка сохранения изображения: {str(e)}")
|
|
return False
|
|
|
|
logger.info(f"Изображение {fileName} успешно сохранено")
|
|
return True
|
|
|
|
def deleteImage(fileName: str):
|
|
if fileName.endswith("default.png"):
|
|
return True
|
|
try:
|
|
import os
|
|
|
|
logger.info(f"Удаляем изображение {fileName}")
|
|
os.remove(f"static/images/{fileName}")
|
|
logger.info(f"Изображение {fileName} успешно удалено")
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"Ошибка удаления изображения: {str(e)}")
|
|
return False
|