refactoring

This commit is contained in:
Maxim Devaev
2022-01-17 13:54:52 +03:00
parent 902fc38778
commit 3ee1948f19
2 changed files with 10 additions and 10 deletions

View File

@@ -26,7 +26,7 @@ import functools
from aiohttp.web import Request
from aiohttp.web import Response
from PIL import Image
from PIL import Image as PilImage
from ....validators.basic import valid_bool
from ....validators.basic import valid_int_f0
@@ -104,7 +104,7 @@ class StreamerApi:
assert not (max_width == snapshot.width and max_height == snapshot.height)
with io.BytesIO(snapshot.data) as snapshot_bio:
with io.BytesIO() as preview_bio:
with Image.open(snapshot_bio) as image:
image.thumbnail((max_width, max_height), Image.ANTIALIAS)
with PilImage.open(snapshot_bio) as image:
image.thumbnail((max_width, max_height), PilImage.ANTIALIAS)
image.save(preview_bio, format="jpeg", quality=quality)
return preview_bio.getvalue()

View File

@@ -25,9 +25,9 @@ import os
import io
import functools
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
from PIL import Image as PilImage
from PIL import ImageDraw as PilImageDraw
from PIL import ImageFont as PilImageFont
from ... import aiotools
@@ -39,8 +39,8 @@ async def make_text_jpeg(width: int, height: int, quality: int, text: str) -> by
@functools.lru_cache(maxsize=10)
def _inner_make_text_jpeg(width: int, height: int, quality: int, text: str) -> bytes:
image = Image.new("RGB", (width, height), color=(0, 0, 0))
draw = ImageDraw.Draw(image)
image = PilImage.new("RGB", (width, height), color=(0, 0, 0))
draw = PilImageDraw.Draw(image)
draw.multiline_text((20, 20), text, font=_get_font(), fill=(255, 255, 255))
with io.BytesIO() as bio:
image.save(bio, format="jpeg", quality=quality)
@@ -48,8 +48,8 @@ def _inner_make_text_jpeg(width: int, height: int, quality: int, text: str) -> b
@functools.lru_cache()
def _get_font() -> ImageFont.FreeTypeFont:
def _get_font() -> PilImageFont.FreeTypeFont:
module_path = sys.modules[__name__].__file__
assert module_path is not None
path = os.path.join(os.path.dirname(module_path), "fonts", "Azbuka04.ttf")
return ImageFont.truetype(path, size=20)
return PilImageFont.truetype(path, size=20)