Add support for longer update intervals
Load display-sizes from supported_models Cleanup old images on new run of Inkycal
This commit is contained in:
parent
1b94162ac4
commit
ae86daf6b8
@ -21,7 +21,7 @@ logs = logging.getLogger(__name__)
|
||||
logs.setLevel(level=logging.INFO)
|
||||
|
||||
# Get the path to the Inkycal folder
|
||||
top_level = os.path.dirname(os.path.abspath(os.path.dirname(__file__))).split("/inkycal")[0]
|
||||
top_level = "/".join(os.path.dirname(os.path.abspath(os.path.dirname(__file__))).split("/")[:-1])
|
||||
|
||||
# Get path of 'fonts' and 'images' folders within Inkycal folder
|
||||
fonts_location = os.path.join(top_level, "fonts/")
|
||||
|
@ -2,15 +2,14 @@
|
||||
Inkycal ePaper driving functions
|
||||
Copyright by aceisace
|
||||
"""
|
||||
import logging
|
||||
import os
|
||||
import traceback
|
||||
from importlib import import_module
|
||||
|
||||
import PIL
|
||||
from PIL import Image
|
||||
|
||||
from inkycal.custom import top_level
|
||||
from inkycal.display.supported_models import supported_models
|
||||
|
||||
|
||||
def import_driver(model):
|
||||
@ -47,14 +46,12 @@ class Display:
|
||||
except FileNotFoundError:
|
||||
raise Exception('SPI could not be found. Please check if SPI is enabled')
|
||||
|
||||
|
||||
def test(self) -> None:
|
||||
"""Test the display by showing a test image"""
|
||||
# TODO implement test image
|
||||
raise NotImplementedError("Devs were too lazy again, sorry, please try again later")
|
||||
|
||||
|
||||
def render(self, im_black: PIL.Image, im_colour: PIL.Image or None=None) -> None:
|
||||
def render(self, im_black: PIL.Image, im_colour: PIL.Image or None = None) -> None:
|
||||
"""Renders an image on the selected E-Paper display.
|
||||
|
||||
Initlializes the E-Paper display, sends image data and executes command
|
||||
@ -166,26 +163,25 @@ class Display:
|
||||
def get_display_size(cls, model_name) -> (int, int):
|
||||
"""Returns the size of the display as a tuple -> (width, height)
|
||||
|
||||
Looks inside "drivers" folder for the given model name, then returns it's
|
||||
Looks inside supported_models file for the given model name, then returns it's
|
||||
size.
|
||||
|
||||
Args:
|
||||
- model_name: str -> The name of the E-Paper display to get it's size.
|
||||
model_name: str -> The name of the E-Paper display to get it's size.
|
||||
|
||||
Returns:
|
||||
(width, height) ->tuple, showing the size of the display
|
||||
(width, height) representing the size of the display
|
||||
|
||||
Raises:
|
||||
AssertionError: If the display name was not found in the supported models.
|
||||
|
||||
You can use this function directly without creating the Display class:
|
||||
|
||||
>>> Display.get_display_size('model_name')
|
||||
"""
|
||||
try:
|
||||
driver = import_driver(model_name)
|
||||
return driver.EPD_WIDTH, driver.EPD_HEIGHT
|
||||
except:
|
||||
logging.error(f'Failed to load driver for ${model_name}. Check spelling?')
|
||||
print(traceback.format_exc())
|
||||
raise AssertionError("Could not import driver")
|
||||
if model_name in supported_models:
|
||||
return supported_models[model_name]
|
||||
raise AssertionError(f'{model_name} not found in supported models')
|
||||
|
||||
@classmethod
|
||||
def get_display_names(cls) -> list:
|
||||
|
19
inkycal/display/supported_models.py
Normal file
19
inkycal/display/supported_models.py
Normal file
@ -0,0 +1,19 @@
|
||||
supported_models = {
|
||||
'epd_12_in_48': (1304, 984),
|
||||
'epd_7_in_5_colour': (640, 384),
|
||||
'9_in_7': (1200, 825),
|
||||
'epd_5_in_83_colour': (600, 448),
|
||||
'epd_12_in_48_colour': (1304, 984),
|
||||
'epd_4_in_2_colour': (400, 300),
|
||||
'epd_7_in_5_v2': (800, 480),
|
||||
'epd_12_in_48_colour_V2': (1304, 984),
|
||||
'epd_7_in_5': (640, 384),
|
||||
'epd5in83b_V2': (648, 480),
|
||||
'epd_7_in_5_v3': (880, 528),
|
||||
'10_in_3': (1872, 1404),
|
||||
'epd_7_in_5_v2_colour': (800, 480),
|
||||
'epd_4_in_2': (400, 300),
|
||||
'7_in_8': (1872, 1404),
|
||||
'epd_7_in_5_v3_colour': (880, 528),
|
||||
'epd_5_in_83': (600, 448)
|
||||
}
|
@ -3,28 +3,22 @@ Main class for inkycal Project
|
||||
Copyright by aceinnolab
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import glob
|
||||
import hashlib
|
||||
import json
|
||||
from logging.handlers import RotatingFileHandler
|
||||
|
||||
import arrow
|
||||
import numpy
|
||||
import asyncio
|
||||
|
||||
|
||||
from inkycal.custom import *
|
||||
from inkycal.display import Display
|
||||
from inkycal.modules.inky_image import Inkyimage as Images
|
||||
|
||||
from PIL import Image
|
||||
|
||||
# On the console, set a logger to show only important logs
|
||||
# (level ERROR or higher)
|
||||
stream_handler = logging.StreamHandler()
|
||||
stream_handler.setLevel(logging.ERROR)
|
||||
|
||||
|
||||
if not os.path.exists(f'{top_level}/logs'):
|
||||
os.mkdir(f'{top_level}/logs')
|
||||
|
||||
@ -66,7 +60,7 @@ class Inkycal:
|
||||
to improve rendering on E-Papers. Set this to False for 9.7" E-Paper.
|
||||
"""
|
||||
|
||||
def __init__(self, settings_path:str or None=None, render:bool=True):
|
||||
def __init__(self, settings_path: str or None = None, render: bool = True):
|
||||
"""Initialise Inkycal"""
|
||||
|
||||
# Get the release version from setup.py
|
||||
@ -87,7 +81,8 @@ class Inkycal:
|
||||
self.settings = settings
|
||||
|
||||
except FileNotFoundError:
|
||||
raise FileNotFoundError(f"No settings.json file could be found in the specified location: {settings_path}")
|
||||
raise FileNotFoundError(
|
||||
f"No settings.json file could be found in the specified location: {settings_path}")
|
||||
|
||||
else:
|
||||
try:
|
||||
@ -108,6 +103,8 @@ class Inkycal:
|
||||
|
||||
self.show_border = self.settings.get('border_around_modules', False)
|
||||
|
||||
self.cleanup()
|
||||
|
||||
# Load drivers if image should be rendered
|
||||
if self.render:
|
||||
# Init Display class with model in settings file
|
||||
@ -146,7 +143,7 @@ class Inkycal:
|
||||
logger.exception(f'Could not find module: "{module}". Please try to import manually')
|
||||
|
||||
# If something unexpected happened, show the error message
|
||||
except Exception as e:
|
||||
except:
|
||||
logger.exception(f"Exception: {traceback.format_exc()}.")
|
||||
|
||||
# Path to store images
|
||||
@ -158,8 +155,16 @@ class Inkycal:
|
||||
# Give an OK message
|
||||
print('loaded inkycal')
|
||||
|
||||
def countdown(self, interval_mins=None):
|
||||
"""Returns the remaining time in seconds until next display update"""
|
||||
def countdown(self, interval_mins: int or None = None) -> int:
|
||||
"""Returns the remaining time in seconds until next display update.
|
||||
|
||||
Args:
|
||||
- interval_mins = int -> the interval in minutes for the update
|
||||
if no interval is given, the value from the settings file is used.
|
||||
|
||||
Returns:
|
||||
- int -> the remaining time in seconds until next update
|
||||
"""
|
||||
|
||||
# Check if empty, if empty, use value from settings file
|
||||
if interval_mins is None:
|
||||
@ -167,20 +172,30 @@ class Inkycal:
|
||||
|
||||
# Find out at which minutes the update should happen
|
||||
now = arrow.now()
|
||||
update_timings = [(60 - int(interval_mins) * updates) for updates in
|
||||
range(60 // int(interval_mins))][::-1]
|
||||
if interval_mins <= 60:
|
||||
update_timings = [(60 - interval_mins * updates) for updates in range(60 // interval_mins)][::-1]
|
||||
|
||||
# Calculate time in minutes until next update
|
||||
minutes = [_ for _ in update_timings if _ >= now.minute][0] - now.minute
|
||||
# Calculate time in minutes until next update
|
||||
minutes = [_ for _ in update_timings if _ >= now.minute][0] - now.minute
|
||||
|
||||
# Print the remaining time in minutes until next update
|
||||
print(f'{minutes} minutes left until next refresh')
|
||||
# Print the remaining time in minutes until next update
|
||||
print(f'{minutes} minutes left until next refresh')
|
||||
|
||||
# Calculate time in seconds until next update
|
||||
remaining_time = minutes * 60 + (60 - now.second)
|
||||
# Calculate time in seconds until next update
|
||||
remaining_time = minutes * 60 + (60 - now.second)
|
||||
|
||||
# Return seconds until next update
|
||||
return remaining_time
|
||||
# Return seconds until next update
|
||||
return remaining_time
|
||||
else:
|
||||
# Calculate time in minutes until next update using the range of 24 hours in steps of every full hour
|
||||
update_timings = [(60 * 24 - interval_mins * updates) for updates in range(60 * 24 // interval_mins)][::-1]
|
||||
minutes = [_ for _ in update_timings if _ >= now.minute][0] - now.minute
|
||||
remaining_time = minutes * 60 + (60 - now.second)
|
||||
|
||||
print(f'{round(minutes / 60, 1)} hours left until next refresh')
|
||||
|
||||
# Return seconds until next update
|
||||
return remaining_time
|
||||
|
||||
def test(self):
|
||||
"""Tests if Inkycal can run without issues.
|
||||
@ -262,7 +277,6 @@ class Inkycal:
|
||||
print("Refresh needed: {a}".format(a=res))
|
||||
return res
|
||||
|
||||
|
||||
async def run(self):
|
||||
"""Runs main program in nonstop mode.
|
||||
|
||||
@ -346,8 +360,8 @@ class Inkycal:
|
||||
|
||||
# render the image on the display
|
||||
if not self.settings.get('image_hash', False) or self._needs_image_update([
|
||||
(f"{self.image_folder}/canvas.png.hash", im_black),
|
||||
(f"{self.image_folder}/canvas_colour.png.hash", im_colour)
|
||||
(f"{self.image_folder}/canvas.png.hash", im_black),
|
||||
(f"{self.image_folder}/canvas_colour.png.hash", im_colour)
|
||||
]):
|
||||
# render the image on the display
|
||||
display.render(im_black, im_colour)
|
||||
@ -362,7 +376,7 @@ class Inkycal:
|
||||
im_black = upside_down(im_black)
|
||||
|
||||
if not self.settings.get('image_hash', False) or self._needs_image_update([
|
||||
(f"{self.image_folder}/canvas.png.hash", im_black),
|
||||
(f"{self.image_folder}/canvas.png.hash", im_black),
|
||||
]):
|
||||
display.render(im_black)
|
||||
|
||||
@ -557,6 +571,16 @@ class Inkycal:
|
||||
else:
|
||||
self._calibration_state = False
|
||||
|
||||
@staticmethod
|
||||
def cleanup():
|
||||
# clean up old images in image_folder
|
||||
for _file in glob.glob(f"{image_folder}*.png"):
|
||||
try:
|
||||
os.remove(_file)
|
||||
except:
|
||||
logger.error(f"could not remove file: {_file}")
|
||||
pass
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print(f'running inkycal main in standalone/debug mode')
|
||||
|
@ -17,12 +17,35 @@ class TestMain(unittest.TestCase):
|
||||
assert inkycal.settings["model"] == "image_file"
|
||||
assert inkycal.settings["update_interval"] == 5
|
||||
assert inkycal.settings["orientation"] == 0
|
||||
assert inkycal.settings["info_section"] == True
|
||||
assert inkycal.settings["info_section"] is True
|
||||
assert inkycal.settings["info_section_height"] == 70
|
||||
assert inkycal.settings["border_around_modules"] == True
|
||||
assert inkycal.settings["border_around_modules"] is True
|
||||
|
||||
def test_run(self):
|
||||
inkycal = Inkycal(self.settings_path, render=False)
|
||||
inkycal.test()
|
||||
|
||||
def test_countdown(self):
|
||||
inkycal = Inkycal(self.settings_path, render=False)
|
||||
|
||||
remaining_time = inkycal.countdown(5)
|
||||
assert 1 <= remaining_time <= 5 * 60
|
||||
remaining_time = inkycal.countdown(10)
|
||||
assert 1 <= remaining_time <= 10 * 60
|
||||
remaining_time = inkycal.countdown(15)
|
||||
assert 1 <= remaining_time <= 15 * 60
|
||||
remaining_time = inkycal.countdown(20)
|
||||
assert 1 <= remaining_time <= 20 * 60
|
||||
remaining_time = inkycal.countdown(30)
|
||||
assert 1 <= remaining_time <= 30 * 60
|
||||
remaining_time = inkycal.countdown(60)
|
||||
assert 1 <= remaining_time <= 60 * 60
|
||||
|
||||
remaining_time = inkycal.countdown(120)
|
||||
assert 1 <= remaining_time <= 120 * 2 * 60
|
||||
remaining_time = inkycal.countdown(240)
|
||||
assert 1 <= remaining_time <= 240 * 2 * 60
|
||||
remaining_time = inkycal.countdown(600)
|
||||
assert 1 <= remaining_time <= 600 * 2 * 60
|
||||
remaining_time = inkycal.countdown(1200)
|
||||
assert 1 <= remaining_time <= 1200 * 2 * 60
|
||||
|
Loading…
Reference in New Issue
Block a user