E-Paper-Calendar/Calendar/DayRowDesign.py

from PIL import ImageDraw, Image
from TextDesign import TextDesign
from settings import week_starts_on, owm_paid_subscription
from DesignEntity import DesignEntity
from datetime import datetime
from Assets import weathericons, wpath, fonts, colors
from SingelDayEventListDesign import SingelDayEventListDesign

daynumber_y_size = (1, 0.65)
weekday_y_size = (daynumber_y_size[0], 1 - daynumber_y_size[1])
weekday_ypos = daynumber_y_size[1]
daynumber_fontsize = daynumber_y_size[1] * 0.8
weekday_fontsize = weekday_y_size[1] * 0.75
weathericon_ypos = 0.1
weathericon_height = 1 - 2 * weathericon_ypos
eventlist_xpadding = 5
eventlist_ypos = 0.1
eventlist_y_fontsize = 0.2

font = fonts["light"]

class DayRowDesign (DesignEntity):
    """Detailed view of a given date."""
    def __init__ (self, size, date):
        super(DayRowDesign, self).__init__(size)
        self.__init_image__()
        self.date = date

    def add_weather (self, weather):
        if weather.is_available is False:
            return
        self.__draw_forecast__(weather)

    def add_calendar (self, calendar):
        self.__draw_event_list__(calendar)

    def add_rssfeed (self, rss):
        pass

    def __draw_event_list__ (self, calendar):
        number_width = daynumber_y_size[0] * self.size[1]
        ypos = eventlist_ypos * self.size[1]
        weather_width = 0
        if owm_paid_subscription:
            weather_width = weathericon_height * self.size[1]
        pos = (number_width + eventlist_xpadding, ypos)
        size = (self.size[0] - pos[0] - weather_width, self.size[1] - pos[1])
        fontsize = eventlist_y_fontsize * self.size[1]
        
        events = calendar.get_day_events(self.date)
        rel_dates = [self.date for _ in range(len(events))]
        event_list = SingelDayEventListDesign(size, events, fontsize, line_spacing=0, event_prefix_rel_dates = rel_dates)
        event_list.pos = pos
        self.draw_design(event_list)

    def __draw_forecast__ (self, weather):
        forecast = weather.get_forecast_in_days(self.date.day - datetime.today().day)
        
        if forecast is None:
            return

        height = int(weathericon_height * self.size[1])
        size = (height, height)
        ypos = weathericon_ypos * self.size[1]
        pos = (self.size[0] - ypos - size[0], ypos)
        icon = Image.open(wpath + weathericons[forecast.icon] + ".jpeg")
        resized_icon = icon.resize(size, resample=Image.LANCZOS)
        self.draw(resized_icon, pos)

    def __finish_image__ (self):
        self.__draw_day_number__()
        self.__draw_weekday__()

    def __draw_weekday__ (self):
        font_size = int(weekday_fontsize * self.size[1])
        size = (weekday_y_size[0] * self.size[1], weekday_y_size[1] * self.size[1])
        ypos = weekday_ypos * self.size[1]
        pos = (0, ypos)

        color = self.__get_day_color__()
        week_day_name = self.date.strftime("%a")
        
        week_day = TextDesign(size, text=week_day_name, font=font, color=color, fontsize=font_size, horizontalalignment="center", verticalalignment="top")
        week_day.pos = pos
        week_day.mask = False
        self.draw_design(week_day)

    def __draw_day_number__ (self):
        font_size = int(daynumber_fontsize * self.size[1])
        size = (daynumber_y_size[0] * self.size[1], daynumber_y_size[1] * self.size[1])
        pos = (0, 0)

        day_text = self.__get_day_text__()
        color = self.__get_day_color__()

        number = TextDesign(size, text=day_text, font=font, color=color, fontsize=font_size, horizontalalignment="center", verticalalignment="bottom")
        number.pos = pos
        number.mask = False
        self.draw_design(number)

    def __abs_co__ (self, coordinates):
        return (int(coordinates[0] * self.size[0]),int(coordinates[1] * self.size[1]))

    def __get_day_text__ (self):
        return str(self.date.day)

    def __get_day_color__ (self):
        """Depending on week_starts_on"""
        if week_starts_on == "Monday" and self.date.strftime("%w") == "0":
            return colors["hl"]
        elif week_starts_on == "Sunday" and self.date.strftime("%w") == "6":
            return colors["hl"]
        else:
            return colors["fg"]