66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
import struct
|
|
|
|
import aiohttp
|
|
|
|
from PIL import Image, ImageOps
|
|
from io import BytesIO
|
|
|
|
sockets = dict()
|
|
|
|
class BinaryEventTypes:
|
|
PREVIEW_IMAGE = 1
|
|
UNENCODED_PREVIEW_IMAGE = 2
|
|
|
|
async def send_image(image_data, sid=None):
|
|
image_type = image_data[0]
|
|
image = image_data[1]
|
|
max_size = image_data[2]
|
|
quality = image_data[3]
|
|
if max_size is not None:
|
|
if hasattr(Image, 'Resampling'):
|
|
resampling = Image.Resampling.BILINEAR
|
|
else:
|
|
resampling = Image.ANTIALIAS
|
|
|
|
image = ImageOps.contain(image, (max_size, max_size), resampling)
|
|
type_num = 1
|
|
if image_type == "JPEG":
|
|
type_num = 1
|
|
elif image_type == "PNG":
|
|
type_num = 2
|
|
elif image_type == "WEBP":
|
|
type_num = 3
|
|
|
|
bytesIO = BytesIO()
|
|
header = struct.pack(">I", type_num)
|
|
bytesIO.write(header)
|
|
image.save(bytesIO, format=image_type, quality=quality, compress_level=1)
|
|
preview_bytes = bytesIO.getvalue()
|
|
await send_bytes(BinaryEventTypes.PREVIEW_IMAGE, preview_bytes, sid=sid)
|
|
|
|
async def send_socket_catch_exception(function, message):
|
|
try:
|
|
await function(message)
|
|
except (aiohttp.ClientError, aiohttp.ClientPayloadError, ConnectionResetError) as err:
|
|
print("send error:", err)
|
|
|
|
def encode_bytes(event, data):
|
|
if not isinstance(event, int):
|
|
raise RuntimeError(f"Binary event types must be integers, got {event}")
|
|
|
|
packed = struct.pack(">I", event)
|
|
message = bytearray(packed)
|
|
message.extend(data)
|
|
return message
|
|
|
|
async def send_bytes(event, data, sid=None):
|
|
message = encode_bytes(event, data)
|
|
|
|
print("sending image to ", event, sid)
|
|
|
|
if sid is None:
|
|
_sockets = list(sockets.values())
|
|
for ws in _sockets:
|
|
await send_socket_catch_exception(ws.send_bytes, message)
|
|
elif sid in sockets:
|
|
await send_socket_catch_exception(sockets[sid].send_bytes, message) |