nemunaire/epaper
1
0
Fork 0
Code Issues Pull requests Releases Activity

Refactor alerts

Browse source
This commit is contained in:
nemunaire 2022-08-14 15:32:52 +02:00
commit 606895f63b
3 changed files with 111 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

97
modules/weather.py
View file

@ -6,6 +6,7 @@ from PIL import Image, ImageDraw, ImageFont
import pygal
from . import display_longtext
from .weather_api import WeatherAPI
def draw_format_infos(infos, width, height, fnt_R, fnt_B, label_margin, align_height=0, margin_bf_first=True, **kwargs):
@ -39,39 +40,6 @@ def draw_format_infos(infos, width, height, fnt_R, fnt_B, label_margin, align_he
return image
def display_longtext(draw, pos, text, font, anchor="lt", maxwidth=9999, **kwargs):
x,y = pos
lines = []
mainlines = text.split("\n")
for line in mainlines:
words = line.split(" ")
cur = ""
while len(words) > 0:
added = (" " if len(cur) > 0 else "") + words[0]
width = font.getsize(cur + added)[0]
if width < maxwidth:
words.pop(0)
cur += added
else:
lines.append(cur)
cur = words.pop(0)
if len(cur) > 0:
lines.append(cur)
line_height = font.getsize("test")[1]
if anchor[1] == "m":
y -= line_height * len(lines) / 2
elif anchor[1] == "b":
y -= line_height * len(lines)
for line in lines:
draw.text((x,y), line, font=font, anchor=anchor, **kwargs)
y += line_height
return line_height * len(lines)
class WeatherToolbarModule:
def __init__(self):
@ -330,24 +298,13 @@ class WeeklyWeatherModule:
return image
class WeatherAlertsModule:
class WeatherAlerts:
def __init__(self):
self.icon_size = 50
def draw_module(self, config, width, height):
image = Image.new('RGBA', (width, height), "#000")
draw = ImageDraw.Draw(image)
align = 0
def gen_alerts(self):
alerts = []
if WeatherAPI().has_alerts():
alerts = WeatherAPI().get_alerts()
fnt_R = ImageFont.truetype(config.fnt_R_path, 16)
fnt_B = ImageFont.truetype(config.fnt_RB_path, 16)
align = 9
for alert in alerts:
for alert in WeatherAPI().get_alerts():
if alert["severity"] == "watch" or alert["title"].startswith("Moderate"):
icon = "wi-small-craft-advisory.png"
elif alert["severity"] == "warning":
@ -355,43 +312,21 @@ class WeatherAlertsModule:
else:
icon = None
if icon is not None:
color_img = Image.new('RGB', (self.icon_size, self.icon_size), "#fff")
icon_img = Image.open("icons/" + icon).resize((self.icon_size, self.icon_size))
image.paste(color_img, (0, align - 5), icon_img)
draw.text(
(self.icon_size - 5, align),
alert["title"],
fill="white", anchor="lt", font=fnt_B
)
startTime = WeatherAPI().read_timestamp(alert["time"])
endTime = WeatherAPI().read_timestamp(alert["expires"])
# Show alert timing if under a day
if startTime.hour != endTime.hour:
draw.text(
(self.icon_size + fnt_B.getsize(alert["title"])[0], align + 3),
startTime.strftime(("%x " if startTime.day != datetime.now().day else "") + "%X") + " - " + endTime.strftime(("%x " if startTime.day != endTime.day else "") + "%X"),
fill="white", anchor="lt", font=fnt_R
)
subtitle = startTime.strftime(("%x " if startTime.day != datetime.now().day else "") + "%X") + " - " + endTime.strftime(("%x " if startTime.day != endTime.day else "") + "%X")
elif startTime.day != datetime.now().day:
draw.text(
(self.icon_size + fnt_B.getsize(alert["title"])[0], align + 3),
startTime.strftime("%x"),
fill="white", anchor="lt", font=fnt_R
)
subtitle = startTime.strftime("%x")
else:
subtitle = ""
align += fnt_B.getsize(alert["title"])[1]
alerts.append({
"icon": icon,
"title": alert["title"],
"subtitle": subtitle,
"description": alert["description"],
})
align += display_longtext(
draw,
(self.icon_size - 5, align),
alert["description"],
fill="white", font=fnt_R,
maxwidth=width-self.icon_size-5
)
align += 7
image = image.crop((0,0,width, align))
return image
return alerts