E-Paper-Calendar/Calendar/AgendaListDesign.py

from DesignEntity import DesignEntity
from Assets import defaultfontsize, colors
from datetime import datetime, date, timedelta
from TableTextDesign import TableTextDesign
from PIL import ImageDraw
from TextFormatter import date_summary_str, event_prefix_str

line_width = 1

class AgendaListDesign (DesignEntity):
    '''Lists upcoming events in chronological order and groups them by days'''
    def __init__ (self, size, calendar, line_spacing = 3, col_spacing = 8, text_size = defaultfontsize, start_date = date.today(), always_add_start_row = True):
        super(AgendaListDesign, self).__init__(size)
        self.calendar = calendar
        self.line_spacing = line_spacing
        self.col_spacing = col_spacing
        self.text_size = text_size
        self.start_dt = datetime(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 + int(self.text_size)
        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
        while len(self.infos) < self.__event_number__:
            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)
        
        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 = TableTextDesign(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=line_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 ]