added function to decode uploaded image data
This commit is contained in:
@@ -1,6 +1,7 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
from base64 import b64decode
|
||||||
from base64 import b64encode
|
from base64 import b64encode
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@@ -122,6 +123,12 @@ class VisualCommunication(BaseModel):
|
|||||||
image.load()
|
image.load()
|
||||||
return VisualCommunication(name=name, image=image)
|
return VisualCommunication(name=name, image=image)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def decode_image(cls, content: str) -> Image.Image:
|
||||||
|
"""Decode image."""
|
||||||
|
_, content_data = content.split(',')
|
||||||
|
return Image.open(BytesIO(b64decode(content_data)))
|
||||||
|
|
||||||
@field_serializer('image')
|
@field_serializer('image')
|
||||||
def serialize_image(image: Image.Image) -> bytes: # type: ignore
|
def serialize_image(image: Image.Image) -> bytes: # type: ignore
|
||||||
buffer = BytesIO()
|
buffer = BytesIO()
|
||||||
|
|||||||
Reference in New Issue
Block a user