E-Paper-Calendar/Calendar/AgendaListDesign.py

from DesignEntity import DesignEntity
from Assets import defaultfontsize, colors, defaultfont, path
from datetime import datetime, date, timedelta
from TableDesign import TableDesign
from PIL import ImageDraw, ImageFont
from TextFormatter import date_summary_str, event_prefix_str
from settings import line_thickness

separator_width = line_thickness


class AgendaListDesign (DesignEntity):
    '''Lists upcoming events in chronological order and groups them by days'''

    def __init__(self, size, calendar, line_spacing=0, col_spacing=8, text_size=defaultfontsize, start_date=date.today(), always_add_start_row=True, day_limit_foresight=91):
        super(AgendaListDesign, self).__init__(size)
        self.calendar = calendar
        self.line_spacing = line_spacing
        self.col_spacing = col_spacing
        self.text_size = text_size
        self.day_limit_foresight = day_limit_foresight
        self.start_dt = date(start_date.year, start_date.month, start_date.day)
        self.always_add_start_row = always_add_start_row

    def __finish_image__(self):
        self.__calculate_parameter__()
        self.__create_infos_events__()
        self.__draw_infos__()
        self.__draw_lines__()

    def __calculate_parameter__(self):
        self.__line_height__ = self.line_spacing + self.__get_text_height__()
        self.__event_number__ = int(int(self.size[1]) // self.__line_height__)
        self.__date_fontsize__ = self.text_size
        self.__date_linespace__ = self.line_spacing

    def __create_infos_events__(self):
        self.infos = []
        self.cell_props = []
        fetch_day = self.start_dt
        days_foresight = 0
        while len(self.infos) < self.__event_number__ and days_foresight < self.day_limit_foresight:
            day_events = self.calendar.get_day_events(fetch_day)
            fetch_day_added_once = False
            for event in day_events:
                row = [""]
                if fetch_day_added_once is False:
                    row.append(date_summary_str(fetch_day))
                    fetch_day_added_once = True
                else:
                    row.append("")

                row.append(event_prefix_str(event, fetch_day))
                row.append(event.title)
                self.cell_props.append(self.__get_row_props__(event))

                self.infos.append(row)
            fetch_day = fetch_day + timedelta(1)
            days_foresight = days_foresight + 1

        if self.infos[0][1] != date_summary_str(self.start_dt) and self.always_add_start_row:
            row = ["", date_summary_str(self.start_dt), "", ""]
            props = self.__get_row_props__()
            self.infos.insert(0, row)
            self.cell_props.insert(0, props)

    def __draw_infos__(self):
        table = TableDesign(self.size, self.infos, fontsize=self.__date_fontsize__,
                            line_spacing=self.__date_linespace__, col_spacing=self.col_spacing, cell_properties=self.cell_props)
        self.draw_design(table)

    def __draw_lines__(self):
        for i, (_, date, _, _) in enumerate(self.infos[1:]):
            if date is not "":
                self.__draw_line__(i + 1)

    def __draw_line__(self, index):
        ypos = index * self.__line_height__ - self.line_spacing / 2
        pos = (0, ypos)
        positions = [pos, (self.size[0], ypos)]

        ImageDraw.Draw(self.__image__).line(
            positions, fill=colors["fg"], width=separator_width)

    def __get_row_props__(self, event=None):
        color = colors["fg"]
        bg_color = colors["bg"]
        default_cell = {
            "color": color,
            "background_color": bg_color
        }
        if event is not None and event.highlight:
            color = colors["hl"]
        cell = {
            "color": color,
            "background_color": bg_color
        }
        return [default_cell, default_cell, cell, cell]

    def __get_text_height__(self):
        return ImageFont.truetype(path + defaultfont, self.text_size).font.height