mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 08:09:37 +00:00
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
import base64
|
|
import re
|
|
from io import BytesIO
|
|
|
|
import PIL
|
|
import requests
|
|
|
|
min_image_size = 320
|
|
max_image_size = 1024
|
|
supported_formats = ["png", "jpeg", "gif"]
|
|
|
|
|
|
def get_data_uri_from_bytes(data: bytes, encoding_format: str) -> str:
|
|
base64_str = base64.b64encode(data).decode("utf-8")
|
|
return "data:{};base64,{}".format(encoding_format, base64_str)
|
|
|
|
|
|
def get_image_from_bytes(data: bytes) -> PIL.Image:
|
|
return PIL.Image.open(BytesIO(data))
|
|
|
|
|
|
def get_image_from_base64_str(base64_str: str) -> PIL.Image:
|
|
image_data = re.sub("^data:image/.+;base64,", "", base64_str)
|
|
return get_image_from_bytes(base64.b64decode(image_data))
|
|
|
|
|
|
def get_image_from_url(url: str) -> PIL.Image:
|
|
response = requests.get(url)
|
|
return get_image_from_bytes(response.content)
|
|
|
|
|
|
def get_mime_type_from_image(image: PIL.Image) -> str:
|
|
return image.get_format_mimetype()
|
|
|
|
|
|
def get_bytes_from_image(image: PIL.Image) -> bytes:
|
|
format = (
|
|
image.format
|
|
if image.format.lower() in supported_formats
|
|
else supported_formats[0]
|
|
)
|
|
|
|
imgByteArr = BytesIO()
|
|
image.save(imgByteArr, format=format)
|
|
return imgByteArr.getvalue()
|
|
|
|
|
|
def resize_image_to_max(image: PIL.Image):
|
|
if image.width > max_image_size or image.height > max_image_size:
|
|
image.thumbnail((max_image_size, max_image_size), PIL.Image.ANTIALIAS)
|
|
|
|
|
|
def validate_image(image: PIL.Image):
|
|
if image.width < min_image_size or image.height < min_image_size:
|
|
raise ValueError(
|
|
f"Image is too small ({image.width}x{image.height}px). At least {min_image_size}x{min_image_size}px."
|
|
)
|