Debugged inkycal_image

Still needs some work:
- input parameter validation (rotation, layout, colours)
- configuration of panel colors in Web UI

Added suggestion for less dynamic calling of module.generate_image()
This commit is contained in:
Robert Sirre 2020-11-23 22:36:04 +01:00
parent 57cc49be21
commit 650e9bdb92
3 changed files with 60 additions and 14 deletions

View File

@ -232,14 +232,17 @@ class Inkycal:
self.info = f"{runtime.format('D MMM @ HH:mm')} "
for number in range(1, self._module_number):
print(f'Generating image {number}')
name = eval(f"self.module_{number}.name")
generate_im = f'black,colour=self.module_{number}.generate_image()'
save_black = f'black.save("{self.image_folder}/module{number}_black.png", "PNG")'
save_colour = f'colour.save("{self.image_folder}/module{number}_colour.png", "PNG")'
full_command = generate_im+'\n'+save_black+'\n'+save_colour
module = eval(f'self.module_{number}')
try:
exec(full_command)
black,colour=module.generate_image()
black.save(f"{self.image_folder}/module{number}_black.png", "PNG")
colour.save(f"{self.image_folder}/module{number}_colour.png", "PNG")
print('OK!')
self.info += f"module {number}: OK "
except Exception as Error:

View File

@ -67,6 +67,7 @@ class Inkyimage(inkycal_module):
self.rotation = self.config['rotation']
self.layout = self.config['layout']
self.colours = self.config['colours']
# give an OK message
print('{0} loaded'.format(self.name))
@ -91,6 +92,8 @@ class Inkyimage(inkycal_module):
im_height = self.height
im_size = im_width, im_height
logger.info('image size: {} x {} px'.format(im_width, im_height))
logger.info('image path: {}'.format(self.image_path))
logger.info('colors: {}'.format(self.colours))
# Try to open the image if it exists and is an image file
try:
@ -109,13 +112,14 @@ class Inkyimage(inkycal_module):
logger.debug(('image-height:', self.image.height))
# Create an image for black pixels and one for coloured pixels
im_black = Image.new('RGB', size = im_size, color = 'white')
im_colour = Image.new('RGB', size = im_size, color = 'white')
# do the required operations
self._remove_alpha()
self._to_layout()
black, colour = self._map_colours()
black, colour = self._map_colours(self.colours)
# paste the images on the canvas
im_black.paste(black, (self.x, self.y))
@ -125,6 +129,9 @@ class Inkyimage(inkycal_module):
im_black.save(images+self.name+'.png', 'PNG')
im_colour.save(images+self.name+'_colour.png', 'PNG')
# return images
return black, colour
def _rotate(self, angle=None):
"""Rotate the image to a given angle
angle must be one of :[0, 90, 180, 270, 360, 'auto']
@ -134,9 +141,9 @@ class Inkyimage(inkycal_module):
angle = self.rotation
# Check if angle is supported
if angle not in self._allowed_rotation:
print('invalid angle provided, setting to fallback: 0 deg')
angle = 0
# if angle not in self._allowed_rotation:
# print('invalid angle provided, setting to fallback: 0 deg')
# angle = 0
# Autoflip the image if angle == 'auto'
if angle == 'auto':
@ -182,11 +189,11 @@ class Inkyimage(inkycal_module):
im = self.image
if mode == None: mode = self.layout
if mode not in self._allowed_layout:
print('{} is not supported. Should be one of {}'.format(
mode, self._allowed_layout))
print('setting layout to fallback: centre')
mode = 'center'
# if mode not in self._allowed_layout:
# print('{} is not supported. Should be one of {}'.format(
# mode, self._allowed_layout))
# print('setting layout to fallback: centre')
# mode = 'center'
# If mode is center, just center the image
if mode == 'center':

View File

@ -0,0 +1,36 @@
import unittest
from inkycal.modules import Inkyimage as Module
tests = [
{
"position": 1,
"name": "Inkyimage",
"config": {
"size": [528,880],
"path": "https://cdn.britannica.com/s:700x500/84/73184-004-E5A450B5/Sunflower-field-Fargo-North-Dakota.jpg",
"rotation": "0",
"layout": "fill",
"padding_x": 0,
"padding_y": 0,
"fontsize": 12,
"language": "en",
"colours": "bwr"
}
},
]
class module_test(unittest.TestCase):
def test_get_config(self):
print('getting data for web-ui...', end = "")
Module.get_config()
print('OK')
def test_generate_image(self):
for test in tests:
print(f'test {tests.index(test)+1} generating image..')
module = Module(test)
module.generate_image()
print('OK')
if __name__ == '__main__':
unittest.main()