Created a single convert_to_bytes function that will take both filename and bytes as input

This commit is contained in:
PySimpleGUI 2020-06-16 10:54:40 -04:00
parent 0cfd770b1a
commit 3da7c7b0da
1 changed files with 17 additions and 20 deletions

View File

@ -5,6 +5,7 @@ import PIL
import io import io
import base64 import base64
import os import os
from typing import Union, Tuple
""" """
Demo Image Album.... displays images on Graph Element and transitions Demo Image Album.... displays images on Graph Element and transitions
@ -18,27 +19,22 @@ import os
G_SIZE = (800,600) G_SIZE = (800,600)
def get_img_filename(f, resize=None):
"""
Resizes an image file and returns the result as a byte string which can be used to update a PySimpleGUI Element
"""
img = PIL.Image.open(f)
return resize_pil_image(img, resize)
def convert_to_bytes(file_or_bytes, resize=None):
'''
Will convert into bytes and optionally resize an image that is a file or a base64 bytes object.
:param file_or_bytes: either a string filename or a bytes base64 image object
:type file_or_bytes: (Union[str, bytes])
:param resize: optional new size
:type resize: (Tuple[int, int] or None)
:return: (bytes) a byte-string object
:rtype: (bytes)
'''
if isinstance(file_or_bytes, str):
img = PIL.Image.open(file_or_bytes)
else:
img = PIL.Image.open(io.BytesIO(base64.b64decode(file_or_bytes)))
def get_img_data(data, resize=None):
"""Generate PIL.Image data using PIL
"""
img = PIL.Image.open(io.BytesIO(base64.b64decode(data)))
return resize_pil_image(img, resize)
def resize_pil_image(img, resize=None):
"""
Resizes a PIL image to a new size or just converts to PNG as a byte string
:param img:
:param resize:
:return:
"""
cur_width, cur_height = img.size cur_width, cur_height = img.size
if resize: if resize:
new_width, new_height = resize new_width, new_height = resize
@ -63,10 +59,11 @@ layout = [[graph]]
window = sg.Window('Scrolling Image Viewer', layout, margins=(0,0), element_padding=(0,0), use_default_focus=False, finalize=True) window = sg.Window('Scrolling Image Viewer', layout, margins=(0,0), element_padding=(0,0), use_default_focus=False, finalize=True)
window.read()
offset, move_amount, direction = 0, 5, 'left' offset, move_amount, direction = 0, 5, 'left'
while True: while True:
file_to_display = os.path.join(folder, fnames[offset]) file_to_display = os.path.join(folder, fnames[offset])
img_data = get_img_filename(file_to_display, resize=G_SIZE) img_data = convert_to_bytes(file_to_display, resize=G_SIZE)
id = graph.draw_image(data=img_data, location=(0, G_SIZE[1])) id = graph.draw_image(data=img_data, location=(0, G_SIZE[1]))
event, values = window.read() event, values = window.read()